Update app.py

app.py

@@ -139,6 +139,8 @@ class StorytellerWorkflow(Workflow):
 # app.py
 
 # The only change is in the function definition line
+# app.py
+
 async def run_turn(user_input, game_state, image_display, story_display, inventory_display):
     if game_state is None:
         game_state = {'inventory': [], 'last_story_part': None}
@@ -164,6 +166,7 @@ async def run_turn(user_input, game_state, image_display, story_display, invento
 
     if isinstance(result_event, StoryContext):
         narrative, choices = result_event.story_part.split("Choices:", 1)
+        # We need a different variable name here to avoid conflict with the component object
         story_display_text = f"{textwrap.fill(narrative, width=80)}\n\nChoices:{choices}"
         
         new_game_state = {
@@ -172,20 +175,20 @@ async def run_turn(user_input, game_state, image_display, story_display, invento
         }
         inventory_text = f"**Inventory:** {', '.join(new_game_state['inventory']) if new_game_state['inventory'] else 'Empty'}"
 
-        # Instantly yield the text
+        # --- KEY CHANGE: THE CORRECT YIELD SYNTAX ---
+        # First, instantly yield the text and state update
         yield {
-            image_display: None,
             story_display: story_display_text,
             inventory_display: inventory_text,
             game_state: new_game_state
         }
 
-        # Generate the image
+        # Then, generate the image in the background
         image_path = None
         if HF_TOKEN:
             image_path = await generate_image(narrative, HF_TOKEN)
         
-        # Yield the final update with the image when it's ready
+        # Finally, yield a second update ONLY for the image component
         if image_path:
             yield {
                 image_display: image_path