Update app.py

app.py

@@ -554,6 +554,13 @@ def create_app():
                 "Image loaded — hover the viewer and click an arrow to generate",
             )
 
+        def _img_from_data_url(data_url):
+            """Decode a base64 data URL back to a PIL Image."""
+            if not data_url or not data_url.startswith("data:"):
+                return None
+            header, b64 = data_url.split(",", 1)
+            return Image.open(BytesIO(base64.b64decode(b64)))
+
         def on_camera_submit(
             current_view, src_img,
             seed_val, rand_seed, guidance, steps, h, w,
@@ -563,6 +570,10 @@ def create_app():
                 az, el, dist = _coerce_view(current_view)
                 prompt = build_camera_prompt(az, el, dist)
 
+                # If no image in gr.Image, try the data URL from the viewer dropzone
+                if src_img is None and isinstance(current_view, dict):
+                    src_img = _img_from_data_url(current_view.get("img", ""))
+
                 if src_img is None:
                     return current_view, prompt, "⚠  Upload an image first", gallery_imgs, gallery_imgs