Update app.py

app.py

@@ -129,48 +129,58 @@ class StorytellerWorkflow(Workflow):
 
 
 # --- Gradio UI and Application Logic ---
-async def run_turn(user_input, game_state):
-    # game_state holds our workflow instance
-    workflow = game_state
+# This is the final, correct version of the main application logic function.
 
-    # On the first turn, initialize the workflow
-    if workflow is None:
-        workflow = StorytellerWorkflow()
+async def run_turn(user_input, game_state):
+    # game_state is now a dictionary: {'inventory': [], 'last_story_part': None}
+    
+    # On the first turn, initialize the game state
+    if game_state is None:
+        game_state = {'inventory': [], 'last_story_part': None}
         event = StartEvent()
     else:
         # For subsequent turns, create a UserChoice event
         event = UserChoice(choice=user_input)
 
     # --- THIS IS THE CORRECTED LOGIC ---
-    # We use stream() to run the workflow until the next input is needed
-    final_event = None
-    async for ev in workflow.stream(event):
-        # The loop will run through generate_story_part and stop when it produces
-        # either a StoryEnd event or a StoryContext event, which triggers the next step.
-        # Since the next step (which we removed) would require user input, the stream pauses.
-        final_event = ev.data
-
-    # Process the result from the workflow stream
-    if isinstance(final_event, StoryEnd):
-        # We need to manually call the end_story step to get the final message
-        stop_event = workflow.end_story(final_event)
-        return None, stop_event.result, "", None # Reset the state
-
-    if isinstance(final_event, StoryContext):
-        narrative, choices = final_event.story_part.split("Choices:", 1)
+    # 1. Create a fresh workflow instance for this turn
+    workflow = StorytellerWorkflow()
+
+    # 2. Create a context object and load our game state into its store
+    # This is how we pass the inventory and story history to the step
+    ctx = Context(store={
+        "inventory": game_state['inventory'],
+        "last_story_part": game_state['last_story_part']
+    })
+
+    # 3. Call the specific step method directly
+    result_event = await workflow.generate_story_part(event, ctx)
+    # ---
+
+    # Process the result from the step
+    if isinstance(result_event, StoryEnd):
+        return None, result_event.final_message, "", None # Reset the state
+
+    if isinstance(result_event, StoryContext):
+        narrative, choices = result_event.story_part.split("Choices:", 1)
         story_display = f"{textwrap.fill(narrative, width=80)}\n\nChoices:{choices}"
         
         image_path = None
-        if final_event.is_new_scene and HF_TOKEN:
+        if result_event.is_new_scene and HF_TOKEN:
             image_path = await generate_image(narrative, HF_TOKEN)
             
-        inventory_text = f"**Inventory:** {', '.join(final_event.inventory) if final_event.inventory else 'Empty'}"
-
-        return image_path, story_display, inventory_text, workflow # Pass the workflow instance to the next turn
+        # Update the game state for the next turn
+        new_game_state = {
+            'inventory': result_event.inventory,
+            'last_story_part': result_event.story_part
+        }
+        inventory_text = f"**Inventory:** {', '.join(new_game_state['inventory']) if new_game_state['inventory'] else 'Empty'}"
+
+        return image_path, story_display, inventory_text, new_game_state
     
     # Graceful fallback
     return None, "An unexpected error occurred. The story cannot continue.", "", None
-
+    
 def create_demo():
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
         # State object to hold the workflow instance between turns