fix: parallel game generation

src/agent/llm_graph.py

@@ -60,7 +60,11 @@ async def node_init_game(state: GraphState) -> GraphState:
     first_scene = await generate_scene.ainvoke(
         {"user_hash": state.user_hash, "last_choice": "start"}
     )
-    change_scene = await generate_image_prompt(state.user_hash, first_scene["description"])
+    init_description = (
+        f"{first_scene['description']}\n"
+        "NOTE FOR THE ASSISTANT: YOU MUST GENERATE A NEW IMAGE FOR THE STARTING SCENE"
+    )
+    change_scene = await generate_image_prompt(state.user_hash, init_description)
     logger.info(f"Change scene: {change_scene}")
     await generate_scene_image.ainvoke(
         {

src/audio/audio_generator.py

@@ -14,6 +14,7 @@ client = genai.Client(api_key=settings.gemini_api_key.get_secret_value(), http_o
 
 async def generate_music(user_hash: str, music_tone: str, receive_audio):
     if user_hash in sessions:
+        logger.info(f"Music generation already started for user hash {user_hash}, skipping new generation")
         return
     async with (
         client.aio.live.music.connect(model='models/lyria-realtime-exp') as session,
@@ -85,6 +86,10 @@ async def cleanup_music_session(user_hash: str):
 
 def update_audio(user_hash):
     """Continuously stream audio from the queue as WAV bytes."""
+    if user_hash == "":
+        return
+    
+    logger.info(f"Starting audio update loop for user hash: {user_hash}")
     while True:
         if user_hash not in sessions:
             time.sleep(0.5)

src/main.py

@@ -5,8 +5,6 @@ from audio.audio_generator import (
     cleanup_music_session,
 )
 import logging
-from agent.llm_agent import process_user_input
-from images.image_generator import modify_image
 from agent.runner import process_step
 import uuid
 from game_constructor import (
@@ -17,9 +15,8 @@ from game_constructor import (
     load_character_suggestion,
     start_game_with_settings,
 )
-import asyncio
-from game_setting import get_user_story
-from config import settings
+
+CONCURRENCY_LIMIT = 100
 
 
 logger = logging.getLogger(__name__)
@@ -37,10 +34,11 @@ async def return_to_constructor(user_hash: str):
         gr.update(visible=False),  # game_interface
         gr.update(visible=False),  # error_message
     )
-
+    
 async def generate_user_hash():
-    new_hash = str(uuid.uuid4())
-    return gr.update(value=new_hash)
+    hash = str(uuid.uuid4())
+    logger.info(f"Generated user hash: {hash}")
+    return gr.update(value=hash)
 
 async def update_scene(user_hash: str, choice):
     logger.info(f"Updating scene with choice: {choice}")
@@ -333,6 +331,7 @@ with gr.Blocks(
             game_choices,
             custom_choice,
         ],
+        concurrency_limit=CONCURRENCY_LIMIT,
     )
 
     back_btn.click(
@@ -350,21 +349,27 @@ with gr.Blocks(
         fn=update_scene,
         inputs=[local_storage, game_choices],
         outputs=[game_text, game_image, game_choices, custom_choice],
+        concurrency_limit=CONCURRENCY_LIMIT,
     )
 
     custom_choice.submit(
         fn=update_scene,
         inputs=[local_storage, custom_choice],
         outputs=[game_text, game_image, game_choices, custom_choice],
+        concurrency_limit=CONCURRENCY_LIMIT,
     )
 
-    demo.load(fn=generate_user_hash, outputs=[local_storage])
-
     demo.unload(cleanup_music_session)
     demo.load(
+        fn=generate_user_hash,
+        inputs=[],
+        outputs=[local_storage],
+    )
+    local_storage.change(
         fn=update_audio,
         inputs=[local_storage],
         outputs=[audio_out],
+        concurrency_limit=CONCURRENCY_LIMIT,
     )
 
 demo.launch(ssr_mode=False)