Graceful failure msg

app.py

@@ -49,13 +49,17 @@ def _infer(p, st, sc, h, w, sd, et, sess_state, request: gr.Request):
         "seed": str(sd),
         "eta": float(et),
     }
-    resp = requests.post(BACKEND_URL, json=payload, timeout=120)
-    resp.raise_for_status()
-    out = resp.json()
-    base_img = b64_to_img(out["base_image"])
-    lora_img = b64_to_img(out["lora_image"])
-
-    limiter.post_consume(ip, out.get("duration", 0.0))
+    try:
+        r = requests.post(BACKEND_URL, json=payload, timeout=120)
+        r.raise_for_status()
+        out = r.json()
+        base_img = b64_to_img(out["base_image"])
+        lora_img = b64_to_img(out["lora_image"])
+        limiter.post_consume(ip, out.get("duration", 0.0))
+        return base_img, lora_img, out.get("status", "ok"), sess_state
+    except Exception as e:
+        blank = Image.new("RGB", (int(w), int(h)), (30, 30, 30))
+        return blank, blank, f"Backend error: {e}", sess_state
 
     return base_img, lora_img, out.get("status", "ok"), sess_state