#feat added reference section to the end

app.py

@@ -218,9 +218,10 @@ def generate_cited_html(cited_text, citations: dict):
     citation_numbers = {}
     next_number = 1
     citation_count = 0  # To track unique instances of each citation
+    references = "<b>References:</b><br><br>"
 
     def replace_citations(match):
-        nonlocal citation_count, next_number
+        nonlocal citation_count, next_number, references
         citation_id = match.group(1)  # Extract citation number from the match
         ref_data = citations.get(int(citation_id))
 
@@ -243,6 +244,9 @@ def generate_cited_html(cited_text, citations: dict):
 
         if citation_id not in citation_numbers:
             citation_numbers[citation_id] = next_number
+            source = ref_data["source"]
+            content = ref_data["content"]
+            references += f"[{next_number}] {source}<br>-     {content}<br><br>"
             next_number += 1
         citation_number = citation_numbers[citation_id]
 
@@ -266,6 +270,7 @@ def generate_cited_html(cited_text, citations: dict):
 
     # Replace inline citations in the text with the generated HTML
     html_code += re.sub(r"<(\d+)>", replace_citations, cited_text)
+    html_code += "<br><br>" + references
     html_code += "</div>"
     return html_code