Update app.py

app.py

@@ -17,6 +17,10 @@ state_agent = False
 state_complete = False
 index_ex = "1"
 
+example_text = [
+    "Research Paper 1: Dataset and Baseline for Automatic Student Feedback Analysis",
+    "Research Paper 2: An Empirical Study on the Impact of Code Review on Software Quality"
+]
 
 # Load example JSON file
 def load_example_data():
@@ -39,6 +43,18 @@ def load_example(example_id):
     paper_text = 'Title:\t' +  example['title'] + '\nAbstract:\t' + example['abstract']
     return paper_text
 
+example_text = [load_example(1), load_example(2)]
+
+# Function to handle example clicks
+def load_example_and_set_index(paper_text_input):
+    global index_ex
+    index_ex = str(example_text.index(paper_text_input) + 1)
+    paper_text =  load_example(index_ex)
+
+    return paper_text,  "", "", "", "", "", ""
+
+
+
 ########## Phase 1 ##############
 
 def extract_research_elements(paper_text):
@@ -176,11 +192,15 @@ def info_to_message(info):
     return msg
 
 
+def handle_example_click(example_index):
+    global index_ex
+    index_ex = example_index
+    return load_example(index_ex)  # Simply return the text to display it in the textbox
 
 # Gradio Interface
 with gr.Blocks() as app:
-    gr.Markdown("# MLR- Copilot: Machine Learning Research based on LLM Agents")
-    gr.Markdown("MLR-Copilot is a framework where LLMs mimic researchers’ thought processes, designed to enhance the productivity of machine learning research by automating the generation and implementation of research ideas.It begins with a research paper, autonomously generating and validating these ideas, while incorporating human feedback to help reach executable research outcomes.")
+    gr.Markdown("# MLR- Copilot: AI Research Assistant with Research Agent")
+    gr.Markdown("### MLR-Copilot is a framework where LLMs mimic researchers’ thought processes, designed to enhance the productivity of machine learning research by automating the generation and implementation of research ideas.It begins with a research paper, autonomously generating and validating these ideas, while incorporating human feedback to help reach executable research outcomes.")
 
 
     # Use state variables to store generated hypothesis and experiment plan
@@ -190,9 +210,6 @@ with gr.Blocks() as app:
     ########## Phase 1:  Research Idea Generation Tab ##############
     with gr.Tab("Phase 1: Research Idea Generation"):
         gr.Markdown("### Extract Research Elements and Generate Research Ideas")
-        with gr.Row():
-            example_1_button = gr.Button("Load Example 1: " + example_data["1"]["title"])
-            example_2_button = gr.Button("Load Example 2: " + example_data["2"]["title"])
         
         with gr.Row():
             with gr.Column():
@@ -212,17 +229,30 @@ with gr.Blocks() as app:
                         hypothesis_output = gr.Textbox(label="Generated Hypothesis", lines=20, interactive=False)
                         experiment_plan_output = gr.Textbox(label="Generated Experiment Plan", lines=20, interactive=False)
 
-       
-        # Pre-step: load example         
-        example_1_button.click(
-            fn=lambda: load_example(1), 
-            outputs=[paper_text_input]
+        # with gr.Row():
+        #     example_1_button = gr.Button("Load Example 1: " + example_data["1"]["title"])
+        #     example_2_button = gr.Button("Load Example 2: " + example_data["2"]["title"])
+        # Example buttons
+        gr.Examples(
+            examples=example_text,
+            inputs=[paper_text_input],
+            outputs=[paper_text_input, tasks_output, gaps_output, keywords_output, recent_works_output, hypothesis_output, experiment_plan_output],
+            fn=load_example_and_set_index,
+            run_on_click = True,
+            label="Click an example to load"
         )
+
+       
+        # # Pre-step: load example         
+        # example_1_button.click(
+        #     fn=lambda: load_example(1), x
+        #     outputs=[paper_text_input]
+        # )
         
-        example_2_button.click(
-            fn=lambda: load_example(2), 
-            outputs=[paper_text_input]
-        )
+        # example_2_button.click(
+        #     fn=lambda: load_example(2), 
+        #     outputs=[paper_text_input]
+        # )
     
         # Step 1: Extract Research Elements
         extract_button.click(
@@ -281,4 +311,4 @@ with gr.Blocks() as app:
 
 if __name__ == "__main__":
     step_index = 0
-    app.launch()
+    app.launch(share=True)