Update app.py

app.py

@@ -5,21 +5,33 @@ import io
 import json
 
 def process(json_input):
-    data = json_input
-    img_bytes = base64.b64decode(data["image_b64"])
-    img = Image.open(io.BytesIO(img_bytes))
+    try:
+        data = json_input
 
-    return {
-        "received": True,
-        "robot_id": data.get("robot_id", "unknown"),
-        "image_size": img.size
-    }
+        # Decode base64 → PIL Image
+        img_bytes = base64.b64decode(data["image_b64"])
+        img = Image.open(io.BytesIO(img_bytes))
+
+        # Response JSON
+        response = {
+            "received": True,
+            "robot_id": data.get("robot_id", "unknown"),
+            "image_size": img.size
+        }
+
+        return response, img  # <-- return JSON + image
+
+    except Exception as e:
+        return {"error": str(e)}, None
 
 demo = gr.Interface(
     fn=process,
     inputs=gr.JSON(label="Input JSON from Jetson"),
-    outputs=gr.JSON(label="Reply to Jetson"),
-    api_name="predict"    # very important
+    outputs=[
+        gr.JSON(label="Reply Message"),
+        gr.Image(type="pil", label="Received Image")  # <-- display image
+    ],
+    api_name="predict"
 )
 
 demo.launch()