Georgii Savin commited on
Commit
b4c5fb5
·
unverified ·
1 Parent(s): 60e195a

fix: runtime error for image and music generation

Browse files
src/agent/image_agent.py CHANGED
@@ -73,7 +73,7 @@ async def generate_image_prompt(user_hash: str, scene_description: str) -> Chang
73
  Generates a detailed image prompt string based on a scene description.
74
  This prompt is intended for use with an AI image generation model.
75
  """
76
- logger.info(f"Generating image prompt for the current scene: {request_id}")
77
 
78
  state = get_user_state(user_hash)
79
  scene = GAME_STATE_PROMPT.format(
@@ -89,8 +89,8 @@ async def generate_image_prompt(user_hash: str, scene_description: str) -> Chang
89
  response = await image_prompt_generator_llm.ainvoke(
90
  [
91
  SystemMessage(content=IMAGE_GENERATION_SYSTEM_PROMPT),
92
- HumanMessage(content=scene_description),
93
  ]
94
  )
95
- logger.info(f"Image prompt generated: {request_id}")
96
  return response
 
73
  Generates a detailed image prompt string based on a scene description.
74
  This prompt is intended for use with an AI image generation model.
75
  """
76
+ logger.info(f"Generating image prompt for the current scene: {scene_description}")
77
 
78
  state = get_user_state(user_hash)
79
  scene = GAME_STATE_PROMPT.format(
 
89
  response = await image_prompt_generator_llm.ainvoke(
90
  [
91
  SystemMessage(content=IMAGE_GENERATION_SYSTEM_PROMPT),
92
+ HumanMessage(content=scene),
93
  ]
94
  )
95
+ logger.info(f"Image prompt generated")
96
  return response
src/game_constructor.py CHANGED
@@ -7,7 +7,7 @@ from agent.llm_agent import process_user_input
7
  from images.image_generator import generate_image
8
  from game_setting import Character, GameSetting
9
  from agent.runner import process_step
10
- from audio.audio_generator import start_music_generation
11
  import asyncio
12
  from config import settings
13
 
@@ -164,6 +164,9 @@ async def start_game_with_settings(
164
  scene_image = scene.get("image", "")
165
  scene_choices = [ch["text"] for ch in scene.get("choices", [])]
166
 
 
 
 
167
  return (
168
  gr.update(visible=False), # loading indicator
169
  gr.update(visible=False), # constructor_interface
 
7
  from images.image_generator import generate_image
8
  from game_setting import Character, GameSetting
9
  from agent.runner import process_step
10
+ from audio.audio_generator import start_music_generation, change_music_tone
11
  import asyncio
12
  from config import settings
13
 
 
164
  scene_image = scene.get("image", "")
165
  scene_choices = [ch["text"] for ch in scene.get("choices", [])]
166
 
167
+ music_prompt = await generate_music_prompt(user_hash, scene_text)
168
+ asyncio.create_task(change_music_tone(user_hash, music_prompt))
169
+
170
  return (
171
  gr.update(visible=False), # loading indicator
172
  gr.update(visible=False), # constructor_interface