app.py

import gradio as gr

# Import tensorflow here
import tensorflow as tf 

from tensorflow.keras.utils import img_to_array,load_img
from tensorflow.keras.models import load_model # Use tensorflow.keras.models
import numpy as np

# Load the pre-trained model from the local path
model_path = 'Mango.h5'

# Define custom objects to handle potential incompatibilities
custom_objects = {'DepthwiseConv2D': tf.keras.layers.DepthwiseConv2D} 

# Load the model with custom_objects
model = load_model(model_path, custom_objects=custom_objects)  # Load the model here

def predict_disease(image_file, model, all_labels):
    
    try:
        # Load and preprocess the image
        img = load_img(image_file, target_size=(224, 224))  # Use load_img from tensorflow.keras.utils
        img_array = img_to_array(img)
        img_array = np.expand_dims(img_array, axis=0)  # Add batch dimension
        img_array = img_array / 255.0  # Normalize the image

        # Predict the class
        predictions = model.predict(img_array)  # Use the loaded model here
        predicted_class = np.argmax(predictions[0])

        # Get the predicted class label
        predicted_label = all_labels[predicted_class]

        # Print the predicted label to the console
        
        if predicted_label=='Mango Anthracrose':
            predicted_label = """<style>
                      li{
                        font-size: 15px;
                        margin-left: 90px;
                        margin-top: 15px;
                        margin-bottom: 15px;
                      }
                      h4{
                        font-size: 17px;
                        margin-top: 15px;
                      }
                      h4:hover{
                        cursor: pointer;
                      }

                      h3:hover{
                        cursor: pointer;
                        color: blue;
                        transform: scale(1.3);
                      }
                      .note{
                        text-align: center;
                        font-size: 16px;
                      }
                      p{
                        font-size: 13px;
                        text-align: center;
                      }

                </style>
                <h3><center><b>Mango Anthracrose</b></center></h3>
                <h4>PESTICIDES TO BE USED:</h4>
                <ul>
                    <li>1. Mancozeb</li>
                    <li>2. Azoxystrobin</li>
                    <li>3. carbendazim</li>
                    <li>4. Propiconazole</li>
                    <li>5. Thiophanate-methyl</li>
                    <li>6. Copper Sulfate</li>
                </ul><br>
                <p class="note"><b>* * * IMPORTANT NOTE * * *</b></p>
                <p>Be sure to follow local regulations and guidelines for application</p>

            """
        elif predicted_label=='Mango Bacterial Canker':
            predicted_label = """
            <style>
                      li{
                        font-size: 15px;
                        margin-left: 90px;
                        margin-top: 15px;
                        margin-bottom: 15px;
                      }
                      h4{
                        font-size: 17px;
                        margin-top: 15px;
                      }
                      h4:hover{
                        cursor: pointer;
                      }

                      h3:hover{
                        cursor: pointer;
                        color: blue;
                        transform: scale(1.3);
                      }
                      .note{
                        text-align: center;
                        font-size: 16px;
                      }
                      p{
                        font-size: 13px;
                        text-align: center;
                      }

                </style>
                <h3><center><b>Mango Bacterial Canker</b></center></h3>
                <h4>PESTICIDES TO BE USED:</h4>
                <ul>
                    <li>1. Copper Hydroxide</li>
                    <li>2. Copper Oxychloride</li>
                    <li>3. Streptomycin</li>
                    <li>4. oxytetracycline</li>
                    <li>5. Neem oil</li>
                    <li>6. Garlic oil</li>
                </ul>
                <p class="note"><b>* * * IMPORTANT NOTE * * *</b></p>
                <p>Be sure to follow local regulations and guidelines for application</p>


            """
        elif predicted_label=='Mango Cutting Weevil':
            predicted_label = """
            <style>
                      li{
                        font-size: 15px;
                        margin-left: 90px;
                        margin-top: 15px;
                        margin-bottom: 15px;
                      }
                      h4{
                        font-size: 17px;
                        margin-top: 15px;
                      }
                      h4:hover{
                        cursor: pointer;
                      }

                      h3:hover{
                        cursor: pointer;
                        color: blue;
                        transform: scale(1.3);
                      }
                      .note{
                        text-align: center;
                        font-size: 16px;
                      }
                      p{
                        font-size: 13px;
                        text-align: center;
                      }

                </style>
                <h3><center><b>Mango Cutting Weevil</b></center></h3>
                <h4>PESTICIDES TO BE USED:</h4>
                <ul>
                    <li>1. Imidacloprid</li>
                    <li>2. Thiamethoxam</li>
                    <li>3. Chlorpyrifos</li>
                    <li>4. Lambda-cyhalothrin</li>
                    <li>5. Fipronil</li>
                    <li>6. Neem oil</li>
                </ul>
                <p class="note"><b>* * * IMPORTANT NOTE * * *</b></p>
                <p>Be sure to follow local regulations and guidelines for application</p>


            """

        elif predicted_label=='Mango Die Back':
            predicted_label = """
            <style>
                      li{
                        font-size: 15px;
                        margin-left: 90px;
                        margin-top: 15px;
                        margin-bottom: 15px;
                      }
                      h4{
                        font-size: 17px;
                        margin-top: 15px;
                      }
                      h4:hover{
                        cursor: pointer;
                      }

                      h3:hover{
                        cursor: pointer;
                        color: blue;
                        transform: scale(1.3);
                      }
                      .note{
                        text-align: center;
                        font-size: 16px;
                      }
                      p{
                        font-size: 13px;
                        text-align: center;
                      }

                </style>
                <h3><center><b>Mango Die Back</b></center></h3>
                <h4>PESTICIDES TO BE USED:</h4>
                <ul>
                    <li>1. Carbendazim</li>
                    <li>2. Mancozeb</li>
                    <li>3. Azoxystrobin</li>
                    <li>4. Triazole</li>
                    <li>5. Potassium bicarbonate</li>
                    <li>6. Sodium bicarbonate</li>
                </ul>
                <p class="note"><b>* * * IMPORTANT NOTE * * *</b></p>
                <p>Be sure to follow local regulations and guidelines for application</p>


            """
        elif predicted_label=='Mango Gall Midge':
            predicted_label = """
            <style>
                      li{
                        font-size: 15px;
                        margin-left: 90px;
                        margin-top: 15px;
                        margin-bottom: 15px;
                      }
                      h4{
                        font-size: 17px;
                        margin-top: 15px;
                      }
                      h4:hover{
                        cursor: pointer;
                      }

                      h3:hover{
                        cursor: pointer;
                        color: blue;
                        transform: scale(1.3);
                      }
                      .note{
                        text-align: center;
                        font-size: 16px;
                      }
                      p{
                        font-size: 13px;
                        text-align: center;
                      }

                </style>
                <h3><center><b>Mango Gall Midge</b></center></h3>
                <h4>PESTICIDES TO BE USED:</h4>
                <ul>
                    <li>1. Imidacloprid</li>
                    <li>2. Thiamethoxam</li>
                    <li>3. Chlorpyrifos</li>
                    <li>4. Lambda-cyhalothrin</li>
                    <li>5. Spinosad</li>
                    <li>6. Pyrethrin</li>
                </ul>
                <p class="note"><b>* * * IMPORTANT NOTE * * *</b></p>
                <p>Be sure to follow local regulations and guidelines for application</p>


            """
        elif predicted_label=='Mango Powdery Mildew':
            predicted_label = """
            <style>
                      li{
                        font-size: 15px;
                        margin-left: 90px;
                        margin-top: 15px;
                        margin-bottom: 15px;
                      }
                      h4{
                        font-size: 17px;
                        margin-top: 15px;
                      }
                      h4:hover{
                        cursor: pointer;
                      }

                      h3:hover{
                        cursor: pointer;
                        color: blue;
                        transform: scale(1.3);
                      }
                      .note{
                        text-align: center;
                        font-size: 16px;
                      }
                      p{
                        font-size: 13px;
                        text-align: center;
                      }

                </style>
                <h3><center><b>Mango Powdery Mildew</b></center></h3>
                <h4>PESTICIDES TO BE USED:</h4>
                <ul>
                    <li>1. Sulfur</li>
                    <li>2. Bicarbonates</li>
                    <li>3. Myclobutanil</li>
                    <li>4. Triadimefon</li>
                    <li>5. Propiconazole</li>
                    <li>6. Azoxystrobin</li>
                </ul>
                <p class="note"><b>* * * IMPORTANT NOTE * * *</b></p>
                <p>Be sure to follow local regulations and guidelines for application</p>


            """

        elif predicted_label=='Mango Sooty Mould':
            predicted_label = """
                <style>
                      li{
                        font-size: 15px;
                        margin-left: 90px;
                        margin-top: 15px;
                        margin-bottom: 15px;
                      }
                      h4{
                        font-size: 17px;
                        margin-top: 15px;
                      }
                      h4:hover{
                        cursor: pointer;
                      }

                      h3:hover{
                        cursor: pointer;
                        color: blue;
                        transform: scale(1.3);
                      }
                      .note{
                        text-align: center;
                        font-size: 16px;
                      }
                      p{
                        font-size: 13px;
                        text-align: center;
                      }

                </style>
                <h3><center><b>Mango Sooty Mould</b></center></h3>
                <h4>PESTICIDES TO BE USED:</h4>
                <ul>
                    <li>1. Imidacloprid (Neonicotinoid)</li>
                    <li>2. Thiamethoxam (Neonicotinoid)</li>
                    <li>3. Bifenthrin (Pyrethroid)</li>
                    <li>4. Lambda-cyhalothrin (Pyrethroid)</li>
                    <li>5. Insecticidal soap</li>
                    <li>6. Horticultural oil</li>
                </ul>
                <p class="note"><b>* * * IMPORTANT NOTE * * *</b></p>
                <p>Be sure to follow local regulations and guidelines for application</p>


            """
        else:
          predicted_label = """<h3 align="center">Mango Healthy</h3><br><br>
          <center>No need use Pesticides</center>"""

        return predicted_label


    except Exception as e:
        print(f"Error: {e}")
        return None

# List of class labels
all_labels = [
    'Mango Anthracrose',
    'Mango Bacterial Canker',
    'Mango Cutting Weevil',
    'Mango Die Back',
    'Mango Gall Midge',
    'Mango Healthy',
    'Mango Powdery Mildew',
    'Mango Sooty Mould'
]

# Define the Gradio interface
def gradio_predict(image_file):
    return predict_disease(image_file, model, all_labels)  # Pass the model to the function

# Create a Gradio interface
gr_interface = gr.Interface(
    fn=gradio_predict,  # Function to call for predictions
    inputs=gr.Image(type="filepath"),  # Upload image as file path
    outputs="html",  # Output will be the class label as text
    title="Mango Disease Predictor",
    description="Upload an image of a plant to predict the disease.",
)

# Launch the Gradio app
gr_interface.launch(share=True)