Update api.py

api.py

@@ -4,6 +4,7 @@ import os
 import requests
 from fastapi import FastAPI, HTTPException
 from fastapi.responses import StreamingResponse
+from fastapi.concurrency import run_in_threadpool
 from PIL import Image
 
 # API Configuration for Hugging Face
@@ -17,7 +18,7 @@ def query(payload):
     return response.content
 
 def generate_image(prompt: str) -> Image.Image:
-    # Generate a random seed so even the same prompt produces a new image.
+    # Use a random seed so even identical prompts produce different images.
     random_seed = random.randint(0, 999999)
     payload = {
         "inputs": prompt,
@@ -29,18 +30,18 @@ def generate_image(prompt: str) -> Image.Image:
 app = FastAPI()
 
 @app.get("/")
-def get_generated_image(text: str):
+async def get_generated_image(text: str):
     """
-    API endpoint that returns the generated image directly.
+    API endpoint that waits for the image to be generated and returns it.
     Example usage:
-      https://bowozzz-gen.hf.space/api?text=jokowi
+        https://bowozzz-gen.hf.space/api?text=jokowi
     """
     try:
-        image = generate_image(text)
+        # Run the generate_image function in a threadpool so that this async endpoint waits for it.
+        image = await run_in_threadpool(generate_image, text)
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
     img_buffer = io.BytesIO()
     image.save(img_buffer, format="PNG")
     img_buffer.seek(0)
-    # This call will block until the image is generated.
     return StreamingResponse(img_buffer, media_type="image/png")