Create api.py

api.py

@@ -0,0 +1,46 @@
+import io
+import random
+import os
+import requests
+from fastapi import FastAPI, HTTPException
+from fastapi.responses import StreamingResponse
+from PIL import Image
+
+# API Configuration for Hugging Face
+API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-dev"
+headers = {"Authorization": f"Bearer {os.getenv('HF')}"}
+
+def query(payload):
+    response = requests.post(API_URL, headers=headers, json=payload)
+    if response.status_code != 200:
+        raise HTTPException(status_code=response.status_code, detail="API call failed")
+    return response.content
+
+def generate_image(prompt: str) -> Image.Image:
+    # Generate a random seed so even the same prompt produces a new image.
+    random_seed = random.randint(0, 999999)
+    payload = {
+        "inputs": prompt,
+        "parameters": {"seed": random_seed}
+    }
+    image_bytes = query(payload)
+    return Image.open(io.BytesIO(image_bytes))
+
+app = FastAPI()
+
+@app.get("/")
+def get_generated_image(text: str):
+    """
+    API endpoint that returns the generated image directly.
+    Example usage:
+      https://bowozzz-gen.hf.space/api?text=jokowi
+    """
+    try:
+        image = 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")