modified response

main.py

@@ -450,6 +450,12 @@ def predict_image(image_bytes: bytes) -> PredictionResponse:
 
         # Final structured response
         return PredictionResponse(
+            success=True,
+            predicted_class=disease_info['disease_name'] if disease_info.disease_name else "Unknown",
+            predicted_class_index=predicted_class_idx,
+            clean_class_name=clean_name,
+            message="Prediction successful",
+            all_predictions=all_predictions,
             class_id=class_id,
             label=clean_name,
             confidence=round(confidence, 4),