From e442b3b1698a095a348385059a882c0628c0a19b Mon Sep 17 00:00:00 2001
From: "Timothy J. Baek" <timothyjrbeck@gmail.com>
Date: Fri, 23 Aug 2024 16:51:34 +0200
Subject: [PATCH] fix: async image gen automatic1111

---
 backend/apps/images/main.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/backend/apps/images/main.py b/backend/apps/images/main.py
index 25ed2c517..ed7543106 100644
--- a/backend/apps/images/main.py
+++ b/backend/apps/images/main.py
@@ -15,6 +15,7 @@ import json
 import logging
 import re
 import requests
+import asyncio
 
 from utils.utils import (
     get_verified_user,
@@ -533,7 +534,9 @@ async def image_generations(
             if form_data.negative_prompt is not None:
                 data["negative_prompt"] = form_data.negative_prompt
 
-            r = requests.post(
+            # Use asyncio.to_thread for the requests.post call
+            r = await asyncio.to_thread(
+                requests.post,
                 url=f"{app.state.config.AUTOMATIC1111_BASE_URL}/sdapi/v1/txt2img",
                 json=data,
                 headers={"authorization": get_automatic1111_api_auth()},
@@ -553,7 +556,6 @@ async def image_generations(
                     json.dump({**data, "info": res["info"]}, f)
 
             return images
-
     except Exception as e:
         error = e
         if r != None: