Update app.py

app.py

@@ -1,4 +1,3 @@
-import streamlit as st
 import cv2
 import numpy as np
 import tensorflow as tf
@@ -92,84 +91,6 @@ treatment_dict = {
     }
 }
 
-# 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", "Phytophthora Blight", "Blossom End Rot", 
-        "Pepper Weevil", "Fusarium Wilt", "Downy
-
-    ]
 def preprocess_image(image, image_size=(224, 224)):
     # Convert image to grayscale
     image = np.array(image.convert('L'))
@@ -201,6 +122,11 @@ if uploaded_file is not None:
     predictions = model.predict(processed_image)
     probabilities = predictions[0]
 
+    # Display probabilities for each class
+    for i, label in enumerate(class_labels):
+        if probabilities[i] > 0:
+            st.write(f"{label}: {probabilities[i]:.2f}")
+
     # Show predicted class
     predicted_class = class_labels[np.argmax(probabilities)]
     st.write(f"Classe prédite: {predicted_class}")
@@ -212,4 +138,4 @@ if uploaded_file is not None:
         if treatment_info['medicines']:
             st.write("Médicaments recommandés:", ', '.join(treatment_info['medicines']))
         else:
-            st.write("Aucun médicament requis.")
+            st.write("Aucun médicament requis.")