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
|