Update app.py

app.py

@@ -8,29 +8,19 @@ def process(json_input):
     try:
         data = json_input
 
-        # Decode base64 → PIL Image
+        # 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
+        return img  # <-- 直接顯示圖片
 
     except Exception as e:
-        return {"error": str(e)}, None
+        return None
 
 demo = gr.Interface(
     fn=process,
     inputs=gr.JSON(label="Input JSON from Jetson"),
-    outputs=[
-        gr.JSON(label="Reply Message"),
-        gr.Image(type="pil", label="Received Image")  # <-- display image
-    ],
+    outputs=gr.Image(type="pil", label="Image From Jetson"),
     api_name="predict"
 )