Fix No image: postprocess bg-removal, URL->path for Daggr display

app.py

@@ -1,5 +1,6 @@
 import base64
 import tempfile
+import urllib.request
 import gradio as gr
 import spaces
 from gradio_client import Client, handle_file
@@ -8,13 +9,35 @@ from daggr import FnNode, GradioNode, Graph
 # ==================== FREE TOOLS COLLECTION ====================
 # These tools are free to use and don't consume credits
 
+
+def _url_to_path(url):
+    """Download image from URL to a temp file; Daggr needs file paths to display images."""
+    if not url or not isinstance(url, str) or not url.startswith("http"):
+        return url
+    try:
+        ext = "png"
+        if ".jpg" in url or ".jpeg" in url:
+            ext = "jpg"
+        elif ".webp" in url:
+            ext = "webp"
+        f = tempfile.NamedTemporaryFile(suffix=f".{ext}", delete=False)
+        req = urllib.request.Request(url, headers={"User-Agent": "Gradio-Daggr/1.0"})
+        with urllib.request.urlopen(req, timeout=60) as r:
+            f.write(r.read())
+        f.close()
+        return f.name
+    except Exception:
+        return url
+
+
 # 1. Background Removal
-# API returns 1 value. hf-applications/background-removal runs on GPU (Zero)
+# API may return (original, result); we only need the result for "final_image".
 bg_remover = GradioNode(
     "hf-applications/background-removal",
     api_name="/image",
     run_locally=False,
     inputs={"image": gr.Image()},
+    postprocess=lambda *vals: vals[-1] if vals else None,
     outputs={
         "final_image": gr.Image(label="Background Removed"),
     },
@@ -82,16 +105,18 @@ def run_upscaler(image, model_selection="4xBHI_dat2_real"):
     except Exception as e:
         raise RuntimeError(f"Upscaler API error: {e}") from e
     # Returns tuple of 2: [comparison, filepath]; we need the upscaled image path (second element)
+    out = None
     if result and len(result) >= 2 and result[1]:
-        return result[1]
-    if result and len(result) >= 1 and result[0]:
-        # result[0] can be tuple (orig, upscaled) - take second
+        out = result[1]
+    elif result and len(result) >= 1 and result[0]:
         r0 = result[0]
         if isinstance(r0, (list, tuple)) and len(r0) >= 2 and r0[1]:
-            return r0[1]
-        if isinstance(r0, str):
-            return r0
-    return None
+            out = r0[1]
+        elif isinstance(r0, str):
+            out = r0
+    if out and isinstance(out, str) and out.startswith("http"):
+        return _url_to_path(out)
+    return out
 
 
 upscaler = FnNode(
@@ -125,13 +150,17 @@ def run_z_image_turbo(prompt, height=1024, width=1024, seed=42):
         )
     except Exception as e:
         raise RuntimeError(f"Z-Image-Turbo API error: {e}") from e
-    # Returns tuple of 2: [0] dict(path=..., url=..., ...), [1] seed used
+    # Returns tuple of 2: [0] dict(path=..., url=..., ...), [1] seed used. Daggr needs local path.
     if result and len(result) >= 1 and result[0]:
         img = result[0]
+        out = None
         if isinstance(img, dict):
-            return img.get("url") or img.get("path")
-        if isinstance(img, str):
-            return img
+            out = img.get("url") or img.get("path")
+        elif isinstance(img, str):
+            out = img
+        if out and isinstance(out, str) and out.startswith("http"):
+            return _url_to_path(out)
+        return out
     return None
 
 
@@ -177,13 +206,17 @@ def run_flux_klein(
         )
     except Exception as e:
         raise RuntimeError(f"FLUX.2-klein-9B API error: {e}") from e
-    # Returns tuple of 2: [0] dict(path=..., url=...), [1] seed used
+    # Returns tuple of 2: [0] dict(path=..., url=...), [1] seed used. Daggr needs local path.
     if result and len(result) >= 1 and result[0]:
         img = result[0]
+        out = None
         if isinstance(img, dict):
-            return img.get("url") or img.get("path")
-        if isinstance(img, str):
-            return img
+            out = img.get("url") or img.get("path")
+        elif isinstance(img, str):
+            out = img
+        if out and isinstance(out, str) and out.startswith("http"):
+            return _url_to_path(out)
+        return out
     return None