Update main.py

main.py

@@ -69,7 +69,7 @@ def preprocess(query: str, k: int) -> tuple[str, str]:
         "ANSWER:\n"
     )
 
-    references = "\n\n## References\n\n"
+    references = ""
     research_abstracts = ""
 
     for i in range(k):
@@ -80,7 +80,7 @@ def preprocess(query: str, k: int) -> tuple[str, str]:
         text = top_five["text"].values[i]
 
         research_abstracts += str(i + i) + ". This excerpt is from: '" + title + "':\n" + text + "\n"
-        references += (
+        references.append(
             "["
             + title.title()
             + "]("
@@ -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".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 ref in list(set(references))])
     
 def postprocess(response: str, bypass_from_preprocessing: str) -> str:
     """