Update app.py

app.py

@@ -92,46 +92,78 @@ treatment_dict = {
     }
 }
 
-def preprocess_image(image, image_size=(224, 224)):
-    # Convert image to grayscale
-    image = np.array(image.convert('L'))
-    # Resize image
-    image = cv2.resize(image, image_size)
-    # Prepare image for the model
-    image = img_to_array(image)
-    image /= 255.0
-    image = np.expand_dims(image, axis=0)
-    return image
-
-# Streamlit app setup
-st.title("Classification des Maladies des Plantes")
-st.write("Téléchargez une image de plante pour la classification")
-
-uploaded_file = st.file_uploader("Choisissez une image...", type=["jpg", "jpeg", "png"])
-
-if uploaded_file is not None:
-    # Display the uploaded image
-    image = Image.open(uploaded_file)
-    st.image(image, caption='Image téléchargée', use_column_width=True)
-
-    st.write("Classification en cours...")
-
-    # Preprocess the image
-    processed_image = preprocess_image(image)
-
-    # Make predictions
-    predictions = model.predict(processed_image)
-    probabilities = predictions[0]
-
-    # Show predicted class
-    predicted_class = class_labels[np.argmax(probabilities)]
-    st.write(f"Classe prédite: {predicted_class}")
-
-    # Display treatment information
-    if predicted_class in treatment_dict:
-        treatment_info = treatment_dict[predicted_class]
-        st.write("Traitement recommandé:", treatment_info['treatment'])
-        if treatment_info['medicines']:
-            st.write("Médicaments recommandés:", ', '.join(treatment_info['medicines']))
-        else:
-            st.write("Aucun médicament requis.")
+# 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