Update app.py

app.py

@@ -122,13 +122,25 @@ class pleiasBot:
             return None, None, None
 
 def format_references(text):
-    ref_pattern = r'<ref name="([^"]+)">"([^"]+)"</ref>\.\s*'  # Modified pattern to include the period and whitespace after ref
+    # First, extract all references in order of appearance
+    ref_pattern = r'<ref name="([^"]+)">"([^"]+)"</ref>'
+    references = re.finditer(ref_pattern, text)
     
+    # Create a mapping of reference IDs to their order of appearance
+    ref_mapping = {}
+    ref_number = 1
+    for match in references:
+        ref_id = match.group(1)
+        if ref_id not in ref_mapping:
+            ref_mapping[ref_id] = ref_number
+            ref_number += 1
+    
+    # Now process the text with the consistent reference numbers
     parts = []
     current_pos = 0
-    ref_number = 1
-
-    for match in re.finditer(ref_pattern, text):
+    
+    # Use finditer again to process the actual text
+    for match in re.finditer(ref_pattern + r'\.\s*', text):
         # Add text before the reference
         text_before = text[current_pos:match.start()].rstrip()
         parts.append(text_before)
@@ -137,12 +149,11 @@ def format_references(text):
         ref_id = match.group(1)
         ref_text = match.group(2).strip()
         
-        # Add the reference, keeping the existing structure but adding <br> where whitespace was
-        tooltip_html = f'<span class="tooltip"><strong>[{ref_number}]</strong><span class="tooltiptext"><strong>{ref_id}</strong>: {ref_text}</span></span>.<br>'
+        # Use the mapped reference number
+        tooltip_html = f'<span class="tooltip"><strong>[{ref_mapping[ref_id]}]</strong><span class="tooltiptext"><strong>{ref_id}</strong>: {ref_text}</span></span>.<br>'
         parts.append(tooltip_html)
         
         current_pos = match.end()
-        ref_number += 1
     
     # Add any remaining text
     parts.append(text[current_pos:])