Update app.py

app.py

@@ -30,15 +30,17 @@ def load_example_data():
     
     for idx in example_data.keys():
         try:
-            with open(os.path.join("example", example_data[idx]["code_init"]), "r") as f:
+            file = example_data[idx]["code_init"]
+            with open(os.path.join("example", file), "r") as f:
                 example_data[idx]["code_init"] = f.read()
         except FileNotFoundError:
-            print(f"File not found: {example_data[idx]["code_init"]}. Skipping key: {idx}")
+            print(f"File not found: {file}. Skipping key: {idx}")
         try:
-            with open(os.path.join("example", example_data[idx]["code_final"]), "r") as f:
+            file = example_data[idx]["code_final"]
+            with open(os.path.join("example", file), "r") as f:
                 example_data[idx]["code_final"] = f.read()
         except FileNotFoundError:
-            print(f"File not found: {example_data[idx]["code_final"]}. Skipping key: {idx}")     
+            print(f"File not found: {file}. Skipping key: {idx}")     
     return example_data
 
 example_data = load_example_data()
@@ -95,7 +97,7 @@ def start_experiment_agent(hypothesis, plan):
     step_index = 0
     state_complete = False
     # predefined_message = f"Implement the following hypothesis and experiment plan:\n\nHypothesis:\n{hypothesis}\n\nExperiment Plan:\n{plan}"
-    return example_data[index_ex]['code_init'], predefined_action_log, ""
+    return example_data[index_ex]['code_init'], predefined_action_log, "", ""
 
 def submit_feedback(user_feedback, history, previous_response):
     if (not state_extract or not  state_generate or not state_agent):
@@ -114,8 +116,7 @@ def submit_feedback(user_feedback, history, previous_response):
         state_complete = True
         response = "Agent Finished."
         
-
-    return msg, response, example_data[index_ex]['code_init'] if state_complete else example_data[index_ex]['code_final']
+    return msg, response, example_data[index_ex]['code_init'] if state_complete else example_data[index_ex]['code_final'], ""
 
 def load_phase_2_inputs(hypothesis, plan):
     return hypothesis, plan, "# Code implementation will be displayed here after Start ExperimentAgent."
@@ -298,13 +299,13 @@ with gr.Blocks(theme=gr.themes.Default()) as app:
         start_exp_agnet.click(
             fn=start_experiment_agent, 
             inputs=[hypothesis_state, experiment_plan_state], 
-            outputs=[code_display, log, response]
+            outputs=[code_display, log, response, feedback]
         )
         
         submit_button.click(
             fn=submit_feedback,
             inputs=[feedback, log, response],
-            outputs=[log, response, code_display]
+            outputs=[log, response, code_display, feedback]
         )
 
 if __name__ == "__main__":