Spaces:
Running
Running
Georgii Savin
commited on
fix: runtime error for image and music generation
Browse files- src/agent/image_agent.py +3 -3
- src/game_constructor.py +4 -1
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: {
|
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=
|
93 |
]
|
94 |
)
|
95 |
-
logger.info(f"Image prompt generated
|
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
|