Update main.py

main.py

@@ -96,7 +96,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 i, ref in references.keys()])
+    return prompt, "\n\n### References\n\n"+"\n".join([str(i+1)+". "+ref for i, ref in enumerate(references.keys())])
     
 def postprocess(response: str, bypass_from_preprocessing: str) -> str:
     """