Update main.py

main.py

@@ -40,7 +40,7 @@ def search(query: str, k: int):
 
     for i in range(k):
         search_results += "### " + top_five["bib_dict"].values[i]["title"] + "\n\n"
-        search_results += str(D[i]) + "\n\n"
+        search_results += str(D[0][i]) + "\n\n"
         if top_five["pub_url"].values[i] is not None:
             search_results += "[Full Text](" + top_five["pub_url"].values[i] + ") "
         if top_five["citedby_url"].values[i] is not None: