fix: switch to gemini-2.5 stable & improve audio generation

src/agent/llm.py

@@ -1,9 +1,7 @@
 """Utility functions for working with the language model."""
 
-import asyncio
 import logging
 
-from google.api_core.exceptions import DeadlineExceeded
 from langchain_google_genai import ChatGoogleGenerativeAI
 
 from config import settings
@@ -12,7 +10,7 @@ from services.google import ApiKeyPool
 logger = logging.getLogger(__name__)
 
 _pool = ApiKeyPool()
-MODEL_NAME = "gemini-2.5-flash-preview-05-20"
+MODEL_NAME = "gemini-2.5-flash"
 
 
 def _get_api_key() -> str:

src/audio/audio_generator.py

@@ -98,21 +98,17 @@ async def cleanup_music_session(user_hash: str):
         del sessions[user_hash]
 
 
-async def update_audio(user_hash: str, request: gr.Request):
+async def update_audio(user_hash: str, game_started: bool):
     """
     Continuously stream audio from the queue as WAV bytes, and clean up
     when the user disconnects.
     """
-    if user_hash == "":
+    if user_hash == "" or not game_started:
         return
 
     logger.info(f"Starting audio update loop for user hash: {user_hash}")
     try:
         while True:
-            if await request.request.is_disconnected():
-                logger.info(f"Client disconnected for user hash {user_hash}.")
-                break
-
             if user_hash not in sessions:
                 await asyncio.sleep(0.5)
                 continue

src/game_constructor.py

@@ -144,6 +144,7 @@ async def start_game_with_settings(
             gr.update(),
             gr.update(),  # game components unchanged
             gr.update(),  # custom choice
+            gr.update(),
         )
 
     character = Character(
@@ -183,4 +184,5 @@ async def start_game_with_settings(
         gr.update(value=scene_image),  # game_image
         gr.update(choices=scene_choices, value=None),  # game_choices
         gr.update(value="", visible=True),  # custom choice
+        gr.update(value=True), # game_started
     )

src/main.py

@@ -118,6 +118,7 @@ async def start_game_with_music(
         gr.update(),
         gr.update(),  # game components unchanged
         gr.update(),  # custom choice unchanged
+        gr.update(), 
     )
 
     # First, get the game interface updates
@@ -143,6 +144,7 @@ with gr.Blocks(
         gr.HTML("<div class='loading-text'>🚀 Starting your adventure...</div>")
 
     ls_user_hash = gr.BrowserState("", "user_hash")
+    ls_game_started = gr.BrowserState(False, "game_started")
 
     # Constructor Interface (visible by default)
     with gr.Column(
@@ -330,6 +332,7 @@ with gr.Blocks(
             game_image,
             game_choices,
             custom_choice,
+            ls_game_started,
         ],
     )
 
@@ -362,9 +365,9 @@ with gr.Blocks(
         inputs=[],
         outputs=[ls_user_hash],
     )
-    ls_user_hash.change(
+    ls_game_started.change(
         fn=update_audio,
-        inputs=[ls_user_hash],
+        inputs=[ls_user_hash, ls_game_started],
         outputs=[audio_out],
     )