Upload app.py

app.py

@@ -487,6 +487,10 @@ def is_final_answer(text: str) -> bool:
 # ============================================================
 def render_citations(text: str, browser: SimpleBrowser) -> str:
     """Convert citation markers to clickable HTML links with tooltips."""
+    # Store citation HTML to protect from markdown conversion
+    citation_store = {}
+    citation_counter = [0]
+
     def replace_citation(m):
         cursor_str = m.group(1)
         full_match = m.group(0)  # Get the full match to extract line info
@@ -527,7 +531,13 @@ def render_citations(text: str, browser: SimpleBrowser) -> str:
                 # Create citation with tooltip (single line to avoid markdown conversion issues)
                 tooltip_html = f'<span class="citation-tooltip"><div class="citation-tooltip-title">{title_display}</div>{line_html}<div class="citation-tooltip-snippet">{snippet_display}</div><div class="citation-tooltip-url">🔗 {html.escape(domain)}</div></span>'
 
-                return f'<span class="citation-wrapper"><a href="{html.escape(url)}" target="_blank" class="citation-link">[{index}]</a>{tooltip_html}</span>'
+                citation_html = f'<span class="citation-wrapper"><a href="{html.escape(url)}" target="_blank" class="citation-link">[{index}]</a>{tooltip_html}</span>'
+
+                # Store citation HTML and return placeholder
+                placeholder = f'___CITATION_{citation_counter[0]}___'
+                citation_store[placeholder] = citation_html
+                citation_counter[0] += 1
+                return placeholder
 
             # Fallback if no URL
             return f'<span class="citation-link">[{index}]</span>'
@@ -536,23 +546,14 @@ def render_citations(text: str, browser: SimpleBrowser) -> str:
             pass
         return m.group(0)
 
-    # First pass: replace citations with linked citations
+    # First pass: replace citations with placeholders
     result = re.sub(r'[【\[](\d+)†.*?[】\]]', replace_citation, text)
 
     # Second pass: Remove standalone URLs that appear after text (common pattern)
     # This removes URLs that directly follow sentences without proper citation
     result = re.sub(r'(?<=[.!?])\s+(https?://[^\s]+)', '', result)
 
-    # Third pass: Deduplicate adjacent identical citations
-    # Matches: <span class="citation-wrapper">...</span> followed by optional whitespace and same wrapper
-    while True:
-        # Match citation wrapper and deduplicate
-        new_result = re.sub(r'(<span class="citation-wrapper">.*?</span>)(\s*)\1', r'\1', result)
-        if new_result == result:
-            break
-        result = new_result
-
-    # Convert basic markdown to HTML
+    # Third pass: Convert basic markdown to HTML
     result = re.sub(r'\*\*(.+?)\*\*', r'<strong>\1</strong>', result)
     result = re.sub(r'\*(.+?)\*', r'<em>\1</em>', result)
     result = re.sub(r'`(.+?)`', r'<code>\1</code>', result)
@@ -560,6 +561,17 @@ def render_citations(text: str, browser: SimpleBrowser) -> str:
     if not result.startswith('<p>'):
         result = f'<p>{result}</p>'
 
+    # Fourth pass: Restore citation HTML from placeholders
+    for placeholder, citation_html in citation_store.items():
+        result = result.replace(placeholder, citation_html)
+
+    # Fifth pass: Deduplicate adjacent identical citations
+    while True:
+        new_result = re.sub(r'(<span class="citation-wrapper">.*?</span>)(\s*)\1', r'\1', result)
+        if new_result == result:
+            break
+        result = new_result
+
     return result
 
 def render_thinking_streaming(text: str) -> str:
@@ -1545,7 +1557,7 @@ def create_interface():
         bottom: 125%;
         left: 50%;
         transform: translateX(-50%);
-        z-index: 9999;
+        z-index: 999999;
         width: 280px;
         background: linear-gradient(135deg, #ffffff 0%, #f9fafb 100%);
         border: 2px solid #10a37f;