Create app.py

app.py

@@ -1,35 +1,44 @@
 import os
 import requests
-import gradio as gr
+from flask import Flask, request, send_file, jsonify, render_template
 from io import BytesIO
-from PIL import Image
+
+app = Flask(__name__, static_folder=".", template_folder=".")
 
 HF_API_KEY = os.environ.get("HF_API_KEY")
 if not HF_API_KEY:
     raise ValueError("HF_API_KEY not set in environment variables")
 
-def generate_image(prompt, model="stabilityai/stable-diffusion-xl-base-1.0"):
+@app.route("/")
+def index():
+    return render_template("index.html")
+
+@app.route("/generate", methods=["POST"])
+def generate():
+    data = request.json
+    prompt = data.get("prompt", "")
+    model = data.get("model", "stabilityai/stable-diffusion-xl-base-1.0")
+
     if not prompt.strip():
-        return None
+        return jsonify({"error": "Prompt cannot be empty"}), 400
+
     headers = {"Authorization": f"Bearer {HF_API_KEY}"}
     payload = {"inputs": prompt}
-    response = requests.post(
-        f"https://api-inference.huggingface.co/models/{model}",
-        headers=headers,
-        json=payload,
-        timeout=120
-    )
+
+    try:
+        response = requests.post(
+            f"https://api-inference.huggingface.co/models/{model}",
+            headers=headers,
+            json=payload,
+            timeout=120
+        )
+    except requests.exceptions.RequestException as e:
+        return jsonify({"error": str(e)}), 500
+
     if response.status_code != 200:
-        raise ValueError(response.text)
-    # Convert bytes to PIL Image
-    return Image.open(BytesIO(response.content))
-
-iface = gr.Interface(
-    fn=generate_image,
-    inputs=[gr.Textbox(label="Prompt", placeholder="Enter your prompt here")],
-    outputs=[gr.Image(type="pil")],
-    title="imgGenPro",
-    description="Generate AI images using Hugging Face Inference API"
-)
-
-iface.launch()
+        return jsonify({"error": response.text}), response.status_code
+
+    return send_file(BytesIO(response.content), mimetype="image/png")
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=7860)