Bug fix.

app.py

@@ -12,8 +12,11 @@ from utils.file_operations import hash_name
 #   9. Load .bibtex file to generate a pre-defined references list. *
 #   1. 把paper改成纯JSON?
 #   2. 实现别的功能
+#   3. Check API Key GPT-4 Support.
+#   8. Re-build some components using `langchain`
+#           - in `references.py`, use PromptTemplates.format -> str
+#           - in `gpt_interation`, use LLM
 # future:
-#   8. Change prompts to langchain
 #   4. add auto_polishing function
 #   12. Change link to more appealing color # after the website is built;
 #   1. Check if there are any duplicated citations

auto_backgrounds.py

@@ -91,7 +91,7 @@ def fake_generator(title, description="", template="ICLR2022", model="gpt-4"):
     return make_archive("sample-output.pdf", filename)
 
 
-def generate_draft(title, description="", template="ICLR2022", model="gpt-4", search_engine="ss", tldr=True, max_kw_refs=14):
+def generate_draft(title, description="", template="ICLR2022", model="gpt-4", search_engine="ss", tldr=True, max_kw_refs=10):
     paper, destination_folder, _ = _generation_setup(title, description, template, model, search_engine, tldr, max_kw_refs)
 
     # todo: `list_of_methods` failed to be generated; find a solution ...

utils/references.py

@@ -112,7 +112,11 @@ def _collect_papers_ss(keyword, counts=3, tldr=False):
         pattern = r'^\w+'
         words = re.findall(pattern, title)
         # return last_name + year_str + title.split(' ', 1)[0]
-        return last_name + year_str + words[0]
+        try:
+            output = last_name + year_str + words[0]
+        except IndexError:
+            output = last_name + year_str + title[:4]
+        return output
 
     def extract_author_info(raw_authors):
         authors = [author['name'] for author in raw_authors]
@@ -120,7 +124,7 @@ def _collect_papers_ss(keyword, counts=3, tldr=False):
         authors_str = " and ".join(authors)
         try:
             last_name = authors[0].split()[-1]
-        except:
+        except IndexError:
             last_name = "ma"
         # pattern = r'^\w+'
         # last_name = re.findall(pattern, authors[0])