Update app.py

app.py

@@ -138,7 +138,8 @@ class StorytellerWorkflow(Workflow):
 
 # app.py
 
-async def run_turn(user_input, game_state):
+# The only change is in the function definition line
+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}
         event = StartEvent()
@@ -163,7 +164,7 @@ async def run_turn(user_input, game_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}"
+        story_display_text = f"{textwrap.fill(narrative, width=80)}\n\nChoices:{choices}"
         
         new_game_state = {
             'inventory': result_event.inventory,
@@ -174,13 +175,12 @@ async def run_turn(user_input, game_state):
         # Instantly yield the text
         yield {
             image_display: None,
-            story_display: story_display,
+            story_display: story_display_text,
             inventory_display: inventory_text,
             game_state: new_game_state
         }
 
-        # --- KEY CHANGE: ALWAYS GENERATE THE IMAGE ---
-        # Removed the `if result_event.is_new_scene` condition
+        # Generate the image
         image_path = None
         if HF_TOKEN:
             image_path = await generate_image(narrative, HF_TOKEN)
@@ -193,7 +193,6 @@ async def run_turn(user_input, game_state):
             
 def create_demo():
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
-        # State object to hold the workflow instance between turns
         game_state = gr.State(None)
 
         gr.Markdown("# LlamaIndex Workflow: Dynamic Storyteller")
@@ -207,21 +206,26 @@ 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...")
         
-        # When the user submits their choice
+        # Define the inputs and outputs lists
+        # --- THIS IS THE KEY CHANGE ---
+        inputs = [user_input, game_state, image_display, story_display, inventory_display]
+        outputs = [image_display, story_display, inventory_display, game_state]
+
         user_input.submit(
             fn=run_turn,
-            inputs=[user_input, game_state],
-            outputs=[image_display, story_display, inventory_display, game_state]
+            inputs=inputs,
+            outputs=outputs
         )
         
-        # When the app loads for the first time
+        # 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]
         demo.load(
             fn=run_turn,
-            inputs=[gr.State(None), game_state], # Pass empty input and state
-            outputs=[image_display, story_display, inventory_display, game_state]
+            inputs=load_inputs,
+            outputs=outputs
         )
     return demo
-
+    
 if __name__ == "__main__":
     if not GROQ_API_KEY or not HF_TOKEN:
         print("ERROR: API keys not found. Make sure to set GROQ_API_KEY and HF_TOKEN in your Hugging Face Space Secrets.")