Update pest/mcp_output/mcp_plugin/mcp_service.py

pest/mcp_output/mcp_plugin/mcp_service.py

@@ -47,19 +47,32 @@ def preprocess_image_tool(image_path: str) -> dict:
         return {"success": False, "result": None, "error": str(e)}
 
 @mcp.tool(name="detect_pest", description="Detect pests in an image using the loaded model")
-def detect_pest_tool(model: dict, image_data: dict) -> dict:
+def detect_pest_tool(model: object, image_data: object) -> dict:
     """
     Detect pests in an image using the loaded model.
 
     Parameters:
-        model (dict): The loaded pest detection model.
-        image_data (dict): Preprocessed image data.
+        model (object): The loaded pest detection model (dict or JSON string).
+        image_data (object): Preprocessed image data (dict or JSON string).
 
     Returns:
         dict: A dictionary containing success status and detection results.
     """
     try:
-        detection_results = detect_pest(model, image_data)
+        import json
+        
+        # Handle both dict and string inputs
+        if isinstance(model, str):
+            model_dict = json.loads(model)
+        else:
+            model_dict = model
+            
+        if isinstance(image_data, str):
+            image_data_dict = json.loads(image_data)
+        else:
+            image_data_dict = image_data
+            
+        detection_results = detect_pest(model_dict, image_data_dict)
         return {"success": True, "result": detection_results, "error": None}
     except Exception as e:
         return {"success": False, "result": None, "error": str(e)}