Changes: image generation made async for openai

servers/fastapi/api/utils.py

@@ -32,9 +32,9 @@ def get_user_config():
         pass
 
     return UserConfig(
-        LLM=os.getenv("LLM") or existing_config.LLM,
-        OPENAI_API_KEY=os.getenv("OPENAI_API_KEY") or existing_config.OPENAI_API_KEY,
-        GOOGLE_API_KEY=os.getenv("GOOGLE_API_KEY") or existing_config.GOOGLE_API_KEY,
+        LLM=existing_config.LLM or os.getenv("LLM"),
+        OPENAI_API_KEY=existing_config.OPENAI_API_KEY or os.getenv("OPENAI_API_KEY"),
+        GOOGLE_API_KEY=existing_config.GOOGLE_API_KEY or os.getenv("GOOGLE_API_KEY"),
     )
 
 

servers/fastapi/image_processor/images_finder.py

@@ -1,3 +1,4 @@
+import asyncio
 import base64
 import os
 import aiohttp
@@ -35,12 +36,14 @@ async def generate_image(
 
 async def generate_image_openai(prompt: str, output_path: str):
     client = OpenAI()
-    result = client.images.generate(
-        model="dall-e-3",
-        prompt=prompt,
-        n=1,
-        quality="standard",
-        size="1024x1024",
+    result = await asyncio.to_thread(
+        client.images.generate(
+            model="dall-e-3",
+            prompt=prompt,
+            n=1,
+            quality="standard",
+            size="1024x1024",
+        )
     )
     image_url = result.data[0].url
     async with aiohttp.ClientSession() as session: