fix citation id ordering

ai_generate.py

@@ -5,11 +5,8 @@ from langchain_core.documents import Document
 from langchain_community.embeddings.sentence_transformer import (
     SentenceTransformerEmbeddings,
 )
-from langchain.schema import StrOutputParser
 from langchain_community.vectorstores import Chroma
 from langchain_text_splitters import RecursiveCharacterTextSplitter
-from langchain import hub
-from langchain_core.output_parsers import StrOutputParser
 from langchain_core.runnables import RunnablePassthrough
 from langchain_groq import ChatGroq
 from langchain_openai import ChatOpenAI

app.py

@@ -212,10 +212,12 @@ def generate_cited_html(cited_text, citations: dict):
     """
 
     # Function to replace each citation with a reference button
+    citation_numbers = {}
+    next_number = 1
     citation_count = 0  # To track unique instances of each citation
 
     def replace_citations(match):
-        nonlocal citation_count
+        nonlocal citation_count, next_number
         citation_id = match.group(1)  # Extract citation number from the match
         ref_data = citations.get(int(citation_id))
 
@@ -236,6 +238,11 @@ def generate_cited_html(cited_text, citations: dict):
         else:
             source_html = f'<span class="source">{ref_data["source"]}</span>'
 
+        if citation_id not in citation_numbers:
+            citation_numbers[citation_id] = next_number
+            next_number += 1
+        citation_number = citation_numbers[citation_id]
+
         # Unique id for each reference button and popup
         unique_id = f"{citation_id}-{citation_count}"
         citation_count += 1
@@ -243,7 +250,7 @@ def generate_cited_html(cited_text, citations: dict):
         # HTML code for the reference button and popup with formatted content
         button_html = f"""
         <span class="reference-container">
-        <label for="ref-toggle-{unique_id}" class="reference-btn" onclick="closeReferencePanes(); document.getElementById('ref-toggle-{unique_id}').checked = true;">{int(citation_id)+1}</label>
+        <label for="ref-toggle-{unique_id}" class="reference-btn" onclick="closeReferencePanes(); document.getElementById('ref-toggle-{unique_id}').checked = true;">{citation_number}</label>
         <input type="radio" id="ref-toggle-{unique_id}" name="reference" />
         <span class="reference-popup">
             <span class="close-btn" onclick="document.getElementById('ref-toggle-{unique_id}').checked = false;">&times;</span>