tree-diagnosis-ai / utils.py
karthikmn's picture
Create utils.py
f8d3a2b verified
raw
history blame contribute delete
550 Bytes
def explain_prediction(label):
return f"The model detected features consistent with {label}, based on trained visual patterns."
def get_treatment_and_fertilizer(label):
treatments = {
"Powdery Mildew": ("Apply sulfur-based fungicide", "Add compost-rich organic fertilizer"),
"Leaf Rust": ("Use copper spray", "Potassium-rich fertilizer"),
}
return treatments.get(label, ("General pruning and monitoring", "Balanced NPK fertilizer"))
def check_alert(label):
return label in ["Leaf Rust", "Anthracnose", "Blight"]