Upload app.py

app.py

@@ -493,12 +493,25 @@ def render_citations(text: str, browser: SimpleBrowser) -> str:
             if info and info.get('url'):
                  # Return clickable index link pointing to reference section
                  # Aligned with generate_html_example.py style (green via CSS class)
-                return f'<a href="#ref-{index}" class="citation-link">[{index}]</a>'
+                return f'<a href="{html.escape(url)}" target="_blank" class="citation-link">[{index}]</a>'
+            
+            # Fallback if no URL
+            return f'<span class="citation-link">[{index}]</span>'
         except:
             pass
         return m.group(0)
 
+    # First pass: replace citations with links
     result = re.sub(r'【(\d+)†L(\d+)(?:-L(\d+))?】', replace_citation, text)
+    
+    # Second pass: Deduplicate adjacent identical citations
+    # Matches: <a ...>[N]</a> followed by optional whitespace and same link
+    # We repeat this until no more changes to handle multiple duplicates
+    while True:
+        new_result = re.sub(r'(<a [^>]+>\[\d+\]</a>)(\s*)\1', r'\1', result)
+        if new_result == result:
+            break
+        result = new_result
 
     # Convert basic markdown to HTML
     result = re.sub(r'\*\*(.+?)\*\*', r'<strong>\1</strong>', result)
@@ -950,8 +963,8 @@ def run_agent_streaming(
             
         # Generate Reference Section
         if browser.used_citations:
-            html_parts.append('<div class="reference-section">')
-            html_parts.append('<div class="reference-title">References</div>')
+            html_parts.append('<details class="reference-section">')
+            html_parts.append('<summary class="reference-title">References</summary>')
             
             for i, cursor in enumerate(browser.used_citations):
                 info = browser.get_page_info(cursor)
@@ -963,17 +976,17 @@ def run_agent_streaming(
                     title = "Unknown Source"
                     
                 ref_item = f'''
-                <a href="{html.escape(url)}" target="_blank" class="reference-item" id="ref-{i}">
+                <div class="reference-item">
                     <div style="display: flex; align-items: baseline;">
                         <span class="ref-number">[{i}]</span>
-                        <span class="ref-text">{html.escape(title)}</span>
+                        <a href="{html.escape(url)}" target="_blank" class="ref-text">{html.escape(title)}</a>
                     </div>
-                    <div class="ref-url">{html.escape(url)}</div>
-                </a>
+                    <div class="ref-url" style="text-align: left;">{html.escape(url)}</div>
+                </div>
                 '''
                 html_parts.append(ref_item)
                 
-            html_parts.append('</div>')
+            html_parts.append('</details>')
             yield ''.join(html_parts)
 
     except Exception as e:
@@ -1464,6 +1477,26 @@ def create_interface():
         border: 1px solid #7dd3fc;
         text-align: right;
     }
+    
+    /* Reference Section Collapsible */
+    .reference-section {
+        margin-top: 40px;
+        border-top: 1px solid #e5e7eb;
+        padding-top: 20px;
+    }
+    
+    .reference-title {
+        font-size: 1.2rem;
+        font-weight: 600;
+        margin-bottom: 16px;
+        color: #111827;
+        cursor: pointer;
+        outline: none;
+    }
+    
+    .ref-url {
+        text-align: left !important;
+    }
 
     .user-message-content {
         line-height: 1.6;