Update main.py

main.py

@@ -91,7 +91,7 @@ def preprocess(query: str, k: int) -> tuple[str, str]:
     prompt = prompt.replace("{{ABSTRACTS_GO_HERE}}", research_abstracts)
     prompt = prompt.replace("{{QUERY_GOES_HERE}}", query)
 
-    return prompt, "\n\n## References\n\n"+"\n".join([str(i+1)+". "+ref for ref in list(set(references))])
+    return prompt, "\n\n## References\n\n"+"\n".join([str(i+1)+". "+ref for i, ref in enumerate(list(set(references)))])
     
 def postprocess(response: str, bypass_from_preprocessing: str) -> str:
     """