Update main.py

main.py

@@ -68,6 +68,7 @@ def preprocess(query: str, k: int) -> tuple[str, str]:
         tuple[str, str]: A tuple containing the prompt and references
     """
     encoded_query = numpy.expand_dims(model.encode(query), axis=0)
+    print(query, encoded_query)
     faiss.normalize_L2(encoded_query)
     D, I = index.search(encoded_query, k)
     top_five = data.loc[I[0]]
@@ -85,11 +86,11 @@ def preprocess(query: str, k: int) -> tuple[str, str]:
     research_abstracts = ""
 
     for i in range(k):
-        research_abstracts += top_five["bib_dict"].values[i]["abstract"] + "\n"
-        references += (
-            str(i + 1)
-            + ". "
-            + ", ".join(
+        year = str(int(top_five["bib_dict"].values[i]["pub_year"]))
+        abstract = top_five["bib_dict"].values[i]["abstract"]
+        url = "(https://scholar.google.com/citations?view_op=view_citation&citation_for_view=" + top_five["author_pub_id"].values[i]
+        title = top_five["bib_dict"].values[i]["title"]
+        authors = ", ".join(
                 [
                     author.split(" ")[-1]
                     for author in top_five["bib_dict"]
@@ -97,13 +98,18 @@ def preprocess(query: str, k: int) -> tuple[str, str]:
                     .split(" and ")
                 ]
             )
+
+        research_abstracts += str(i + i) + ". '" + title + "' by " + authors + "\n" + abstract + "\n"
+        references += (
+            str(i + 1)
+            + ". "
+            + authors
             + ". ("
-            + str(int(top_five["bib_dict"].values[i]["pub_year"]))
+            + year
             + "). ["
-            + top_five["bib_dict"].values[i]["title"]
+            + title
             + "]"
-            + "(https://scholar.google.com/citations?view_op=view_citation&citation_for_view="
-            + top_five["author_pub_id"].values[i]
+            + url
             + ").\n"
         )