Update app.py

app.py

@@ -131,6 +131,8 @@ class StorytellerWorkflow(Workflow):
 # --- Gradio UI and Application Logic ---
 # This is the final, correct version of the main application logic function.
 
+# This is the final, correct version of the main application logic function.
+
 async def run_turn(user_input, game_state):
     # game_state is now a dictionary: {'inventory': [], 'last_story_part': None}
     
@@ -146,14 +148,14 @@ async def run_turn(user_input, game_state):
     # 1. Create a fresh workflow instance for this turn
     workflow = StorytellerWorkflow()
 
-    # 2. Create a context object and load our game state into its store
-    # This is how we pass the inventory and story history to the step
-    ctx = Context(store={
-        "inventory": game_state['inventory'],
-        "last_story_part": game_state['last_story_part']
-    })
+    # 2. Create an empty context object
+    ctx = Context()
+    
+    # 3. Asynchronously set the values in the context's store
+    await ctx.store.set("inventory", game_state['inventory'])
+    await ctx.store.set("last_story_part", game_state['last_story_part'])
 
-    # 3. Call the specific step method directly
+    # 4. Call the specific step method directly, passing the populated context
     result_event = await workflow.generate_story_part(event, ctx)
     # ---