Update app.py

Added the Treatment Code

app.py

@@ -31,18 +31,56 @@ CITRUS_CLASS_NAMES = [
     'Yellow dragon', 'Yellow leaves'
 ]
 
+# Orange / Citrus Treatment Knowledge Base
+CITRUS_TREATMENTS = {
+    'Citrus canker': 'Apply copper-based bactericides. Prune and destroy infected branches immediately.',
+    'Citrus greening': 'No known cure. Remove and destroy infected trees to prevent spread. Control psyllid insect populations.',
+    'Citrus mealybugs': 'Spray with insecticidal soap or neem oil. Introduce natural predators like ladybugs.',
+    'Die back': 'Prune affected branches well below the dying part. Ensure proper soil drainage and avoid overwatering.',
+    'Foliage damaged': 'Protect plants from extreme weather. Check for pests and ensure balanced fertilization.',
+    'Healthy leaf': 'Your citrus plant looks completely healthy! Maintain regular watering and nutrient care.',
+    'Powdery mildew': 'Apply sulfur-based fungicides or neem oil. Prune to improve air circulation around the leaves.',
+    'Shot hole': 'Apply copper-based fungicides during the dormant season. Avoid overhead watering to keep leaves dry.',
+    'Spiny whitefly': 'Use horticultural oils or insecticidal soaps. Clean the leaves to remove black sooty mold.',
+    'Yellow dragon': 'This is another name for Citrus Greening. There is no cure; remove the tree and control insect vectors.',
+    'Yellow leaves': 'Check for nutrient deficiencies (like Nitrogen or Iron). Do not overwater, and ensure good soil drainage.'
+}
+
 # --- CORN CONFIGURATION ---
 CORN_MODEL_PATH = "CornModel.tflite"
 CORN_CLASS_NAMES = ['Blight', 'Common_Rust', 'Gray_Leaf_Spot', 'Healthy']
 
+# Corn Treatment Knowledge Base
+CORN_TREATMENTS = {
+    'Blight': 'Apply appropriate fungicides. Rotate crops and manage crop residue to reduce fungal spores.',
+    'Common_Rust': 'Apply fungicides early when pustules first appear. Use rust-resistant corn hybrids for future planting.',
+    'Gray_Leaf_Spot': 'Use fungicides containing azoxystrobin or pyraclostrobin. Improve air circulation and practice crop rotation.',
+    'Healthy': 'Your corn plant looks healthy! Ensure consistent watering and balanced nitrogen fertilization.'
+}
+
 # --- APPLE CONFIGURATION ---
 APPLE_MODEL_PATH = "apple_modelH.tflite"
 APPLE_CLASS_NAMES = ['Apple Scab', 'Black Rot', 'Cedar Apple Rust', 'Healthy']
 
+# Apple Treatment Knowledge Base
+APPLE_TREATMENTS = {
+    'Apple Scab': 'Apply fungicides like captan or myclobutanil. Rake and destroy fallen leaves to prevent fungal spores from overwintering.',
+    'Black Rot': 'Prune out dead or diseased wood. Remove mummified fruit from the tree and ground. Apply fungicides early in the season.',
+    'Cedar Apple Rust': 'Remove nearby eastern red cedar hosts if possible. Apply protective fungicides in the spring before rain events.',
+    'Healthy': 'Your apple tree looks perfectly healthy! Maintain good pruning practices to ensure proper airflow and sunlight.'
+}
+
 # --- POTATO CONFIGURATION ---
 POTATO_MODEL_PATH = "potato_model.tflite"
 POTATO_CLASS_NAMES = ['Potato___Early_blight', 'Potato___Late_blight', 'Potato___healthy']
 
+# Potato Treatment Knowledge Base
+POTATO_TREATMENTS = {
+    'Potato___Early_blight': 'Remove affected leaves. Apply fungicides like chlorothalonil or copper. Practice crop rotation next season.',
+    'Potato___Late_blight': 'Apply targeted fungicides immediately. Destroy heavily infected plants to prevent rapid spread to nearby crops.',
+    'Potato___healthy': 'Your potato plant looks healthy! Maintain consistent watering and hill up soil around the base of the plant.'
+}
+
 # --- TOMATO CONFIGURATION ---
 TOMATO_MODEL_PATH = "TomatoModelH.tflite" 
 
@@ -54,10 +92,32 @@ TOMATO_CLASS_NAMES = [
     'Tomato___healthy'
 ]
 
+# 🏥 NEW: Tomato Treatment Knowledge Base
+TOMATO_TREATMENTS = {
+    'Tomato___Bacterial_spot': 'Apply copper-based fungicides. Avoid overhead watering to keep leaves dry.',
+    'Tomato___Early_blight': 'Remove affected lower leaves. Apply a fungicide containing chlorothalonil or copper.',
+    'Tomato___Late_blight': 'Apply fungicides specifically labeled for late blight. Destroy deeply infected plants.',
+    'Tomato___Leaf_Mold': 'Improve air circulation by pruning. Apply preventative fungicides.',
+    'Tomato___Septoria_leaf_spot': 'Remove infected leaves. Mulch around the base of the plant and apply fungicide.',
+    'Tomato___Spider_mites Two-spotted_spider_mite': 'Spray with insecticidal soap or neem oil. Introduce ladybugs.',
+    'Tomato___Target_Spot': 'Improve airflow and reduce humidity. Apply targeted fungicides.',
+    'Tomato___Tomato_Yellow_Leaf_Curl_Virus': 'Remove and destroy infected plants. Control whitefly populations.',
+    'Tomato___Tomato_mosaic_virus': 'No cure. Remove infected plants immediately. Wash hands and tools thoroughly.',
+    'Tomato___healthy': 'Your plant looks completely healthy! Keep up the good work with regular care.'
+}
+
 # --- RICE CONFIGURATION ---
 RICE_MODEL_PATH = "RiceModel.tflite"
 RICE_CLASS_NAMES = ['Bacterialblight', 'Blast', 'Brownspot', 'Tungro']
 
