Update app.py

app.py

@@ -12,6 +12,7 @@ import json
 # Global variables to store session state
 env = None
 agent = None
+state_example = False
 state_extract = False
 state_generate = False
 state_agent = False
@@ -57,7 +58,8 @@ 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
+    global index_ex, state_example
+    state_example = True
     index_ex = str(example_text.index(paper_text_input) + 1)
     paper_text =  load_example(index_ex)
 
@@ -68,7 +70,9 @@ def load_example_and_set_index(paper_text_input):
 ########## Phase 1 ##############
 
 def extract_research_elements(paper_text):
-    global state_extract, index_ex
+    global state_extract, index_ex, state_example
+    if not state_example:
+        return "", "", "", ""
     state_extract = True
     example = example_data[index_ex]
     tasks = example['research_tasks']
@@ -80,7 +84,7 @@ def extract_research_elements(paper_text):
 
 # Step 2: Generate Research Hypothesis and Experiment Plan
 def generate_and_store(tasks, gaps, keywords, recent_works):
-    if (not state_extract):
+    if (not state_extract or not state_example):
         return "", "", "", ""
     global state_generate, index_ex
     state_generate = True
@@ -90,7 +94,7 @@ def generate_and_store(tasks, gaps, keywords, recent_works):
 
 ##########  Phase 2 & 3  ##############
 def start_experiment_agent(hypothesis, plan):
-    if (not state_extract or not  state_generate):
+    if (not state_extract or not state_generate or not state_example):
         return "", "", ""
     global state_agent, step_index, state_complete
     state_agent = True
@@ -100,7 +104,7 @@ def start_experiment_agent(hypothesis, plan):
     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):
+    if (not state_extract or not  state_generate or not state_agent or not state_example):
         return "", "", ""
     global step_index, state_complete
     step_index += 1