Update app.py

app.py

@@ -1,9 +1,29 @@
-from flask import Flask
-app = Flask(__name__)
+from flask import Flask, request, jsonify, send_from_directory
+from gradio_client import Client
+import base64, os, uuid
+
+app = Flask(__name__, static_folder="static")
+client = Client("mrfakename/Z-Image-Turbo")
 
 @app.route("/")
-def hello():
-    return "<h1>🟢 Docker Space is alive</h1><p>Hello from HedronCreeper!</p>"
+def index():
+    return send_from_directory("static", "index.html")
+
+@app.route("/generate", methods=["POST"])
+def generate():
+    data = request.json
+    prompt = data.get("prompt", "a beautiful landscape")
+    try:
+        result = client.predict(
+            prompt=prompt,
+            api_name="/generate"
+        )
+        # result is a file path — read and base64 encode it
+        with open(result, "rb") as f:
+            img_b64 = base64.b64encode(f.read()).decode()
+        return jsonify({"image": img_b64})
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
 
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)