try to implement session

app.py

@@ -107,77 +107,76 @@ class AppSimulator:
         return self.simulation.one_conversation_round(user_input)
 
 
-# Initialize the simulator
-simulator_app = AppSimulator(openai_api_key=openai_api_key)
-
-
-def chat(user_input, history):
+# Updated chat function
+def chat(user_input, history, simulator):
     """Chat handler that validates input and interacts with the simulator."""
-    response = simulator_app.chat_interaction(user_input, history)
+    response = simulator.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.app_name
-        idx = simulator_app.smallest_index
-        task = simulator_app.task
+        app_name = simulator.app_name
+        idx = simulator.smallest_index
+        task = simulator.task
         write_task_data(app_name, idx, task, task_complete, task_completed_step)
 
-    return response
+    return response, history
 
+def initialize_and_start_chat(sitemap, simulator):
+    """Initialize the simulator for the current session."""
+    return simulator.initialize_simulator(sitemap)
 
-def give_up():
+def give_up(simulator):
     """Handle the Give-Up action by marking the first incomplete task as abandoned."""
     task_completed = 0
     task_completed_steps = 0
 
-    app_name = simulator_app.app_name
-    idx = simulator_app.smallest_index
-    task = simulator_app.task
+    app_name = simulator.app_name
+    idx = simulator.smallest_index
+    task = simulator.task
 
     write_task_data(app_name, idx, task, task_completed, task_completed_steps)
-
     return "Task marked as abandoned (Give-Up action)."
 
-
-# Gradio Interface using ChatInterface
+# Gradio Interface with gr.State for session-specific simulator
 with gr.Blocks(fill_height=True) as demo:
     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")
 
-    # Status block to display initialization progress with elapsed time
-    status = gr.Textbox(label="Status", interactive=False)
-
-    # Chat interface to handle user interactions
+    # Chat interface and give-up button
     chat_interface = gr.ChatInterface(fn=chat, type='messages')
-
     give_up_button = gr.Button("Give Up")
 
-    # Define the callback function to initialize the simulator and update status
-    def initialize_and_start_chat(sitemap):
-        return simulator_app.initialize_simulator(sitemap)  # Use progress tracking
+    # Status block to display initialization progress
+    status = gr.Textbox(label="Status", interactive=False)
+
+    # Simulator instance for each session
+    simulator = gr.State(AppSimulator(openai_api_key=openai_api_key))
 
-    # Set up the button click to initialize simulator and update status only
+    # Define button clicks with gr.State
     initialize_button.click(
         fn=initialize_and_start_chat,
-        inputs=[sitemap_input],
+        inputs=[sitemap_input, simulator],
         outputs=status  # Update only the status block
     )
 
-    # Set up the Give-Up button click to update the dataset
+    chat_interface.submit(
+        fn=chat,
+        inputs=["user_input", "history", simulator],
+        outputs=["chat_interface", "history"]
+    )
+
     give_up_button.click(
         fn=give_up,
-        inputs=[],
-        outputs=status  # Update the status with the give-up message
+        inputs=[simulator],
+        outputs=status
     )
 
 # Launch the app