try to implement session

app.py

@@ -107,76 +107,80 @@ class AppSimulator:
         return self.simulation.one_conversation_round(user_input)
 
 
-# Updated chat function
-def chat(user_input, history, simulator):
+def chat(user_input, history, simulator_app):
     """Chat handler that validates input and interacts with the simulator."""
-    response = simulator.chat_interaction(user_input, history)
+    response = simulator_app.chat_interaction(user_input, history)
 
     # Initialize variables for task completion and steps
+
+    # Define the pattern for matching the response
     pattern = r"Task completed! You took (\d+) steps\."
     match = re.match(pattern, response)
 
     if match:
         task_complete = 1
         task_completed_step = int(match.group(1))
-        app_name = simulator.app_name
-        idx = simulator.smallest_index
-        task = simulator.task
+        app_name = simulator_app.app_name
+        idx = simulator_app.smallest_index
+        task = simulator_app.task
         write_task_data(app_name, idx, task, task_complete, task_completed_step)
 
-    return response, history
+    return response
 
-def initialize_and_start_chat(sitemap, simulator):
-    """Initialize the simulator for the current session."""
-    return simulator.initialize_simulator(sitemap)
 
-def give_up(simulator):
+def give_up(simulator_app):
     """Handle the Give-Up action by marking the first incomplete task as abandoned."""
     task_completed = 0
     task_completed_steps = 0
 
-    app_name = simulator.app_name
-    idx = simulator.smallest_index
-    task = simulator.task
+    app_name = simulator_app.app_name
+    idx = simulator_app.smallest_index
+    task = simulator_app.task
 
     write_task_data(app_name, idx, task, task_completed, task_completed_steps)
+
     return "Task marked as abandoned (Give-Up action)."
 
-# Gradio Interface with gr.State for session-specific simulator
+
+# Gradio Interface using ChatInterface
 with gr.Blocks(fill_height=True) as demo:
+    # Simulator instance stored in gr.State for each session
+    simulator_state = gr.State(AppSimulator(openai_api_key=openai_api_key))
+
     gr.Markdown("## Simulator Setup")
 
     # Input fields for initialization
-    sitemap_input = gr.Textbox(label="Sitemap", placeholder="Enter the Hugging Face link to sitemap... (eg. jjz5463/AppStore_synthetic_sitemap)")
+    sitemap_input = gr.Textbox(label="Sitemap", placeholder="Enter the Hugging Face link to sitemap... (eg.jjz5463/AppStore_synthetic_sitemap)")
     initialize_button = gr.Button("Initialize Simulator")
 
-    # Chat interface and give-up button
-    chat_interface = gr.ChatInterface(fn=chat, type='messages')
-    give_up_button = gr.Button("Give Up")
-
-    # Status block to display initialization progress
+    # Status block to display initialization progress with elapsed time
     status = gr.Textbox(label="Status", interactive=False)
 
-    # Simulator instance for each session
-    simulator = gr.State(AppSimulator(openai_api_key=openai_api_key))
+    # Chat interface to handle user interactions
+    chat_interface = gr.ChatInterface(
+        fn=chat,
+        type="messages",
+        additional_inputs=[simulator_state]
+    )
+
+    give_up_button = gr.Button("Give Up")
+
+    # Define the callback function to initialize the simulator and update status
+    def initialize_and_start_chat(sitemap, simulator_app):
+        return simulator_app.initialize_simulator(sitemap)  # Use progress tracking
 
-    # Define button clicks with gr.State
+    # Set up the button click to initialize simulator and update status only
     initialize_button.click(
         fn=initialize_and_start_chat,
-        inputs=[sitemap_input, simulator],
+        inputs=[sitemap_input, simulator_state],
         outputs=status  # Update only the status block
     )
 
-    chat_interface.submit(
-        fn=chat,
-        inputs=["user_input", "history", simulator],
-        outputs=["chat_interface", "history"]
-    )
-
+    # Set up the Give-Up button click to update the dataset
     give_up_button.click(
         fn=give_up,
-        inputs=[simulator],
-        outputs=status
+        inputs=[simulator_state],
+        outputs=status  # Update the status with the give-up message
     )
 
 # Launch the app