Update app.py

app.py

@@ -25,7 +25,6 @@ HF_TOKEN = os.environ.get("HF_TOKEN")
 class StoryContext(Event):
     story_part: str
     inventory: list[str]
-    is_new_scene: bool
 
 class SceneReadyEvent(Event):
     pass
@@ -38,7 +37,7 @@ class StoryEnd(Event):
 
 # Helper function to generate an image and return its path
 async def generate_image(prompt: str, hf_token: str) -> str | None:
-    API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
+    API_URL = "https://api-inference.huggingface.co/models/stabilityai/sd-turbo"
     headers = {"Authorization": f"Bearer {hf_token}"}
     full_prompt = f"epic fantasy art, digital painting, cinematic lighting, masterpiece, {prompt}"
     payload = {"inputs": full_prompt}
@@ -62,13 +61,15 @@ class StorytellerWorkflow(Workflow):
         super().__init__(timeout=300, **kwargs)
 
     @step
+# app.py
+
+    @step
     async def generate_story_part(self, ev: StartEvent | UserChoice, ctx: Context) -> StoryContext | StoryEnd:
         inventory = await ctx.store.get("inventory", [])
         prompt = ""
-        is_new_scene_flag = False
 
         if isinstance(ev, StartEvent):
-            is_new_scene_flag = True
+            # The prompt for the first turn remains the same.
             prompt = """
             You are a creative text adventure game master. Your output is for a console game.
             Start a new story about a curious explorer entering a recently discovered, glowing cave.
@@ -77,12 +78,14 @@ class StorytellerWorkflow(Workflow):
             """
         elif isinstance(ev, UserChoice):
             last_story_part = await ctx.store.get("last_story_part")
+            # --- SIMPLIFIED PROMPT ---
+            # Removed the instruction about [NEW_SCENE]
             prompt = f"""
             You are a creative text adventure game master.
             The story so far: "{last_story_part}"
             The player chose: "{ev.choice}"
             The player's inventory: {inventory}
-            Continue the story. IMPORTANT: If the story describes moving to a new location, add the tag `[NEW_SCENE]` to the beginning of the STORY section. Otherwise, do not add the tag.
+            Continue the story.
             If a choice results in an item, use `[ADD_ITEM: item name]`. If the story should end, write "[END]".
             Format your response exactly like this: STORY: [The story text goes here] CHOICES: 1. [First choice] 2. [Second choice]
             """
@@ -91,9 +94,7 @@ class StorytellerWorkflow(Workflow):
         response = await llm.acomplete(prompt)
         response_text = str(response)
 
-        if "[NEW_SCENE]" in response_text:
-            is_new_scene_flag = True
-            response_text = response_text.replace("[NEW_SCENE]", "").strip()
+        # --- REMOVED THE [NEW_SCENE] CHECK ---
 
         items_found = re.findall(r"\[ADD_ITEM: (.*?)\]", response_text)
         if items_found:
@@ -115,9 +116,10 @@ class StorytellerWorkflow(Workflow):
         
         await ctx.store.set("last_story_part", full_story_part)
         await ctx.store.set("inventory", inventory)
-        return StoryContext(story_part=full_story_part, inventory=inventory, is_new_scene=is_new_scene_flag)
-
-    # --- THIS IS THE STEP THAT WAS MISSING ---
+        
+        # Return the simplified event, without the is_new_scene flag
+        return StoryContext(story_part=full_story_part, inventory=inventory)
+        
     @step
     def end_story(self, ev: StoryEnd) -> StopEvent:
         """This step satisfies the workflow validator by providing a path to a StopEvent."""
@@ -134,56 +136,61 @@ class StorytellerWorkflow(Workflow):
 # This is the final, correct version of the main application logic function.
 # This is the final, correct version of the main application logic function.
 
+# app.py
+
 async def run_turn(user_input, game_state):
-    # game_state is now a dictionary: {'inventory': [], 'last_story_part': None}
-    
-    # On the first turn, initialize the game state
     if game_state is None:
         game_state = {'inventory': [], 'last_story_part': None}
         event = StartEvent()
     else:
-        # For subsequent turns, create a UserChoice event
         event = UserChoice(choice=user_input)
 
-    # --- THIS IS THE CORRECTED LOGIC ---
-    # 1. Create a fresh workflow instance for this turn
     workflow = StorytellerWorkflow()
-
-    # 2. Create a context object, PASSING THE WORKFLOW to it.
-    ctx = Context(workflow=workflow) # This is the fix.
-    
-    # 3. Asynchronously set the values in the context's store
+    ctx = Context(workflow=workflow)
     await ctx.store.set("inventory", game_state['inventory'])
     await ctx.store.set("last_story_part", game_state['last_story_part'])
 
-    # 4. Call the specific step method directly, passing the populated context
     result_event = await workflow.generate_story_part(event, ctx)
-    # ---
 
-    # Process the result from the step
     if isinstance(result_event, StoryEnd):
-        return None, result_event.final_message, "", None # Reset the state
+        yield {
+            image_display: None,
+            story_display: result_event.final_message,
+            inventory_display: "",
+            game_state: None
+        }
+        return
 
     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}"
         
-        image_path = None
-        if result_event.is_new_scene and HF_TOKEN:
-            image_path = await generate_image(narrative, HF_TOKEN)
-            
-        # Update the game state for the next turn
         new_game_state = {
             'inventory': result_event.inventory,
             'last_story_part': result_event.story_part
         }
         inventory_text = f"**Inventory:** {', '.join(new_game_state['inventory']) if new_game_state['inventory'] else 'Empty'}"
 
-        return image_path, story_display, inventory_text, new_game_state
-    
-    # Graceful fallback
-    return None, "An unexpected error occurred. The story cannot continue.", "", None
-    
+        # Instantly yield the text
+        yield {
+            image_display: None,
+            story_display: story_display,
+            inventory_display: inventory_text,
+            game_state: new_game_state
+        }
+
+        # --- KEY CHANGE: ALWAYS GENERATE THE IMAGE ---
+        # Removed the `if result_event.is_new_scene` condition
+        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
+        if image_path:
+            yield {
+                image_display: image_path
+            }
+            
 def create_demo():
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
         # State object to hold the workflow instance between turns