Update app.py

app.py

@@ -141,7 +141,10 @@ class StorytellerWorkflow(Workflow):
 # 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):
+# This is the final and correct version of the main application logic function.
+
+async def run_turn(user_input, game_state):
+    # This function no longer needs the component objects passed in.
     if game_state is None:
         game_state = {'inventory': [], 'last_story_part': None}
         event = StartEvent()
@@ -156,12 +159,8 @@ async def run_turn(user_input, game_state, image_display, story_display, invento
     result_event = await workflow.generate_story_part(event, ctx)
 
     if isinstance(result_event, StoryEnd):
-        yield {
-            image_display: None,
-            story_display: result_event.final_message,
-            inventory_display: "",
-            game_state: None
-        }
+        # Return a single tuple of 4 values
+        yield (None, result_event.final_message, "", None)
         return
 
     if isinstance(result_event, StoryContext):
@@ -174,25 +173,22 @@ 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'}"
 
-        # --- THIS IS THE FINAL, CORRECTED LOGIC ---
+        # --- THIS IS THE FINAL, CORRECTED YIELD 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,
-        }
+        # 1. Instantly yield a full tuple with a placeholder for the image.
+        # This provides all 4 required output values.
+        yield (None, story_display_text, inventory_text, new_game_state)
 
         # 2. Generate the image.
         image_path = None
         if HF_TOKEN:
             image_path = await generate_image(narrative, HF_TOKEN)
         
-        # 3. Yield the final update, containing BOTH the image and the new game state.
-        yield {
-            image_display: image_path,
-            game_state: new_game_state,
-        }
+        # 3. Yield a second, complete tuple with the new image path.
+        yield (image_path, story_display_text, inventory_text, new_game_state)
         
+# The create_demo function should look like this.
+
 def create_demo():
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
         game_state = gr.State(None)
@@ -208,9 +204,8 @@ def create_demo():
                 story_display = gr.Textbox(label="Story", lines=15, interactive=False)
                 user_input = gr.Textbox(label="What do you do?", placeholder="Type your choice and press Enter...")
         
-        # Define the inputs and outputs lists
-        # --- THIS IS THE KEY CHANGE ---
-        inputs = [user_input, game_state, image_display, story_display, inventory_display]
+        # The inputs and outputs lists should be simple.
+        inputs = [user_input, game_state]
         outputs = [image_display, story_display, inventory_display, game_state]
 
         user_input.submit(
@@ -219,8 +214,7 @@ def create_demo():
             outputs=outputs
         )
         
-        # The `load` event doesn't have a `user_input`, so we pass None
-        load_inputs = [gr.State(None), game_state, image_display, story_display, inventory_display]
+        load_inputs = [gr.State(None), game_state]
         demo.load(
             fn=run_turn,
             inputs=load_inputs,