Update app.py

app.py

@@ -6,29 +6,25 @@ import json
 
 def process(json_input):
     try:
-        data = json.loads(json_input)
+        data = json_input  # no need to json.loads
 
-        # Decode Base64
         img_bytes = base64.b64decode(data["image_b64"])
         img = Image.open(io.BytesIO(img_bytes))
 
-        # You can do CV/LLM here later
         response = {
             "received": True,
             "robot_id": data.get("robot_id", "unknown"),
             "image_size": img.size
         }
-
-        return json.dumps(response)
+        return response
 
     except Exception as e:
-        return json.dumps({"error": str(e)})
+        return {"error": str(e)}
 
-# Gradio interface
 demo = gr.Interface(
     fn=process,
-    inputs=gr.Textbox(lines=10, label="Input JSON from Jetson"),
-    outputs=gr.Textbox(lines=10, label="Reply to Jetson"),
+    inputs=gr.JSON(label="Input JSON from Jetson"),
+    outputs=gr.JSON(label="Reply to Jetson"),
 )
 
 demo.launch()