Update app.py

app.py

@@ -1,169 +1,68 @@
+### Health Management APP
+from dotenv import load_dotenv
+
+load_dotenv() ## load all the environment variables
+
 import streamlit as st
-import cv2
-import numpy as np
-import tensorflow as tf
-from tensorflow.keras.models import load_model
-from tensorflow.keras.preprocessing.image import img_to_array
+import os
+import google.generativeai as genai
 from PIL import Image
 
-# Load the pre-trained model
-model = load_model('plant_diseases.h5')
+genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
+
+## Function to load Google Gemini Pro Vision API And get response
+
+def get_gemini_repsonse(input,image,prompt):
+    model=genai.GenerativeModel('gemini-pro-vision')
+    response=model.generate_content([input,image[0],prompt])
+    return response.text
+
+def input_image_setup(uploaded_file):
+    # Check if a file has been uploaded
+    if uploaded_file is not None:
+        # Read the file into bytes
+        bytes_data = uploaded_file.getvalue()
+
+        image_parts = [
+            {
+                "mime_type": uploaded_file.type,  # Get the mime type of the uploaded file
+                "data": bytes_data
+            }
+        ]
+        return image_parts
+    else:
+        raise FileNotFoundError("No file uploaded")
+    
+##initialize our streamlit app
+
+st.set_page_config(page_title="Crop Disease Detection App")
+
+st.header("Gemini Crop Disease App")
+input=st.text_input("Input Prompt: ",key="input")
+uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
+image=""   
+if uploaded_file is not None:
+    image = Image.open(uploaded_file)
+    st.image(image, caption="Uploaded Image.", use_column_width=True)
+
+
+submit=st.button("Predict Crop/Plant Health")
 
-# Class labels (replace with your own classes)
-class_labels = [
-    'Piment: Bacterial_spot', 
-    'Piment: healthy', 
-    'Pomme de terre: Early_blight', 
-    'Pomme de terre: Late_blight', 
-    'Pomme de terre: Healthy', 
-    'Tomate: Bacterial Spot', 
-    'Tomate: Early Blight', 
-    'Tomate: Late Blight', 
-    'Tomate: Leaf mold', 
-    'Tomate: Septoria leaf spot', 
-    'Tomate: Spider mites', 
-    'Tomate: Spot', 
-    'Tomate: Yellow Leaf Curl', 
-    'Tomate: Virus Mosaïque', 
-    'Tomate: Healthy'
-]
+input_prompt="""
+        "You are an expert in computer vision and agriculture who can easily predict the disease of the plant. "
+        "Analyze the following image and provide 6 outputs in a structured table format: "
+        "1. Crop in the image, "
+        "2. Whether it is infected or healthy, "
+        "3. Type of disease (if any), "
+        "4. How confident out of 100% whether image is healthy or infected "
+        "5. Reason for the disease such as whether it is happening due to fungus, bacteria, insect bite, poor nutrition, etc., "
+        "6. Precautions for it."
+"""
 
-# Treatment dictionary mapping disease classes to treatment options
-treatment_dict = {
-    'Piment: Bacterial_spot': {
-        'treatment': 'Apply Copper Fungicide',
-        'medicines': ['Fungicide A', 'Fungicide B']
-    },
-    'Piment: healthy': {
-        'treatment': 'No treatment required',
-        'medicines': []
-    },
-    'Pomme de terre: Early_blight': {
-        'treatment': 'Use Neem Oil',
-        'medicines': ['Neem Oil', 'Fungicide C']
-    },
-    'Pomme de terre: Late_blight': {
-        'treatment': 'Apply Systemic Fungicide',
-        'medicines': ['Fungicide D']
-    },
-    'Pomme de terre: Healthy': {
-        'treatment': 'No treatment required',
-        'medicines': []
-    },
-    'Tomate: Bacterial Spot': {
-        'treatment': 'Remove infected leaves and apply bactericide',
-        'medicines': ['Bactericide A']
-    },
-    'Tomate: Early Blight': {
-        'treatment': 'Apply preventative fungicides',
-        'medicines': ['Fungicide E']
-    },
-    'Tomate: Late Blight': {
-        'treatment': 'Use fungicides with metalaxyl',
-        'medicines': ['Fungicide F']
-    },
-    'Tomate: Leaf mold': {
-        'treatment': 'Improve air circulation',
-        'medicines': ['Fungicide G']
-    },
-    'Tomate: Septoria leaf spot': {
-        'treatment': 'Remove infected leaves and spray with fungicide',
-        'medicines': ['Fungicide H']
-    },
-    'Tomate: Spider mites': {
-        'treatment': 'Use miticides or insecticidal soap',
-        'medicines': ['Miticide A']
-    },
-    'Tomate: Spot': {
-        'treatment': 'Check for pest issues and apply appropriate treatment',
-        'medicines': []
-    },
-    'Tomate: Yellow Leaf Curl': {
-        'treatment': 'Remove infected plants',
-        'medicines': []
-    },
-    'Tomate: Virus Mosaïque': {
-        'treatment': 'Remove infected plants',
-        'medicines': []
-    },
-    'Tomate: Healthy': {
-        'treatment': 'No treatment required',
-        'medicines': []
-    }
-}
+## If submit button is clicked
 