+# 🏥 NEW: Rice Treatment Knowledge Base
+RICE_TREATMENTS = {
+    'Bacterialblight': 'Apply copper-based bactericides. Avoid applying excessive nitrogen fertilizer and ensure good field drainage.',
+    'Blast': 'Apply fungicides like tricyclazole or azoxystrobin. Manage water levels carefully to avoid drought stress.',
+    'Brownspot': 'Improve soil fertility, especially with potassium and balanced nitrogen. Apply fungicides like propiconazole if severe.',
+    'Tungro': 'Control the green leafhopper insect vector using targeted insecticides. Immediately remove and destroy infected plants.'
+}
+
 # --- 2. LOAD Orange TFLITE MODEL ---
 citrus_interpreter = tf.lite.Interpreter(model_path=CITRUS_MODEL_PATH)
 citrus_interpreter.allocate_tensors()
@@ -135,6 +195,15 @@ async def predict_citrus(file: UploadFile = File(...)):
             "confidence": confidence
         }
 
+    # 🏥 NEW: Fetch the treatment advice
+    recommended_treatment = CITRUS_TREATMENTS.get(predicted_class, "Consult a local agricultural expert for treatment.")
+
+    return {
+        "class": predicted_class,
+        "confidence": confidence,
+        "treatment": recommended_treatment 
+    }
+
         
 # --- Corn's Code ---
 @app.post("/predict_corn")
@@ -162,9 +231,13 @@ async def predict_corn(file: UploadFile = File(...)):
     confidence = float(np.max(output_data[0]))
     predicted_class = CORN_CLASS_NAMES[prediction]
 
+    # 🏥 NEW: Fetch the treatment advice
+    recommended_treatment = CORN_TREATMENTS.get(predicted_class, "Consult a local agricultural expert for treatment.")
+
     return {
         "class": predicted_class,
-        "confidence": confidence
+        "confidence": confidence,
+        "treatment": recommended_treatment
     }
 
 # --- Apple's Code ---
@@ -193,9 +266,13 @@ async def predict_apple(file: UploadFile = File(...)):
     confidence = float(np.max(output_data[0]))
     predicted_class = APPLE_CLASS_NAMES[prediction]
 
+    # 🏥 NEW: Fetch the treatment advice
+    recommended_treatment = APPLE_TREATMENTS.get(predicted_class, "Consult a local agricultural expert for treatment.")
+
     return {
         "class": predicted_class,
-        "confidence": confidence
+        "confidence": confidence,
+        "treatment": recommended_treatment
     }
     
 # --- Potato's Code ---
@@ -224,10 +301,14 @@ async def predict_potato(file: UploadFile = File(...)):
     confidence = float(np.max(output_data[0]))
     predicted_class = POTATO_CLASS_NAMES[prediction]
 
+    # 🏥 NEW: Fetch the treatment advice
+    recommended_treatment = POTATO_TREATMENTS.get(predicted_class, "Consult a local agricultural expert for treatment.")
+
     return {
         "class": predicted_class,
-        "confidence": confidence
-    }    
+        "confidence": confidence,
+        "treatment": recommended_treatment
+    }   
 
 # --- Tomato's Code ---
 @app.post("/predict_tomato")
@@ -277,9 +358,16 @@ async def predict_tomato(file: UploadFile = File(...)):
         # Clean up the class name for the app response (e.g., "Tomato___Early_blight" -> "Early blight")
         display_name = predicted_class.replace('Tomato___', '').replace('_', ' ')
 
+        # Clean up the class name for the app response
+        display_name = predicted_class.replace('Tomato___', '').replace('_', ' ')
+
+        # 🏥 NEW: Fetch the treatment advice
+        recommended_treatment = TOMATO_TREATMENTS.get(predicted_class, "Consult a local agricultural expert for treatment.")
+
         return {
                 "class": display_name,
-                "confidence": confidence
+                "confidence": confidence,
+                "treatment": recommended_treatment
             }
             
     except Exception as e:
@@ -316,7 +404,14 @@ async def predict_rice(file: UploadFile = File(...)):
     # Add spaces to the class names for a cleaner app display
     display_name = predicted_class.replace('Bacterialblight', 'Bacterial Blight').replace('Brownspot', 'Brown Spot')
 
+    # Add spaces to the class names for a cleaner app display
+    display_name = predicted_class.replace('Bacterialblight', 'Bacterial Blight').replace('Brownspot', 'Brown Spot')
+
+    # 🏥 NEW: Fetch the treatment advice
+    recommended_treatment = RICE_TREATMENTS.get(predicted_class, "Consult a local agricultural expert for treatment.")
+
     return {
         "class": display_name,
-        "confidence": confidence
-    }    
+        "confidence": confidence,
+        "treatment": recommended_treatment
+    }