Update app.py

app.py

@@ -4,70 +4,66 @@ from ultralytics import YOLO
 import gradio as gr
 from gradio_client import Client, handle_file
 
-# 🌿 Dictionary containing disease names, descriptions, remedies, and recommended fertilizers
+# 🌿 Dictionary containing disease details
 disease_info = {
     "bacterial spot": {
-        "description": "Bacterial spot is a common disease affecting tomato and pepper plants, causing dark, water-soaked spots on leaves and fruits.",
+        "description": "Causes dark, water-soaked spots on leaves and fruits.",
         "remedy": "Remove infected plant debris, use copper-based fungicides.",
         "fertilizer": "Use potassium-rich fertilizers to strengthen plant immunity."
     },
     "early blight": {
-        "description": "Early blight is a fungal disease that causes dark concentric rings on leaves, leading to defoliation.",
+        "description": "Causes dark concentric rings on leaves, leading to defoliation.",
         "remedy": "Apply fungicides, practice crop rotation.",
         "fertilizer": "Use nitrogen-rich fertilizers to improve plant resistance."
     },
     "healthy": {
-        "description": "The plant is healthy and does not show any signs of disease.",
+        "description": "No signs of disease detected.",
         "remedy": "No action needed.",
-        "fertilizer": "Balanced NPK fertilizers can maintain plant health."
+        "fertilizer": "Balanced NPK fertilizers to maintain health."
     },
     "late blight": {
-        "description": "Late blight is a devastating fungal disease that spreads rapidly under wet conditions.",
-        "remedy": "Remove infected plants, use fungicides.",
-        "fertilizer": "Calcium-based fertilizers help reduce susceptibility."
+        "description": "Rapidly spreading fungal disease, common in wet conditions.",
+        "remedy": "Remove infected plants, apply fungicides.",
+        "fertilizer": "Calcium-based fertilizers reduce susceptibility."
     },
     "leaf miner": {
-        "description": "Leaf miners create winding tunnels in leaves as larvae feed on plant tissue.",
+        "description": "Larvae create winding tunnels in leaves.",
         "remedy": "Use insecticidal sprays, remove affected leaves.",
-        "fertilizer": "Organic compost or vermicompost can strengthen plants against pests."
+        "fertilizer": "Organic compost or vermicompost to strengthen plants."
     },
     "leaf mold": {
-        "description": "Leaf mold appears as a yellowing of leaves with fuzzy mold on the undersides.",
-        "remedy": "Improve air circulation, use fungicides.",
-        "fertilizer": "Apply phosphorus-based fertilizers to enhance root health."
+        "description": "Yellowing leaves with fuzzy mold on undersides.",
+        "remedy": "Improve air circulation, apply fungicides.",
+        "fertilizer": "Phosphorus-based fertilizers to enhance root health."
     },
     "mosaic virus": {
-        "description": "Mosaic virus causes a mottled, yellow-green appearance on leaves, affecting photosynthesis.",
+        "description": "Causes mottled, yellow-green appearance on leaves.",
         "remedy": "Remove infected plants, control aphids.",
-        "fertilizer": "Avoid nitrogen excess; use balanced NPK fertilizers."
+        "fertilizer": "Balanced NPK fertilizers, avoid excess nitrogen."
     },
     "septoria": {
-        "description": "Septoria leaf spot appears as numerous small, circular brown spots on lower leaves.",
+        "description": "Small, circular brown spots on lower leaves.",
         "remedy": "Remove infected leaves, use fungicides.",
-        "fertilizer": "Use sulfur-based fertilizers to reduce fungal spread."
+        "fertilizer": "Sulfur-based fertilizers to reduce fungal spread."
     },
     "spider mites": {
-        "description": "Spider mites cause stippling and webbing on leaves, leading to leaf drop.",
+        "description": "Causes stippling and webbing on leaves.",
         "remedy": "Use miticides, introduce beneficial insects.",
-        "fertilizer": "Silicon-based fertilizers improve plant resistance to pests."
+        "fertilizer": "Silicon-based fertilizers improve resistance to pests."
     },
     "yellow leaf curl virus": {
-        "description": "This virus causes curled, yellowing leaves and stunted growth, transmitted by whiteflies.",
+        "description": "Curled, yellowing leaves and stunted growth.",
         "remedy": "Remove infected plants, control whiteflies.",
-        "fertilizer": "Zinc and boron-based fertilizers can reduce symptoms."
+        "fertilizer": "Zinc and boron-based fertilizers to reduce symptoms."
     }
 }
 
 def custom_images(img, target_size=(512, 512)):  # Fixed size
