Update app.py

app.py

@@ -9,7 +9,6 @@ import urllib.parse
 import asyncio
 import aiohttp
 import threading
-import PIL
 
 app = FastAPI()
 
@@ -421,6 +420,8 @@ async def ask_website(url: str, question: str, model: str = "llama-3-70b", proxy
 from huggingface_hub import InferenceClient
 client_sd3 = InferenceClient("stabilityai/stable-diffusion-3-medium-diffusers")
 
+from PIL import Image
+
 @app.get("/api/sd3")
 def sd3(prompt :str = "",
         steps: int = "20",
@@ -431,9 +432,10 @@ def sd3(prompt :str = "",
         image = client_sd3.text_to_image(prompt = f"{prompt}, hd, high quality, 4k, masterpiece",
                 num_inference_steps = steps, 
                 width = width, height = height )
-        return image.save(f"{prompt}.jpg")
+        image = Image.open(io.BytesIO(image))
+        return image
     except Exception as e:
-        raise HTTPException(detail=f"Error during image generation: {e}")        
+        raise HTTPException(status_code=500, detail=f"Error during image generation: {e}")        
         
 @app.get("/api/maps")
 async def maps(