Upload app.py

app.py

@@ -171,6 +171,11 @@ class SimpleBrowser:
         return self.used_citations.index(cursor)
 
     def get_page_info(self, cursor: int) -> Optional[Dict[str, str]]:
+        # Prioritize link_map as it stores search result metadata
+        if cursor in self.link_map:
+            return self.link_map[cursor]
+            
+        # Fallback to page_stack for opened pages
         if 0 <= cursor < len(self.page_stack):
             url = self.page_stack[cursor]
             page = self.pages.get(url)
@@ -487,7 +492,8 @@ def render_citations(text: str, browser: SimpleBrowser) -> str:
             info = browser.get_page_info(cursor)
             if info and info.get('url'):
                  # Return clickable index link pointing to reference section
-                return f'<a href="#ref-{index}" class="citation-link" style="color:#2563eb;text-decoration:none;font-weight:600;">[{index}]</a>'
+                 # Aligned with generate_html_example.py style (green via CSS class)
+                return f'<a href="#ref-{index}" class="citation-link">[{index}]</a>'
         except:
             pass
         return m.group(0)