Prefill reference values via node data, wire condition image, handle canvas file URLs in edit_image

app.py

@@ -10,7 +10,10 @@ PyTorch runs in CUDA emulation mode, and a real ZeroGPU is attached only while
 a decorated function executes.
 """
 
+import base64
 import os
+import urllib.parse
+import urllib.request
 
 import gradio as gr
 from gradio_client import utils as client_utils
@@ -39,14 +42,24 @@ pipe.to("cuda")
 
 def _to_pil(image) -> Image.Image:
     """Accept whatever the canvas hands a bound function for an image port:
-    a PIL image, a file path, or Gradio's {path, url, is_file} file dict."""
+    a PIL image, a local path, a /gradio_api/file= reference, an http(s) or
+    data: URL, or a file dict carrying any of those. Mirrors _file_ref in
+    gradio.workflow — canvas file values carry only `url`, no `path`."""
+    import io
+
     if isinstance(image, Image.Image):
         return image
     if isinstance(image, dict):
-        image = image.get("path") or image.get("url")
-    if isinstance(image, str):
-        return Image.open(image)
-    raise ValueError(f"Unsupported image input: {type(image)!r}")
+        image = image.get("path") or image.get("url") or ""
+    if not isinstance(image, str) or not image:
+        raise ValueError(f"Unsupported image input: {type(image)!r}")
+    if image.startswith("/gradio_api/file="):
+        image = urllib.parse.unquote(image.removeprefix("/gradio_api/file="))
+    if image.startswith("data:"):
+        return Image.open(io.BytesIO(base64.b64decode(image.split(",", 1)[1])))
+    if image.startswith(("http://", "https://")):
+        return Image.open(io.BytesIO(urllib.request.urlopen(image).read()))
+    return Image.open(image)
 
 
 def _save(image: Image.Image) -> dict:

workflow.json

@@ -23,7 +23,10 @@
           "type": "text",
           "default_value": "A capybara wearing a wizard hat, oil painting"
         }
-      ]
+      ],
+      "data": {
+        "out": "A capybara wearing a wizard hat, oil painting"
+      }
     },
     {
       "id": "ref_image",
@@ -45,7 +48,12 @@
           "label": "Image",
           "type": "image"
         }
-      ]
+      ],
+      "data": {
+        "out": {
+          "url": "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/input_image_vermeer.png"
+        }
+      }
     },
     {
       "id": "ref_instruction",
@@ -68,7 +76,10 @@
           "type": "text",
           "default_value": "Move it to a snowy mountain top"
         }
-      ]
+      ],
+      "data": {
+        "out": "Move it to a snowy mountain top"
+      }
     }
   ],
   "operators": [
@@ -101,7 +112,10 @@
           "type": "image",
           "output_index": 0
         }
-      ]
+      ],
+      "data": {
+        "steps": 40
+      }
     },
     {
       "id": "op_edit",
@@ -138,7 +152,10 @@
           "type": "image",
           "output_index": 0
         }
-      ]
+      ],
+      "data": {
+        "steps": 40
+      }
     }
   ],
   "subjects": [
@@ -221,8 +238,8 @@
       "type": "image"
     },
     {
-      "id": "e_chain",
-      "from_node_id": "sub_generated",
+      "id": "e3",
+      "from_node_id": "ref_image",
       "from_port_id": "out",
       "to_node_id": "op_edit",
       "to_port_id": "image",