Update app.py

app.py

@@ -4,9 +4,11 @@ from PIL import Image
 import io
 import json
 
-def process(json_input):
+# 修改函式以確保它接收一個字典（這是 gradio_client 預設發送的格式）
+def process(payload):
     try:
-        data = json_input
+        # 如果客戶端已經傳送字典，直接使用 payload:
+        data = payload 
 
         # decode base64 image
         img_bytes = base64.b64decode(data["image_b64"])
@@ -19,19 +21,23 @@ def process(json_input):
             "size": img.size
         }
 
-        # return only the msg: (Jetson JSON)
-        return reply
+        # *** 關鍵修改：回傳一個包含圖片和 JSON 回覆的元組 (tuple) ***
+        # Gradio 會自動將第一個值賦給第一個輸出元件 (gr.Image)
+        # 第二個值賦給第二個輸出元件 (gr.JSON)
+        return img, reply
 
     except Exception as e:
+        # 發生錯誤時，確保回傳兩個值，其中圖片值為 None
         return None, {"error": str(e)}
 
 
 demo = gr.Interface(
     fn=process,
-    inputs=gr.JSON(label="Jetson JSON"),
+    # 我們將輸入定義為 JSON，這允許後端接收字典格式
+    inputs=gr.JSON(label="Input Payload (Dict format)"),
     outputs=[
-        gr.Image(type="pil", label="Image Preview"),
-        gr.JSON(label="Reply to Jetson")
+        gr.Image(type="pil", label="Image Preview"), # 現在將接收 img 物件
+        gr.JSON(label="Reply to Jetson")            # 現在將接收 reply 字典
     ],
     api_name="predict"
 )