Update app.py

app.py

@@ -1,25 +1,35 @@
-from flask import Flask, Response, request
+from flask import Flask, Response, request, send_file
 import os, urllib.parse, requests
 
 app = Flask(__name__)
 
-# Get URL template from env (set in HF secrets)
+# Pollinations URL template from env
 URL_TEMPLATE = os.environ.get("FLUX")
 if not URL_TEMPLATE:
     raise RuntimeError("Missing FLUX env variable")
 
 @app.route("/", methods=["GET"])
-def generate_image():
-    # Default parameters
-    prompt = request.args.get("prompt", "a plain simple glowing board with the word 'OptimFLUX'")
-    width = request.args.get("width", "1024")
-    height = request.args.get("height", "1024")
-    seed = request.args.get("seed", "0")
+def generate_or_default():
+    # If no query string present, serve default.png
+    if not request.query_string:
+        # Make sure default.png is in the working dir (e.g. next to app.py)
+        return send_file(
+            "default.png",
+            mimetype="image/png",
+            as_attachment=False,
+            download_name="default.png"
+        )
 
-    # Sanitize prompt
-    encoded_prompt = urllib.parse.quote(prompt)
+    # Otherwise, generate via Pollinations
+    prompt = request.args.get(
+        "prompt",
+        "a glowing board with the word 'KindSynapse'"
+    )
+    width  = request.args.get("width", "1024")
+    height = request.args.get("height", "1024")
+    seed   = request.args.get("seed", "0")
 
-    # Fill in URL
+    encoded_prompt = urllib.parse.quote(prompt, safe="")
     url = (
         URL_TEMPLATE
         .replace("[prompt]", encoded_prompt)
@@ -28,10 +38,15 @@ def generate_image():
         .replace("[seed]", seed)
     )
 
-    # Request image
-    response = requests.get(url, stream=True)
-    if response.status_code != 200:
-        return Response(f"Failed to fetch image. Upstream status: {response.status_code}", status=502)
+    resp = requests.get(url, stream=True)
+    if resp.status_code != 200:
+        return Response(
+            f"Failed to fetch image. Upstream status: {resp.status_code}",
+            status=502
+        )
+
+    return Response(resp.content, content_type=resp.headers.get("Content-Type"))
 
-    # Serve image
-    return Response(response.content, content_type=response.headers.get("Content-Type"))
+if __name__ == "__main__":
+    # Dev server
+    app.run(host="0.0.0.0", port=7860)