Update app.py

app.py

@@ -33,6 +33,19 @@ def predict_disease(image_file, model, all_labels, target_language):
 
         # Predict the class
         predictions = model.predict(img_array)
+        confidence_threshold = 0.98  # Require at least 98% confidence
+        confidence_scores = predictions[0]
+        max_confidence = np.max(confidence_scores)
+
+        if max_confidence < confidence_threshold:
+            print(f"Prediction confidence ({max_confidence:.2f}) is too low.")
+            return f"""
+    <h3 style="color:red; text-align:center;">
+        The model's confidence is below 98% ({max_confidence*100:.2f}%). 
+        Please upload a clearer image of the plant.
+    </h3>
+    """
+
         predicted_class = np.argmax(predictions[0])
         
         # Get the predicted class label