-# Crop disease dictionary
-crop_disease_dict = {
-    "Wheat": [
-        "Rust", "Powdery Mildew", "Fusarium Head Blight", "Septoria Leaf Blotch", 
-        "Barley Yellow Dwarf", "Leaf Spot", "Bacterial Blight", "Loose Smut", 
-        "Sharp Eyespot", "Root Rot", "Take-all", "Yellow Rust", "Brown Rust", 
-        "Black Point", "Stem Rust", "Glume Blotch", "Dwarf Wheat Rust", "Wheat Blast"
-    ],
-    "Rice": [
-        "Brown Spot", "Leaf Blast", "Bacterial Blight", "Sheath Blight", 
-        "Rice Tungro Virus", "False Smut", "Leaf Scald", "Stem Rot", 
-        "Glume Discoloration", "Narrow Brown Leaf Spot", "Mosaic Virus", 
-        "Ricelblast", "White Tip", "Bacterial Leaf Blight", "Fusarium Head Blight"
-    ],
-    "Corn": [
-        "Corn Smut", "Northern Corn Leaf Blight", "Gray Leaf Spot", "Goss's Wilt", 
-        "Southern Rust", "Common Rust", "Seedling Blight", "Ear Rot", 
-        "Fusarium Ear Rot", "Anthracnose", "Bacterial Leaf Streak", "Diplodia Ear Rot",
-        "Eyespot", "Root Rot", "Crown Rot", "Stalk Rot", "Kernel Rot"
-    ],
-    "Soybean": [
-        "Soybean Cyst Nematode", "Phytophthora Root Rot", "Bacterial Blight", "Downy Mildew", 
-        "Brown Spot", "Sclerotinia Stem Rot", "Frogeye Leaf Spot", "Pod Blight", 
-        "Root Rot", "Anthracnose", "Virus Diseases", "Septoria Leaf Spot", 
-        "Rust", "White Mold", "Sudden Death Syndrome", "Charcoal Rot"
-    ],
-    "Tomato": [
-        "Late Blight", "Early Blight", "Leaf Spot", "Fusarium Wilt", 
-        "Bacterial Spot", "Blossom End Rot", "Powdery Mildew", "Septoria Leaf Spot", 
-        "Tobacco Mosaic Virus", "Gray Mold", "Canker", "Phytophthora Root Rot", 
-        "Verticillium Wilt", "Downy Mildew", "Black Mold", "Nematode Infestation"
-    ],
-    "Potato": [
-        "Late Blight", "Early Blight", "Black Leg", "Scab", 
-        "Powdery Scab", "Fusarium Wilt", "Tuber Blight", "Soft Rot", 
-        "Verticillium Wilt", "Phytophthora Blight", "Golden Nematode", 
-        "Bacterial Ring Rot", "Leaf Roll Virus", "Potato Virus X", "Cyst Nematodes"
-    ],
-    "Cotton": [
-        "Cotton Wilt", "Boll Rot", "Alternaria Leaf Spot", "Bacterial Blight", 
-        "Fusarium Wilt", "Pink Bollworm", "Cotton Leaf Curl Virus", "Root Knot Nematode", 
-        "Seedling Disease", "Boll Weevil", "Leaf Spot", "Webworm", 
-        "Southern Blight", "Target Spot", "Anthracnose", "Thrips Damage"
-    ],
-    "Barley": [
-        "Barley Yellow Dwarf", "Powdery Mildew", "Net Blotch", "Fusarium Head Blight", 
-        "Scald", "Hulled Barley Disease", "Root Rot", "Bacterial Leaf Blight", 
-        "Corn Smut", "Kernel Blight", "Mosaic Virus", "Spot Blotch", 
-        "Streak", "Barley Yellow Mosaic"
-    ],
-    "Cabbage": [
-        "Black Rot", "Downy Mildew", "Cabbage Looper", "Alternaria Leaf Spot", 
-        "Clubroot", "Bacterial Soft Rot", "Fusarium Wilt", "Cabbage Maggot", 
-        "Diamondback Moth", "Bacterial Blight", "Mosaic Virus", "White Rust", 
-        "Leaf Spot", "Sclerotinia Rot"
-    ],
-    "Carrot": [
-        "Alternaria Leaf Blight", "Cavity Spot", "Fusarium Wilt", "Bacterial Soft Rot", 
-        "Downy Mildew", "Carrot Rust Fly", "Crown Rot", "Powdery Mildew", 
-        "Bacterial Blight", "Sclerotinia Rot", "Root Knot Nematode", "Mosaic Virus", 
-        "Leaf Spot"
-    ],
-    "Onion": [
-        "Downy Mildew", "Onion Fly", "White Rot", "Botrytis Leaf Blight", 
-        "Fusarium Basal Rot", "Bacterial Soft Rot", "Pink Root", "Mosaic Virus", 
-        "Leaf Blight", "Pythium Root Rot", "Black Mold", "Fusarium Wilt"
-    ],
-    "Grapes": [
-        "Powdery Mildew", "Downy Mildew", "Botrytis Bunch Rot", "Black Rot", 
-        "Phomopsis Cane and Leaf Spot", "Crown Gall", "Pierce's Disease", 
-        "Bacterial Blight", "Eutypa Dieback", "Grapevine Leafroll", 
-        "Sooty Mold", "Anthracnose", "Mealybug Infestation", "Fungal Leaf Spot"
-    ],
-    "Peppers": [
-        "Bacterial Spot", "Powdery Mildew", "Ph
+if submit:
+    image_data=input_image_setup(uploaded_file)
+    response=get_gemini_repsonse(input_prompt,image_data,input)
+    st.subheader("The Response is")
+    st.write(response)