Create app.py

app.py

@@ -0,0 +1,48 @@
+import requests
+import base64
+from PIL import Image
+from io import BytesIO
+import gradio as gr
+import uuid
+import os
+
+API_KEY = "sk-k8c1MQmc6OLzrJRs9zJb9ox3VeLmO4Fho7bQo49uLxTX6OTs"
+
+def generate_image(prompt):
+    response = requests.post(
+        "https://api.stability.ai/v1/generation/stable-diffusion-v1-5/text-to-image",
+        headers={
+            "Authorization": f"Bearer {API_KEY}",
+            "Content-Type": "application/json"
+        },
+        json={
+            "text_prompts": [{"text": prompt}],
+            "cfg_scale": 7,
+            "height": 512,
+            "width": 512,
+            "samples": 1,
+            "steps": 30
+        }
+    )
+    
+    if response.status_code != 200:
+        return f"Error: {response.status_code} - {response.text}"
+    
+    data = response.json()
+    b64_image = data["artifacts"][0]["base64"]
+    image_data = base64.b64decode(b64_image)
+    image = Image.open(BytesIO(image_data))
+
+    # Save temporarily
+    filename = f"output_{uuid.uuid4().hex[:8]}.png"
+    filepath = f"/tmp/{filename}"  # Hugging Face allows writing to /tmp
+    image.save(filepath)
+
+    return filepath  # return file path, not just image
+
+gr.Interface(
+    fn=generate_image,
+    inputs="text",
+    outputs=gr.Image(type="filepath", label="Generated Image"),  # <- Important!
+    title="Speech2Image Generator (Fast API Powered)",
+).launch()