fix: make image generation async

src/audio/audio_generator.py

@@ -5,7 +5,6 @@ from config import settings
 import wave
 import queue
 import logging
-import gradio as gr
 import io
 import time
 

src/images/image_generator.py

@@ -9,7 +9,7 @@ import logging
 
 logger = logging.getLogger(__name__)
 
-client = genai.Client(api_key=settings.gemini_api_key.get_secret_value())
+client = genai.Client(api_key=settings.gemini_api_key.get_secret_value()).aio
 
 async def generate_image(prompt: str) -> tuple[str, str] | None:
     """
@@ -26,11 +26,11 @@ async def generate_image(prompt: str) -> tuple[str, str] | None:
     os.makedirs(output_dir, exist_ok=True)
 
     try:
-        response = client.models.generate_content(
+        response = await client.models.generate_content(
             model="gemini-2.0-flash-preview-image-generation",
             contents=prompt,
             config=types.GenerateContentConfig(
-                response_modalities=['TEXT', 'IMAGE']
+                response_modalities=['TEXT', 'IMAGE'],
             )
         )
 
@@ -62,7 +62,7 @@ async def generate_image(prompt: str) -> tuple[str, str] | None:
         return None, None
 
 
-def modify_image(image_path: str, modification_prompt: str) -> str | None:
+async def modify_image(image_path: str, modification_prompt: str) -> str | None:
     """
     Modify an existing image using Google's Gemini model based on a text prompt.
     
@@ -84,14 +84,14 @@ def modify_image(image_path: str, modification_prompt: str) -> str | None:
     
     key = settings.gemini_api_key.get_secret_value()
     
-    client = genai.Client(api_key=key)
+    client = genai.Client(api_key=key).aio
 
     try:
         # Load the input image
         input_image = Image.open(image_path)
         
         # Make the API call with both text and image
-        response = client.models.generate_content(
+        response = await client.models.generate_content(
             model="gemini-2.0-flash-preview-image-generation",
             contents=[modification_prompt, input_image],
             config=types.GenerateContentConfig(