Update app.py

app.py

@@ -9,22 +9,23 @@ import gradio as gr
 HORDE_URL = "https://stablehorde.net/api/v2/generate/async"
 HORDE_STATUS = "https://stablehorde.net/api/v2/generate/status/{id}"
 
-# HF Space secret recommended for priority, but optional
+# HF Space secret recommended for priority
 HORDE_API_KEY = os.getenv("HORDE_API_KEY", "")
 CLIENT_AGENT = "StitchMaster/0.1 (https://huggingface.co/spaces/your-space)"
 
 DEFAULT_STEPS = 24
-DEFAULT_W = 768
-DEFAULT_H = 768
+DEFAULT_W = 704   # keep defaults under the 715px threshold
+DEFAULT_H = 704
 POLL_INTERVAL = 2.5
 POLL_TIMEOUT = 240            # bump if queues are long
 MODEL = None                  # or set e.g. "SDXL 1.0"
 
 def _headers():
-    h = {"Client-Agent": CLIENT_AGENT}
-    if HORDE_API_KEY:
-        h["apikey"] = HORDE_API_KEY
-    return h
+    # Always send an apikey; fallback to anonymous for testing
+    return {
+        "Client-Agent": CLIENT_AGENT,
+        "apikey": HORDE_API_KEY if HORDE_API_KEY else "0000000000"
+    }
 
 # =========================
 # Horde client with debugging
@@ -53,32 +54,47 @@ def horde_txt2img(prompt: str,
     if model:
         payload["models"] = [model]
 
-    # Submit
+    # -------- Submit (with KudosUpfront fallback) --------
     try:
-        submit = requests.post(
-            HORDE_URL, json=payload, headers=_headers(), timeout=30
-        )
+        submit = requests.post(HORDE_URL, json=payload, headers=_headers(), timeout=30)
+
+        # Auto-fallback if KudosUpfront required
+        if submit.status_code == 403:
+            try:
+                body = submit.json()
+            except Exception:
+                body = {"message": submit.text}
+            msg = (body.get("message") or "").lower()
+            rc = body.get("rc") or ""
+            if "kudos" in msg or rc == "KudosUpfront":
+                # Clamp params to stay under upfront limits
+                payload["params"]["steps"]  = min(int(payload["params"]["steps"]), 30)
+                payload["params"]["width"]  = min(int(payload["params"]["width"]), 704)
+                payload["params"]["height"] = min(int(payload["params"]["height"]), 704)
+                dbg.append("Fallback applied: steps<=30, width/height<=704. Retrying submit...")
+                submit = requests.post(HORDE_URL, json=payload, headers=_headers(), timeout=30)
+
         dbg.append(f"SUBMIT status={submit.status_code}")
         if submit.status_code >= 300:
             dbg.append(f"SUBMIT body={submit.text[:500]}")
             submit.raise_for_status()
+
         submit_j = submit.json()
         job_id = submit_j.get("id")
         if not job_id:
             dbg.append(f"SUBMIT json={submit_j}")
             raise gr.Error("Horde submit succeeded but no job id returned.")
         dbg.append(f"JOB id={job_id}")
-    except Exception as e:
+
+    except Exception:
         dbg.append("SUBMIT exception:\n" + traceback.format_exc())
         return None, "\n".join(dbg)
 
-    # Poll
+    # -------- Poll --------
     start = time.time()
     while True:
         try:
-            status_r = requests.get(
-                HORDE_STATUS.format(id=job_id), headers=_headers(), timeout=30
-            )
+            status_r = requests.get(HORDE_STATUS.format(id=job_id), headers=_headers(), timeout=30)
             if status_r.status_code >= 300:
                 dbg.append(f"POLL status={status_r.status_code}")
                 dbg.append(f"POLL body={status_r.text[:500]}")
@@ -123,14 +139,13 @@ def horde_txt2img(prompt: str,
             return None, "\n".join(dbg)
 
 def generate_image(prompt, steps, size):
-    # size like "768x768"
+    # size like "704x704"
     try:
         w, h = [int(x.strip()) for x in size.lower().split("x")]
     except Exception:
         w, h = DEFAULT_W, DEFAULT_H
     img, debug = horde_txt2img(prompt, steps=steps, width=w, height=h)
     if img is None:
-        # Show a friendly error plus the debug log
         gr.Warning("Generation failed. See debug log for details.")
     return img, debug
 
@@ -162,22 +177,22 @@ CUSTOM_CSS = """
 }
 """
 
-with gr.Blocks(css= CUSTOM_CSS, title="Image Checkpoints – Stable Horde") as demo:
+with gr.Blocks(css=CUSTOM_CSS, title="Image Checkpoints – Stable Horde") as demo:
     gr.Markdown("### Image Checkpoints (Stable Horde)\nGenerate per-prompt frames. If a run fails, check the debug panel.")
 
     with gr.Row():
         steps = gr.Slider(8, 50, value=DEFAULT_STEPS, step=1, label="Steps (quality/time)")
         size = gr.Dropdown(
-            choices=["512x512", "768x768", "1024x576", "1024x768"],
+            choices=["512x512", "704x704", "704x512", "640x640"],
             value=f"{DEFAULT_W}x{DEFAULT_H}",
             label="Resolution"
         )
 
-    # One shared debug panel for simplicity
+    # Shared debug panel
     debug_box = gr.Code(label="Debug log", interactive=False)
 
     prompt_boxes, gen_buttons, img_outputs = [], [], []
-    for i in range(1, 5):
+    for i in range(1, 4 + 1):
         with gr.Row():
             with gr.Column(scale=1, min_width=320):
                 p = gr.Textbox(