Update app.py

app.py

@@ -166,7 +166,6 @@ 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 = {
@@ -175,25 +174,25 @@ 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'}"
 
-        # --- KEY CHANGE: THE CORRECT YIELD SYNTAX ---
-        # First, instantly yield the text and state update
+        # --- THIS IS THE FINAL, CORRECTED LOGIC ---
+
+        # 1. Instantly yield ONLY the visible text components. Do not touch the state yet.
         yield {
             story_display: story_display_text,
             inventory_display: inventory_text,
-            game_state: new_game_state
         }
 
-        # Then, generate the image in the background
+        # 2. Generate the image.
         image_path = None
         if HF_TOKEN:
             image_path = await generate_image(narrative, HF_TOKEN)
         
-        # Finally, yield a second update ONLY for the image component
-        if image_path:
-            yield {
-                image_display: image_path
-            }
-            
+        # 3. Yield the final update, containing BOTH the image and the new game state.
+        yield {
+            image_display: image_path,
+            game_state: new_game_state,
+        }
+        
 def create_demo():
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
         game_state = gr.State(None)