Update app.py

app.py

@@ -10,6 +10,7 @@ import asyncio
 import aiohttp
 import threading
 
+
 app = FastAPI()
 
 @app.get("/")
@@ -416,6 +417,24 @@ async def ask_website(url: str, question: str, model: str = "llama-3-70b", proxy
         raise HTTPException(status_code=500, detail=f"Error fetching or processing URL: {e}")
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Error during question answering: {e}")
+
+from huggingface_hub import InferenceClient
+client_sd3 = InferenceClient("stabilityai/stable-diffusion-3-medium-diffusers")
+
+@app.get("/api/sd3")
+def sd3(prompt :str = "",
+        steps: int = "20",
+        width: int = 1000,
+        height: int = 1000
+       ):
+    try:
+        seed = random.randint(0, 9999999)
+        image = client_sd3.text_to_image(prompt = f"{prompt} , {seed}, hd, high quality, 4k, masterpiece",
+                num_inference_steps = steps, 
+                width = width, height = height )
+        return image
+    except Exception as e:
+        raise HTTPException(detail=f"Error during image generation: {e}")        
         
 @app.get("/api/maps")
 async def maps(