Update app.py

app.py

@@ -78,16 +78,22 @@ 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]
+            # --- UPDATED, MORE EXPLICIT PROMPT ---
             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.
             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]
+
+            Format your response exactly like this:
+            STORY:
+            [The story text goes here]
+            CHOICES:
+            1. [First choice on its own line]
+            2. [Second choice on its own line]
             """
 
         llm = Groq(model="llama3-8b-8192", api_key=GROQ_API_KEY)
@@ -164,8 +170,15 @@ async def run_turn(user_input, game_state):
         return
 
     if isinstance(result_event, StoryContext):
-        narrative, choices = result_event.story_part.split("Choices:", 1)
-        story_display_text = f"{textwrap.fill(narrative, width=80)}\n\nChoices:{choices}"
+        narrative, choices_text = result_event.story_part.split("Choices:", 1)
+        
+        # --- THIS IS THE NEW DEFENSIVE CODE ---
+        # It finds any choice number (like " 2." or " 3.") that has a space before it
+        # and replaces that space with a newline character. This forces them to be on separate lines.
+        choices_text = re.sub(r" (\d\.)", r"\n\1", choices_text)
+        # --- END OF NEW CODE ---
+
+        story_display_text = f"{textwrap.fill(narrative, width=80)}\n\nChoices:{choices_text}"
         
         new_game_state = {
             'inventory': result_event.inventory,
@@ -173,12 +186,9 @@ async def run_turn(user_input, game_state):
         }
         inventory_text = f"**Inventory:** {', '.join(new_game_state['inventory']) if new_game_state['inventory'] else 'Empty'}"
 
-        # --- THIS IS THE FINAL, CORRECTED YIELD LOGIC ---
-
-        # 1. Instantly yield a full tuple with a placeholder for the image.
-        # This provides all 4 required output values.
+        # The rest of the function remains the same...
         yield (None, story_display_text, inventory_text, new_game_state)
-
+        
         # 2. Generate the image.
         image_path = None
         if HF_TOKEN: