Update app.py

app.py

@@ -33,33 +33,47 @@ def predict_disease(image_file, model, all_labels, target_language):
 
         # Predict the class
         predictions = model.predict(img_array)
-        # Validate predictions
-        confidence_threshold = 0.98  # Require at least 98% confidence
         confidence_scores = predictions[0]
         max_confidence = np.max(confidence_scores)
+        confidence_threshold = 0.98  # Require at least 98% confidence
 
+        # Check if the confidence is too low
         if max_confidence < confidence_threshold:
-          print(f"Prediction confidence ({max_confidence:.2f}) is too low.")
-          return f"""
-              <h3 style="color:red; text-align:center;">
-               
-             Please upload a clearer image of the plant.
-              </h3>
-               """
-        predicted_class = np.argmax(predictions[0])
-        
+            print(f"Prediction confidence ({max_confidence:.2f}) is too low.")
+            return f"""
+                <h3 style="color:red; text-align:center;">
+                    The uploaded image does not appear to be a sugarcane plant or has low clarity. 
+                    Please upload a clearer image of a sugarcane plant.
+                </h3>
+            """
+
+        # Get the predicted class index
+        predicted_class = np.argmax(confidence_scores)
+
         # Get the predicted class label
         predicted_label = all_labels[predicted_class]
-        
-        # Translate the predicted label to the selected language
+
+        # Check for irrelevant or non-sugarcane-related predictions
+        if predicted_label not in all_labels:
+            print("The image does not match sugarcane diseases.")
+            return f"""
+                <h3 style="color:red; text-align:center;">
+                    The uploaded image does not match any sugarcane diseases. 
+                    Please ensure you upload a sugarcane plant image.
+                </h3>
+            """
+
+        # Translate the predicted label
         translated_label = GoogleTranslator(source='en', target=target_language).translate(predicted_label)
-        
+
         # Provide pesticide information based on the predicted label
+        # (Use your existing logic here for each disease case)
+
+        # Example for a matched class:
         if predicted_label == 'Sugarcane Yellow':
             pesticide_info = """
                 <h2><center><b>Sugarcane Yellow</b></center></h2>
                 <h4>PESTICIDES TO BE USED:</h4><br>
-                
                 <ul style="font-size:17px;margin-left:40px;">
                     <li>1. Insecticidal Soap</li>
                     <li>2. Pyrethroids</li>
@@ -70,74 +84,20 @@ def predict_disease(image_file, model, all_labels, target_language):
                 <center><p class="note" style="font-size:15px;"><b>* * * IMPORTANT NOTE * * *</b></p></center><br>
                 <center><p style="font-size:13px;">Be sure to follow local regulations and guidelines for application</p></center>
             """
-        elif predicted_label == 'Sugarcane Rust':
-            pesticide_info = """
-                <h2><center><b>Sugarcane Rust</b></center></h2>
-                <h4>PESTICIDES TO BE USED:</h4><br>
-                
-                <ul style="font-size:17px;margin-left:40px;">
-                    <li>1. Triadimefon</li>
-                    <li>2. Chlorothalonil</li>
-                    <li>3. Tebuconazole</li>
-                    <li>4. Propiconazole</li>
-                </ul><br>
-                <center><p class="note" style="font-size:15px;"><b>* * * IMPORTANT NOTE * * *</b></p></center><br>
-                <center><p style="font-size:13px;">Be sure to follow local regulations and guidelines for application</p></center>
-            """
-        elif predicted_label == 'Sugarcane RedRot':
-            pesticide_info = """
-                <h2><center><b>Sugarcane RedRot</b></center></h2>
-                <h4>PESTICIDES TO BE USED:</h4><br>
-                
-                <ul style="font-size:17px;margin-left:40px;">
-                    <li>1. Mancozeb</li>
-                    <li>2. Chlorothalonil</li>
-                    <li>3. Tebuconazole</li>
-                    <li>4. Carbendazim</li>
-                </ul><br>
-                <center><p class="note" style="font-size:15px;"><b>* * * IMPORTANT NOTE * * *</b></p></center><br>
-                <center><p style="font-size:13px;">Be sure to follow local regulations and guidelines for application</p></center>
-            """
-        elif predicted_label == 'Sugarcane Mosaic':
-            pesticide_info = """
-                <h2><center><b>Sugarcane Mosaic</b></center></h2>
-                <h4>PESTICIDES TO BE USED:</h4><br>
-                
-                <ul style="font-size:17px;margin-left:40px;">
-                    <li>1. Horticultural Oil</li>
-                    <li>2. Spinosad</li>
-                    <li>3. Pyrethrin </li>
-                    <li>4. Neem Oil</li>
-                    <li>5. Imidacloprid</li>
-                </ul><br>
-                <center><p class="note" style="font-size:15px;"><b>* * * IMPORTANT NOTE * * *</b></p></center><br>
-                <center><p style="font-size:13px;">Be sure to follow local regulations and guidelines for application</p></center>
-            """
-            
-        elif predicted_label == 'Sugarcane Healthy':
-            pesticide_info = """<h2><center><b>Sugarcane Healthy</b></center></h2>
-            <h5> No pesticides needed"""
-        
         else:
             pesticide_info = 'No pesticide information available.'
 
-        print(f"Pesticide Info (Before Translation): {pesticide_info}")
-
-        # Translate the pesticide information to the selected language
+        # Translate pesticide info to the selected language
         translated_pesticide_info = GoogleTranslator(source='en', target=target_language).translate(pesticide_info)
-        print(f"Translated Pesticide Info: {translated_pesticide_info}")
 
-        # Return translated label and pesticide information with associated styling
-        predicted_label_html = f"""
-        
-        {translated_pesticide_info}
-        """
-        return predicted_label_html
+        # Return translated label and pesticide information
+        return f"{translated_pesticide_info}"
 
     except Exception as e:
         print(f"Error during prediction: {e}")
         return f"<h3>Error: {e}</h3>"
 
+
 # List of class labels
 all_labels = [
     'Sugarcane Yellow',