Update app.py

app.py

@@ -129,39 +129,47 @@ class StorytellerWorkflow(Workflow):
 
 
 # --- Gradio UI and Application Logic ---
-
 async def run_turn(user_input, game_state):
-    # game_state holds our workflow instance and the current step
-    workflow, current_step_name, inventory_text = game_state
+    # game_state holds our workflow instance
+    workflow = game_state
 
     # On the first turn, initialize the workflow
     if workflow is None:
         workflow = StorytellerWorkflow()
-        current_step_name = "generate_story_part"
-        # Manually create the first event to kick off the workflow
-        result_event = await workflow.run_step(current_step_name, StartEvent())
+        event = StartEvent()
     else:
-        # For subsequent turns, create a UserChoice event and run the next step
-        result_event = await workflow.run_step("generate_story_part", UserChoice(choice=user_input))
-
-    # Process the result from the workflow
-    if isinstance(result_event, StoryEnd):
-        return None, result_event.final_message, "", (None, None, None)
-
-    if isinstance(result_event, StoryContext):
-        narrative, choices = result_event.story_part.split("Choices:", 1)
+        # 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)
         story_display = f"{textwrap.fill(narrative, width=80)}\n\nChoices:{choices}"
         
         image_path = None
-        if result_event.is_new_scene and HF_TOKEN:
+        if final_event.is_new_scene and HF_TOKEN:
             image_path = await generate_image(narrative, HF_TOKEN)
             
-        inventory_text = f"**Inventory:** {', '.join(result_event.inventory) if result_event.inventory else 'Empty'}"
+        inventory_text = f"**Inventory:** {', '.join(final_event.inventory) if final_event.inventory else 'Empty'}"
 
-        return image_path, story_display, inventory_text, (workflow, "generate_story_part", inventory_text)
+        return image_path, story_display, inventory_text, workflow # Pass the workflow instance to the next turn
     
-    # Should not happen, but a graceful fallback
-    return None, "An unexpected error occurred.", "", 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: