Update app.py

app.py

@@ -74,10 +74,9 @@ def safe_parse_json_from_text(text: str):
 
 
 # -------------------------------
-# MCP TOOL
+# TRUE CORE FUNCTION
 # -------------------------------
-@mcp.tool()
-def robot_watch(payload: Dict[str, Any]):
+def robot_watch_core(payload: Dict[str, Any]):
     if isinstance(payload, str):
         try:
             payload = json.loads(payload)
@@ -93,12 +92,12 @@ def robot_watch(payload: Dict[str, Any]):
     if not image_b64:
         return {"error": "image_b64 missing"}
 
-    # Upload image
+    # Upload
     _, hf_url, _, size_bytes = upload_image(image_b64, hf_token)
     if not hf_url:
         return {"error": "Image upload failed"}
 
-    # VLM call
+    # VLM
     system_prompt = """
 Respond in STRICT JSON ONLY.
 {
@@ -151,28 +150,29 @@ Respond in STRICT JSON ONLY.
 
 
 # -------------------------------
-# GRADIO WRAPPER
+# REGISTER MCP TOOL (wrapper)
+# -------------------------------
+@mcp.tool()
+def robot_watch(payload: Dict[str, Any]):
+    return robot_watch_core(payload)
+
+
+# -------------------------------
+# Gradio wrapper
 # -------------------------------
 def process_json(payload):
-    """
-    Gradio wrapper: directly calls the MCP tool inside the same process.
-    """
-    return robot_watch(payload)
+    return robot_watch_core(payload)
 
 
 app = gr.Interface(
     fn=process_json,
-    inputs=gr.JSON(label="Input JSON (hf_token, image_b64, robot_id)"),
+    inputs=gr.JSON(label="Input JSON"),
     outputs=gr.JSON(label="Result JSON"),
     title="Robot MCP Server",
-    description="This is only a JSON wrapper. The real logic lives in the MCP tool.",
+    description="JSON endpoint for robot vision pipeline.",
     api_name="predict"
 )
 
 
-# -------------------------------
-# ENTRY
-# -------------------------------
 if __name__ == "__main__":
-    # Launch MCP server + Gradio together
     app.launch(mcp_server=True)