-    """
-    Resizes the input image to a fixed target size.
-    """
+    """Resizes the input image to a fixed target size."""
     return cv2.resize(img, target_size)
 
 def plant_disease_detect(img, model, original_size):
-    """
-    Performs plant disease detection using the YOLO model.
-    """
+    """Runs YOLO disease detection on the image."""
     detect_result = model(img)
     detect_img = detect_result[0].plot()
     detections = detect_result[0].boxes.data.tolist()
@@ -76,40 +72,48 @@ def plant_disease_detect(img, model, original_size):
     return detect_img, classes
 
 def predict(img):
+    """Processes image and returns detections with disease info."""
     model = YOLO('crop_disease_model.pt')
     original_size = (img.shape[1], img.shape[0])
-    resized_image = custom_images(img)  # Automatically resized
+    resized_image = custom_images(img)
     detect_img, classes = plant_disease_detect(resized_image, model, original_size)
-    
+
+    # 🟢 **Format table data for better readability**
     unique_classes = list(set(classes))
-    class_table = [[
-        cls, 
-        disease_info.get(cls.lower(), {}).get("description", "No description available"), 
-        disease_info.get(cls.lower(), {}).get("remedy", "No remedy available"), 
-        disease_info.get(cls.lower(), {}).get("fertilizer", "No fertilizer recommendation available")
-    ] for cls in unique_classes]
-    
-    return detect_img, class_table
+    table_data = []
+    for cls in unique_classes:
+        disease = cls
+        details = disease_info.get(cls.lower(), {})
+        description = details.get("description", "No description available")
+        remedy = details.get("remedy", "No remedy available")
+        fertilizer = details.get("fertilizer", "No recommendation available")
+        table_data.append([disease, description, remedy, fertilizer])
+
+    return detect_img, table_data  # Returns detection image + structured table
 
-# 🌱 Gradio Interface
+# 🌱 **Improved Gradio Interface**
 with gr.Blocks() as demo:
-    gr.Markdown("# 🌿 Crop Disease Diagnosis")
-    gr.Markdown("Upload an image of a plant to detect diseases and get recommendations.")
+    gr.Markdown("# 🌿 **Crop Disease Diagnosis**")
+    gr.Markdown("Upload a plant image to detect diseases and get expert recommendations.")
 
     with gr.Row():
         with gr.Column():
-            img_input = gr.Image(type="numpy", label="📷 Upload Image")
-            submit_btn = gr.Button("🔍 Detect Disease")
+            img_input = gr.Image(type="numpy", label="Upload Image")
+            submit_btn = gr.Button("Detect Disease")
         with gr.Column():
-            img_output = gr.Image(label="🖼️ Processed Image")
-            class_table = gr.Dataframe(headers=["Disease", "Description", "Remedy", "Recommended Fertilizer"], label="📋 Disease Details")
+            img_output = gr.Image(label="Processed Image")
+            class_table = gr.Dataframe(
+                headers=["Disease", "Description", "Remedy", "Recommended Fertilizer"],
+                datatype=["str", "str", "str", "str"],  # ✅ Defines column types
+                label="Disease Details"
+            )
 
     submit_btn.click(fn=predict, inputs=[img_input], outputs=[img_output, class_table])
 
-# Enable API Mode
+# 🌎 **Enable API Mode for External Requests**
 demo.launch(server_name="0.0.0.0", server_port=7860, share=True)
 
-# Hugging Face API Access
+# 🛰️ **Hugging Face API Access**
 client = Client("sankalp2606/Plant-Disease_diagnosis")
 print("HELLO")
 print(client.endpoints)