Update app.py

app.py

@@ -6,21 +6,33 @@ import json
 
 def process(json_input):
     try:
-        # json_input from gr.JSON() is STRING, so must parse
         data = json.loads(json_input)
 
+        # decode base64 image
         img_bytes = base64.b64decode(data["image_b64"])
         img = Image.open(io.BytesIO(img_bytes))
 
-        return img  # show image
+        # This goes to Jetson
+        reply = {
+            "received": True,
+            "robot_id": data.get("robot_id"),
+            "size": img.size
+        }
+
+        # return only the msg: (Jetson JSON)
+        return reply
 
     except Exception as e:
-        return None
+        return None, {"error": str(e)}
+
 
 demo = gr.Interface(
     fn=process,
-    inputs=gr.JSON(label="Input JSON from Jetson"),
-    outputs=gr.Image(type="pil", label="Image From Jetson"),
+    inputs=gr.JSON(label="Jetson JSON"),
+    outputs=[
+        gr.Image(type="pil", label="Image Preview"),
+        gr.JSON(label="Reply to Jetson")
+    ],
     api_name="predict"
 )