Update app.py

app.py

@@ -17,11 +17,11 @@ from llama_index.core.workflow import (
 )
 from llama_index.llms.groq import Groq
 
-# --- Secret Management ---
+# --- Secrets ---
 GROQ_API_KEY = os.environ.get("Groq_Token")
 HF_TOKEN = os.environ.get("HF_TOKEN")
 
-# --- Event and Workflow Definitions (Copied from our notebook) ---
+# --- Event and Workflow Definitions ---
 class StoryContext(Event):
     story_part: str
     inventory: list[str]
@@ -54,22 +54,19 @@ async def generate_image(prompt: str, hf_token: str) -> str | None:
         print(f"Image generation failed: {e}")
         return None
 
-# The full workflow class, now with the required end_story step
+# The full workflow class
 
 class StorytellerWorkflow(Workflow):
     def __init__(self, **kwargs):
         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 = ""
 
         if isinstance(ev, StartEvent):
-            # The prompt for the first turn remains the same.
+            # The prompt for the first turn
             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.
@@ -78,7 +75,7 @@ class StorytellerWorkflow(Workflow):
             """
         elif isinstance(ev, UserChoice):
             last_story_part = await ctx.store.get("last_story_part")
-            # --- UPDATED, MORE EXPLICIT PROMPT ---
+            # --- EXPLICIT PROMPT ---
             prompt = f"""
             You are a creative text adventure game master.
             The story so far: "{last_story_part}"
@@ -100,8 +97,6 @@ class StorytellerWorkflow(Workflow):
         response = await llm.acomplete(prompt)
         response_text = str(response)
 
-        # --- REMOVED THE [NEW_SCENE] CHECK ---
-
         items_found = re.findall(r"\[ADD_ITEM: (.*?)\]", response_text)
         if items_found:
             for item in items_found:
@@ -137,17 +132,6 @@ 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.
-# This is the final, correct version of the main application logic function.
-
-# app.py
-
-# The only change is in the function definition line
-# app.py
-
-# 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.
@@ -165,7 +149,6 @@ async def run_turn(user_input, game_state):
     result_event = await workflow.generate_story_part(event, ctx)
 
     if isinstance(result_event, StoryEnd):
-        # Return a single tuple of 4 values
         yield (None, result_event.final_message, "", None)
         return
 
@@ -176,7 +159,6 @@ async def run_turn(user_input, game_state):
         # 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}"
         
@@ -186,7 +168,6 @@ async def run_turn(user_input, game_state):
         }
         inventory_text = f"**Inventory:** {', '.join(new_game_state['inventory']) if new_game_state['inventory'] else 'Empty'}"
 
-        # The rest of the function remains the same...
         yield (None, story_display_text, inventory_text, new_game_state)
         
         # 2. Generate the image.
@@ -197,7 +178,6 @@ async def run_turn(user_input, 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: