fix: extract prediction label for accurate weight management in consensus calculation

app_mcp.py

@@ -326,11 +326,12 @@ def predict_image_with_json(img, confidence_threshold, augment_methods, rotate_d
     
     for model_id, prediction in model_predictions_raw.items(): # Use raw predictions for weighting
         # Ensure the prediction label is valid for weighted_predictions
-        if prediction in weighted_predictions:
-            weighted_predictions[prediction] += adjusted_weights[model_id]
+        prediction_label = prediction.get("Label") # Extract the label
+        if prediction_label in weighted_predictions:
+            weighted_predictions[prediction_label] += adjusted_weights[model_id]
         else:
             # Handle cases where prediction might be an error or unexpected label
-            logger.warning(f"Unexpected prediction label '{prediction}' from model '{model_id}'. Skipping its weight in consensus.")
+            logger.warning(f"Unexpected prediction label '{prediction_label}' from model '{model_id}'. Skipping its weight in consensus.")
 
     final_prediction_label = "UNCERTAIN"
     if weighted_predictions["AI"] > weighted_predictions["REAL"] and weighted_predictions["AI"] > weighted_predictions["UNCERTAIN"]: