bug fix.

auto_backgrounds.py

@@ -68,7 +68,7 @@ def _generation_setup(title, description="", template="ICLR2022",
     # print("Generation setup...")
     # paper = {}
     # paper_body = {}
-    llm = GPTModel()
+    llm = GPTModel(model="gpt-3.5-turbo")
 
     # Create a copy in the outputs folder.
     bibtex_path, destination_folder = copy_templates(template, title)

chat.py

@@ -7,29 +7,62 @@ from utils.prompts import SYSTEM
 
 openai_key = os.getenv("OPENAI_API_KEY")
 default_model = os.getenv("DEFAULT_MODEL")
-openai.api_key = openai_key
+if default_model is None:
+    default_model = "gpt-3.5-turbo-16k"
 
-contribution_system_prompt_1 = '''You are an assistant designed to propose potential contributions of a given title of the paper. Ensure follow the following instructions:
-Instruction:
-- Your response should follow the JSON format.
-- Your response should have the following structure: {"contribution1": {"statement": "briefly describe what the contribution is", "reason": "reason why this contribution has not been made by other literatures"}, "contribution2": {"statement": "briefly describe what the contribution is", "reason": "reason why this contribution has not been made by other literatures"}, ...}'''
+openai.api_key = openai_key
 
-contribution_system_prompt_2 = '''You are an assistant designed to criticize the contributions of a paper. You will be provided Paper's Title, References and Contributions. Ensure follow the following instructions:
+paper_system_prompt =  '''You are an assistant designed to propose choices of research direction.
+The user will input questions or some keywords of a fields. You need to generate some paper titles and main contributions. Ensure follow the following instructions:
 Instruction:
 - Your response should follow the JSON format.
 - Your response should have the following structure: 
-{"contribution1": {"statement": "briefly describe what the contribution is", "reason": "reason why this becomes a contribution from the user", "think":"your thought about if this is a novel contribution", "criticism": "reason why this can or cannot be a novel contribution"}, 
-"contribution2": {"statement": "briefly describe what the contribution is", "reason": "reason why this becomes a contribution from the user", "think":"your thought about if this is a novel contribution", "criticism": "reason why this can or cannot be a novel contribution"}, ...}
-- You need to carefully check if the claimed contribution has been made in the provided references, which makes the contribution not novel.  
+{
+    "your suggested paper title": 
+        {
+            "summary": "an overview introducing what this paper will include",
+            "contributions": {
+                "contribution1": {"statement": "briefly describe this contribution", "reason": "reason why this contribution can make this paper outstanding"}, 
+                "contribution2": {"statement": "briefly describe this contribution", "reason": "reason why this contribution can make this paper outstanding"}, 
+                ...
+            }
+        }
+    "your suggested paper title": 
+        {
+            "summary": "an overview introducing what this paper will include",
+            "contributions": {
+                "contribution1": {"statement": "briefly describe this contribution", "reason": "reason why this contribution can make this paper outstanding"}, 
+                "contribution2": {"statement": "briefly describe this contribution", "reason": "reason why this contribution can make this paper outstanding"}, 
+                ...
+            }
+        }
+    ...
+}
+- Please list three to five suggested title and at least three contributions for each paper.
 '''
 
-suggestions_system_prompt = '''You are an assistant designed to help improve the novelty of a paper. You will be provided Paper's Title, References, Criticism, and Contributions. Ensure follow the following instructions:
+
+contribution_system_prompt = '''You are an assistant designed to criticize the contributions of a paper. You will be provided Paper's Title, References and Contributions. Ensure follow the following instructions:
 Instruction:
 - Your response should follow the JSON format.
 - Your response should have the following structure: 
-{"title": {"suggestion": "your suggestion on the title", "new title": "your suggested title based on your suggestion", "reason": "your reason why you want to make such modification based on the references and criticism"},
-"contributions": {"new contribution 1": {"statement": "your proposed new contribution", "reason": "why this is a novel contribution"}, "new contribution 2": {"statement": "your proposed new contribution", "reason": "why this is a novel contribution"}, ...}}
-- Your reason should be based on the references and solve the criticism.
+{
+    "title": "the title provided by the user",
+    "comment": "your thoughts on if this title clearly reflects the key ideas of this paper and explain why"
+    "contributions": {
+        "contribution1": {"statement": "briefly describe what the contribution is", 
+                          "reason": "reason why the user claims it is a contribution", 
+                          "judge": "your thought about if this is a novel contribution and explain why", 
+                          "suggestion": "your suggestion on how to modify the research direction to enhance the novelty "},
+        "contribution2": {"statement": "briefly describe what the contribution is", 
+                          "reason": "reason why the user claims it is a contribution", 
+                          "judge": "your thought about if this is a novel contribution and explain why", 
+                          "suggestion": "your suggestion on how to modify the research direction to enhance the novelty "},
+        ...
+    }
+} 
+- You need to carefully check if the claimed contribution has been made in the provided references, which makes the contribution not novel.  
+- You also need to propose your concerns on if any of contributions could be incremental or just a mild modification on an existing work.
 '''
 
 
@@ -39,47 +72,16 @@ ANNOUNCEMENT = """
 Criticize your paper's contribution by searching related references online! This nice bot will also give you some suggestions.
 """
 
-
-def reset():
-    return "", "", {}, {}, {}
-
-
-def search_refs(title, contributions):
-    ref = References(title=title, description=contributions)
-
+def criticize_my_idea(title, contributions, max_tokens=4096):
+    ref = References(title=title, description=f"{contributions}")
     keywords, _ = llm(systems=SYSTEM["keywords"], prompts=title, return_json=True)
     keywords = {keyword: 10 for keyword in keywords}
-
     ref.collect_papers(keywords)
-    return ref.to_prompts(max_tokens=8192)
-
+    ref_prompt = ref.to_prompts(max_tokens=max_tokens)
 
-def criticize_my_idea(title, contributions, refined_contributions, suggestions):
-    if refined_contributions:
-        cont = {k: {"statement": v["statement"]} for k, v in refined_contributions.items()}
-
-        ref = References(title=title, description=f"{cont}")
-        keywords, _ = llm(systems=SYSTEM["keywords"], prompts=title, return_json=True)
-        keywords = {keyword: 10 for keyword in keywords}
-        ref.collect_papers(keywords)
-        ref_prompt = ref.to_prompts(max_tokens=4096)
-
-        prompt = f"Title: {title}\n References: {ref_prompt}\n Contributions: {cont}"
-        output, _ = llm(systems=contribution_system_prompt_2, prompts=prompt, return_json=True)
-
-        suggestions, _ = llm(systems=suggestions_system_prompt, prompts=str(output), return_json=True)
-
-        return output, ref_prompt, suggestions
-    else:
-        ref = References(title=title, description=f"{contributions}")
-        keywords, _ = llm(systems=SYSTEM["keywords"], prompts=title, return_json=True)
-        keywords = {keyword: 10 for keyword in keywords}
-        ref.collect_papers(keywords)
-        ref_prompt = ref.to_prompts(max_tokens=4096)
-
-        prompt = f"Title: {title}\n References: {ref_prompt}\n Contributions: {contributions}"
-        output, _ = llm(systems=contribution_system_prompt_1, prompts=prompt, return_json=True)
-        return output, ref_prompt, {}
+    prompt = f"Title: {title}\n References: {ref_prompt}\n Contributions: {contributions}"
+    output, _ = llm(systems=contribution_system_prompt, prompts=prompt, return_json=True)
+    return output, ref_prompt
 
 def paste_title(suggestions):
     if suggestions:
@@ -90,7 +92,9 @@ def paste_title(suggestions):
     else:
         return "", "", {}, {}, {}
 
-
+def generate_choices(thoughts):
+    output, _ = llm(systems=paper_system_prompt, prompts=thoughts, return_json=True)
+    return output
 
 
 
@@ -99,26 +103,32 @@ with gr.Blocks() as demo:
     gr.Markdown(ANNOUNCEMENT)
 
     with gr.Row():
-        with gr.Column():
+        with gr.Tab("Make it an idea!"):
+            thoughts_input = gr.Textbox(label="Thoughts")
+            with gr.Accordion("Show prompts", open=False):
+                prompts_1 = gr.Textbox(label="Prompts", interactive=False, value=paper_system_prompt)
+
+            with gr.Row():
+                button_generate_idea = gr.Button("Make it an idea!", variant="primary")
+
+        with gr.Tab("Criticize my idea!"):
             title_input = gr.Textbox(label="Title")
             contribution_input = gr.Textbox(label="Contributions", lines=5)
+            with gr.Accordion("Show prompts", open=False):
+                prompts_2 = gr.Textbox(label="Prompts", interactive=False, value=contribution_system_prompt)
 
             with gr.Row():
-                button_reset = gr.Button("Reset")
-                button_submit = gr.Button("Submit", variant="primary")
+                button_submit = gr.Button("Criticize my idea!", variant="primary")
+
+        with gr.Tab("Make it a paper!"):
+            gr.Markdown("## Coming Soon!")
 
         with gr.Column(scale=1):
             contribution_output = gr.JSON(label="Contributions")
-            suggestions_output = gr.JSON(label="Suggestions")
-            button_copy = gr.Button("Send Title and Contributions to the Left")
-            references_output = gr.JSON(label="References")
-
-    button_reset.click(fn=reset, inputs=[], outputs=[title_input, contribution_input, contribution_output, references_output, suggestions_output])
-    button_submit.click(fn=criticize_my_idea, inputs=[title_input, contribution_input, contribution_output, suggestions_output], outputs=[contribution_output, references_output, suggestions_output])
-    button_copy.click(fn=paste_title, inputs=suggestions_output, outputs=[title_input, contribution_input, contribution_output, references_output, suggestions_output])
-    # clear_button_refs.click(fn=clear_inputs_refs, inputs=[title_refs, slider_refs], outputs=[title_refs, slider_refs])
-    # submit_button_refs.click(fn=wrapped_references_generator,
-    #                          inputs=[title_refs, slider_refs, key], outputs=json_output)
+            with gr.Accordion("References", open=False):
+                references_output = gr.JSON(label="References")
 
+    button_submit.click(fn=criticize_my_idea, inputs=[title_input, contribution_input], outputs=[contribution_output, references_output])
+    button_generate_idea.click(fn=generate_choices, inputs=thoughts_input, outputs=contribution_output)
 demo.queue(concurrency_count=1, max_size=5, api_open=False)
 demo.launch(show_error=True)

utils/gpt_interaction.py

@@ -79,7 +79,7 @@ class GPTModel_API2D_SUPPORT:
 
 
 class GPTModel:
-    def __init__(self, model="gpt-4", temperature=0.9, presence_penalty=0,
+    def __init__(self, model="gpt-3.5-turbo", temperature=0.9, presence_penalty=0,
                  frequency_penalty=0, max_attempts=1, delay=20):
         self.model = model
         self.temperature = temperature