sc_ma commited on
Commit
4a233ea
1 Parent(s): 05783f8

Remove some incomplete components.

Browse files
Files changed (2) hide show
  1. app.py +6 -1
  2. auto_backgrounds.py +5 -4
app.py CHANGED
@@ -7,7 +7,12 @@ from utils.file_operations import hash_name
7
  # todo:
8
  # 2. update QQ group and Organization cards
9
  # 4. add auto_polishing function
10
- # 5. Use Completion to substitute some simple task (including: writing abstract, conclusion, generate keywords, generate figures...)
 
 
 
 
 
11
 
12
  openai_key = os.getenv("OPENAI_API_KEY")
13
  access_key_id = os.getenv('AWS_ACCESS_KEY_ID')
 
7
  # todo:
8
  # 2. update QQ group and Organization cards
9
  # 4. add auto_polishing function
10
+ # 5. Use some simple method for simple tasks (including: writing abstract, conclusion, generate keywords, generate figures...)
11
+ # 5.1 Use GPT 3.5 for abstract, conclusion, ...
12
+ # 5.2 Use local LLM to generate keywords, figures, ...
13
+ # 5.3 Use embedding to find most related papers (find a paper dataset)
14
+ # 5.4 Use Segmantic Scholar API instead of Arxiv API.
15
+ # 6. get logs when the procedure is not completed.
16
 
17
  openai_key = os.getenv("OPENAI_API_KEY")
18
  access_key_id = os.getenv('AWS_ACCESS_KEY_ID')
auto_backgrounds.py CHANGED
@@ -90,12 +90,13 @@ def fake_generator(title, description="", template="ICLR2022", model="gpt-4"):
90
  def generate_draft(title, description="", template="ICLR2022", model="gpt-4"):
91
  paper, destination_folder, _ = _generation_setup(title, description, template, model)
92
 
93
- print("Generating figures ...")
94
- usage = figures_generation(paper, destination_folder, model="gpt-3.5-turbo")
95
- log_usage(usage, "figures")
 
96
 
97
  # for section in ["introduction", "related works", "backgrounds", "methodology", "experiments", "conclusion", "abstract"]:
98
- for section in ["introduction", "related works", "backgrounds", "experiments", "conclusion", "abstract"]:
99
  try:
100
  usage = section_generation(paper, section, destination_folder, model=model)
101
  log_usage(usage, section)
 
90
  def generate_draft(title, description="", template="ICLR2022", model="gpt-4"):
91
  paper, destination_folder, _ = _generation_setup(title, description, template, model)
92
 
93
+ # todo: `list_of_methods` failed to be generated; find a solution ...
94
+ # print("Generating figures ...")
95
+ # usage = figures_generation(paper, destination_folder, model="gpt-3.5-turbo")
96
+ # log_usage(usage, "figures")
97
 
98
  # for section in ["introduction", "related works", "backgrounds", "methodology", "experiments", "conclusion", "abstract"]:
99
+ for section in ["introduction", "related works", "backgrounds", "abstract"]:
100
  try:
101
  usage = section_generation(paper, section, destination_folder, model=model)
102
  log_usage(usage, section)