On live.

app.py

@@ -2,13 +2,10 @@ import gradio as gr
 import openai
 from auto_backgrounds import generate_backgrounds
 
-# todo: 1. remove repeated entry in bibfile (go to references.py)
-#       2. (maybe) multiple commas error (see Overleaf)
+# todo:
 #       3. create a huggingface space. test it using multiple devices!
 #       4. further polish auto_backgrounds.py. Make backgrounds have multiple subsection.
 #       5. Design a good layout of huggingface space.
-def generate_backgrounds(t1, t2):
-    return "README.md"
 
 def clear_inputs(text1, text2):
     return ("", "")

utils/references.py

@@ -44,7 +44,7 @@ def _collect_papers_arxiv(keyword, counts=3):
             for author in authors:
                 name = author.find(f"{namespace}name").text
                 author_list.append(name)
-            authors_str = " , ".join(author_list)
+            authors_str = " and ".join(author_list)
 
             # Extract the year
             published = entry.find(f"{namespace}published").text
@@ -100,8 +100,6 @@ class References:
         for key, counts in keywords_dict.items():
             self.papers = self.papers + process(key, counts)
 
-        # TODO: remove repeated entries
-        # test this
         seen = set()
         papers = []
         for paper in self.papers:
@@ -147,4 +145,11 @@ class References:
         prompts = {}
         for paper in self.papers:
             prompts[paper["paper_id"]] = paper["abstract"]
-        return prompts
+        return prompts
+
+if __name__ == "__main__":
+    refs = References()
+    keywords_dict = {"machine learning 1": 10, "machine learning 2":10}
+    refs.collect_papers(keywords_dict)
+    for p in refs.papers:
+        print(p["paper_id"])