From 36795672e0bf3d986a9888d87e93540e722bbd35 Mon Sep 17 00:00:00 2001
From: sauravniraula <developmentsaurav@gmail.com>
Date: Tue, 13 May 2025 00:23:07 +0545
Subject: [PATCH] Changes: image generation made async for openai

---
 servers/fastapi/api/utils.py                     |  6 +++---
 servers/fastapi/image_processor/images_finder.py | 15 +++++++++------
 2 files changed, 12 insertions(+), 9 deletions(-)

diff --git a/servers/fastapi/api/utils.py b/servers/fastapi/api/utils.py
index f970408d..cd68ab62 100644
--- a/servers/fastapi/api/utils.py
+++ b/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"),
     )
 
 
diff --git a/servers/fastapi/image_processor/images_finder.py b/servers/fastapi/image_processor/images_finder.py
index a06f6562..cfe28e52 100644
--- a/servers/fastapi/image_processor/images_finder.py
+++ b/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: