Update app.py

app.py

@@ -41,56 +41,19 @@ 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 = "CoRn_model.tflite"
 CORN_CLASS_NAMES = ['Common Rust', 'Corn Leaf Blight', 'Gray Leaf Spot', 'Healthy', 'Insects damage']
 
-# Corn Treatment Knowledge Base
-CORN_TREATMENTS = {
-    'Common Rust': 'Apply fungicides early when pustules first appear. Use rust-resistant corn hybrids for future planting.',
-    'Corn Leaf Blight': 'Apply appropriate fungicides. Rotate crops and manage crop residue to reduce fungal spores.',
-    '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.',
-    'Insects damage': 'Inspect leaves for insect activity. Apply appropriate insecticides or neem oil. Remove heavily damaged leaves.'
-}
-
 # --- 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 = "pot_ato_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 = "TOMAto_model.tflite" 
@@ -108,31 +71,11 @@ 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 = "New_Rice_Model.tflite"
 RICE_CLASS_NAMES = ['Bacterial Leaf Blight', 'Brown Spot', 'Healthy Rice Leaf', 'Leaf Blast', 'Leaf scald', 'Narrow Brown Leaf Spot', 'Rice Hispa', 'Sheath Blight']
 
-# 🏥 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)
@@ -205,13 +148,11 @@ async def predict_citrus(file: UploadFile = File(...)):
     confidence = float(np.max(output_data[0]))
     predicted_class = CITRUS_CLASS_NAMES[prediction]
 
-    # 🏥 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 
+        "confidence": confidence
     }
 
         
@@ -241,13 +182,10 @@ 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,
-        "treatment": recommended_treatment
+        "confidence": confidence
     }
 
 # --- Apple's Code ---
@@ -276,13 +214,11 @@ 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,
-        "treatment": recommended_treatment
+        "confidence": confidence
     }
     
 # --- Potato's Code ---
@@ -311,13 +247,11 @@ 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,
-        "treatment": recommended_treatment
+        "confidence": confidence
     }   
 
 # --- Tomato's Code ---
@@ -337,7 +271,7 @@ async def predict_tomato(file: UploadFile = File(...)):
         image = image.resize((224, 224))
         
         # 2. Custom Normalization: Converting [0, 255] to [-1.0, 1.0]
-        input_data = np.array(image).astype(np.float32)/255.0
+        input_data = np.array(image).astype(np.float32)
         input_data = np.expand_dims(input_data, axis=0)
 
         # --- RUN INFERENCE ---
@@ -354,13 +288,10 @@ async def predict_tomato(file: UploadFile = File(...)):
         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,
-                "treatment": recommended_treatment
+                "confidence": confidence
             }
             
     except Exception as e:
@@ -397,11 +328,7 @@ 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')
 
-    # 🏥 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,
-        "treatment": recommended_treatment
+        "confidence": confidence
     }