fix: runtime error with user choice

src/agent/image_agent.py

@@ -68,7 +68,7 @@ class ChangeScene(BaseModel):
 
 image_prompt_generator_llm = create_light_llm(0.1).with_structured_output(ChangeScene)
 
-async def generate_image_prompt(user_hash: str, scene_description: str) -> ChangeScene:
+async def generate_image_prompt(user_hash: str, scene_description: str, last_choice = "No choice yet") -> ChangeScene:
     """
     Generates a detailed image prompt string based on a scene description.
     This prompt is intended for use with an AI image generation model.

src/agent/llm_graph.py

@@ -94,7 +94,7 @@ async def node_player_step(state: GraphState) -> GraphState:
                 "last_choice": state.choice_text,
             }
         )
-        change_scene = await generate_image_prompt(state.user_hash, next_scene["description"])
+        change_scene = await generate_image_prompt(state.user_hash, next_scene["description"], state.choice_text)
         current_image = None
         if scene_id and scene_id in user_state.scenes:
             current_image = user_state.scenes[scene_id].image
@@ -107,7 +107,7 @@ async def node_player_step(state: GraphState) -> GraphState:
                 "change_scene": change_scene,
             }
         )
-        music_task = generate_music_prompt(next_scene["description"])
+        music_task = generate_music_prompt(state.user_hash, next_scene["description"], state.choice_text)
         _, music_prompt = await asyncio.gather(image_task, music_task)
         asyncio.create_task(change_music_tone(state.user_hash, music_prompt))
         state.scene = next_scene

src/agent/music_agent.py

@@ -40,7 +40,7 @@ class MusicPrompt(BaseModel):
 llm = create_light_llm(0.1).with_structured_output(MusicPrompt)
 
 
-async def generate_music_prompt(user_hash: str, scene_description: str) -> str:
+async def generate_music_prompt(user_hash: str, scene_description: str, last_choice = "No choice yet") -> str:
     logger.info(f"Generating music prompt for the current scene: {scene_description}")
 
     state = get_user_state(user_hash)