fixes to reference formatting

app.py

@@ -7,6 +7,15 @@ from ai_generate import generate
 import requests
 from gptzero_free import GPT2PPL
 import language_tool_python
+import re
+
+
+def remove_bracketed_numbers(text):
+    # This regex pattern matches any occurrence of [{int}]
+    pattern = r"^\[\d+\]"
+    # re.sub() replaces all occurrences of the pattern with an empty string
+    cleaned_text = re.sub(pattern, "", text)
+    return cleaned_text
 
 
 def clean_text(text: str) -> str:
@@ -83,6 +92,7 @@ def regenerate_prompt(settings: Dict[str, str]) -> str:
     
     Comments:
     - {settings['user_comments']}
+    - Keep the references the same as the given text in the same format.
 
     Additional requirements:
     - Include {settings['num_examples']} relevant examples or case studies
@@ -236,7 +246,7 @@ def format_references(text: str) -> str:
     in_references = False
 
     for line in lines:
-        if "references" in line.strip().lower():
+        if line.strip().lower() == "references" or line.strip().lower() == "references:":
             in_references = True
             continue
         if in_references:
@@ -246,8 +256,7 @@ def format_references(text: str) -> str:
 
     formatted_refs = []
     for i, ref in enumerate(references, 1):
-        if f"[{i}]" in ref:
-            ref = ref.replace(f"[{i}]", "")
+        ref = remove_bracketed_numbers(ref)
         formatted_refs.append(f"[{i}] {ref}\n")
 
     return "\n\n".join(article_text) + "\n\nReferences:\n" + "\n".join(formatted_refs)