fix: generate image and change music in parallel

src/images/image_generator.py

@@ -24,6 +24,8 @@ async def generate_image(prompt: str) -> tuple[str, str] | None:
     # Ensure the generated/images directory exists
     output_dir = "generated/images"
     os.makedirs(output_dir, exist_ok=True)
+    
+    logger.info(f"Generating image with prompt: {prompt}")
 
     try:
         response = await client.models.generate_content(
@@ -37,9 +39,7 @@ async def generate_image(prompt: str) -> tuple[str, str] | None:
         # Process the response parts
         image_saved = False
         for part in response.candidates[0].content.parts:
-            if part.text is not None:
-                logger.info(f"Generated text: {part.text}")
-            elif part.inline_data is not None:
+            if part.inline_data is not None:
                 # Create a filename with timestamp
                 timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
                 filename = f"gemini_{timestamp}.png"
@@ -102,9 +102,7 @@ async def modify_image(image_path: str, modification_prompt: str) -> str | None:
         # Process the response parts
         image_saved = False
         for part in response.candidates[0].content.parts:
-            if part.text is not None:
-                logger.info(f"Generated text: {part.text}")
-            elif part.inline_data is not None:
+            if part.inline_data is not None:
                 # Create a filename with timestamp
                 timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
                 filename = f"gemini_modified_{timestamp}.png"

src/main.py

@@ -18,6 +18,7 @@ from game_constructor import (
     load_character_suggestion,
     start_game_with_settings,
 )
+import asyncio
 
 logger = logging.getLogger(__name__)
 
@@ -54,15 +55,18 @@ async def update_scene(user_hash: str, choice):
         story[new_scene]["choices"] = [
             option.option_description for option in response.player_options
         ]
-
-        if response.change_scene.change_scene:
-            img_path, _ = await generate_image(response.change_scene.scene_description)
+        
+        # run both tasks in parallel
+        img_res, _ = await asyncio.gather(
+            generate_image(response.change_scene.scene_description) if response.change_scene.change_scene else asyncio.sleep(0),
+            change_music_tone(user_hash, response.change_music.music_description) if response.change_music.change_music else asyncio.sleep(0)
+        )
+        
+        if img_res and response.change_scene.change_scene:
+            img_path, _ = img_res
             if img_path:
                 story[new_scene]["image"] = img_path
 
-        if response.change_music.change_music:
-            await change_music_tone(user_hash, response.change_music.music_description)
-
     scene = story[state["scene"]]
     return (
         scene["text"],