Upload 7 files

.env

@@ -0,0 +1 @@
+GEMINI_API_KEY=AQ.Ab8RN6IV4H0ieIzGArl227Vlu-nx9kzvUkAn2uNekV8pt-g6qw

Dockerfile

@@ -0,0 +1,11 @@
+FROM python:3.10-slim
+
+WORKDIR /code
+
+COPY ./requirements.txt /code/requirements.txt
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+COPY . /code
+
+# Hugging Face strictly requires apps to listen on port 7860
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

disease_config.json

@@ -0,0 +1,361 @@
+{
+  "model": {
+    "name": "FarmGuard Nigeria - Plant Disease Detection",
+    "version": "1.0.0",
+    "description": "MobileNetV2-based plant disease detection for Nigerian crops",
+    "framework": "tensorflow",
+    "input_shape": [224, 224, 3],
+    "supported_formats": ["jpeg", "jpg", "png"]
+  },
+  "classes": [
+    {
+      "id": "Corn_(maize)___Cercospora_leaf_spot Gray_leaf_spot",
+      "name": "Cercospora Leaf Spot / Gray Leaf Spot",
+      "plant": "Maize",
+      "disease": "Cercospora Leaf Spot",
+      "severity": "moderate",
+      "symptoms": ["Rectangular grey or tan lesions between leaf veins", "Lesions run parallel to leaf veins", "Premature leaf death in severe cases"],
+      "causes": "Fungal infection — Cercospora zeae-maydis, favoured by warm humid weather",
+      "treatment": {
+        "chemical": "Apply strobilurin fungicides (azoxystrobin, pyraclostrobin). Available as Amistar or Headline at agro-dealers.",
+        "cultural": "Remove and destroy infected crop residue after harvest. Improve field drainage.",
+        "preventive": "Plant resistant maize varieties (IITA TELA maize). Practice 2-year crop rotation with legumes."
+      },
+      "urgency": "Moderate — treat within 2 weeks of first symptoms",
+      "economic_impact": "Can reduce yield by 20-50% if untreated in severe outbreaks"
+    },
+    {
+      "id": "Corn_(maize)___Common_rust_",
+      "name": "Common Rust",
+      "plant": "Maize",
+      "disease": "Common Rust",
+      "severity": "moderate",
+      "symptoms": ["Small, circular to elongated brick-red pustules on both leaf surfaces", "Pustules turn dark brown/black as they mature", "Severe infection causes yellowing and premature death of leaves"],
+      "causes": "Fungal infection — Puccinia sorghi, spreads by wind-blown spores",
+      "treatment": {
+        "chemical": "Apply mancozeb (Dithane M-45) or propiconazole (Tilt 25EC) fungicide. Mix 2g/litre and spray every 10-14 days.",
+        "cultural": "Remove heavily infected plants. Avoid planting in areas with poor air circulation.",
+        "preventive": "Use rust-resistant varieties. Plant early to avoid peak rust season. Apply potassium fertiliser to strengthen plant resistance."
+      },
+      "urgency": "Moderate — treat within 1-2 weeks",
+      "economic_impact": "Can reduce yield by 10-35% in susceptible varieties"
+    },
+    {
+      "id": "Corn_(maize)___Northern_Leaf_Blight",
+      "name": "Northern Leaf Blight",
+      "plant": "Maize",
+      "disease": "Northern Leaf Blight",
+      "severity": "high",
+      "symptoms": ["Long, cigar-shaped grey-green to tan lesions (5-15cm) on leaves", "Lesions may have wavy margins", "Severe infection causes large areas of leaf tissue to die"],
+      "causes": "Fungal infection — Exserohilum turcicum, favoured by cool, moist conditions",
+      "treatment": {
+        "chemical": "Apply propiconazole (Tilt 25EC) at 0.5ml/litre or mancozeb at first sign of disease. Repeat every 14 days.",
+        "cultural": "Remove and burn infected crop debris. Improve field drainage and air circulation.",
+        "preventive": "Plant resistant varieties (IITA TZSR series). Rotate with legumes like cowpea or soybean."
+      },
+      "urgency": "High — treat within 1 week of detection",
+      "economic_impact": "Can reduce yield by 30-50% in severe cases"
+    },
+    {
+      "id": "Corn_(maize)___healthy",
+      "name": "Healthy",
+      "plant": "Maize",
+      "disease": "Healthy",
+      "severity": "none",
+      "symptoms": ["Deep green, upright leaves with no lesions", "Strong healthy stalk", "Normal growth pattern"],
+      "causes": "No disease present",
+      "treatment": {
+        "chemical": "No treatment needed",
+        "cultural": "Continue regular care — ensure adequate NPK fertiliser and weeding",
+        "preventive": "Monitor weekly for early signs of fall armyworm and rust. Apply nitrogen at 4-6 leaf stage."
+      },
+      "urgency": "None — plant is healthy",
+      "economic_impact": "No negative impact — maintain good agronomic practices for optimal yield"
+    },
+    {
+      "id": "Orange___Haunglongbing_(Citrus_greening)",
+      "name": "Citrus Greening (Huanglongbing)",
+      "plant": "Orange / Citrus",
+      "disease": "Huanglongbing (HLB)",
+      "severity": "high",
+      "symptoms": ["Asymmetric yellowing (blotchy mottle) on leaves — unlike nutrient deficiency which is symmetric", "Small, lopsided, bitter fruits", "Twig die-back from top of tree downward", "Fruit remains green on one side when ripe"],
+      "causes": "Bacterial infection — Candidatus Liberibacter, spread by Asian citrus psyllid insect",
+      "treatment": {
+        "chemical": "No cure exists. Control psyllid vector with imidacloprid (Confidor) soil drench or foliar spray to protect healthy trees.",
+        "cultural": "Remove and destroy infected trees immediately to prevent spread to healthy trees. Do not leave stumps.",
+        "preventive": "Use certified disease-free grafted seedlings from reputable nurseries. Apply systemic insecticide to control psyllid vector."
+      },
+      "urgency": "Critical — remove infected trees immediately, no cure",
+      "economic_impact": "Tree becomes unproductive within 3-5 years. Orchard can be completely destroyed if spread is not controlled."
+    },
+    {
+      "id": "Pepper,_bell___Bacterial_spot",
+      "name": "Bacterial Spot",
+      "plant": "Pepper",
+      "disease": "Bacterial Spot",
+      "severity": "moderate",
+      "symptoms": ["Small water-soaked spots on leaves that turn brown with yellow halo", "Spots may merge causing large brown areas", "Infected fruits have raised, scabby, water-soaked lesions", "Premature leaf and fruit drop"],
+      "causes": "Bacterial infection — Xanthomonas campestris pv. vesicatoria, spread by rain splash and handling",
+      "treatment": {
+        "chemical": "Apply copper oxychloride (Cobox/Funguran) at 3g/litre. Spray every 7-10 days during wet season. Available at agro-dealers nationwide.",
+        "cultural": "Remove infected leaves and fruits. Avoid overhead irrigation. Do not work in field when plants are wet.",
+        "preventive": "Use certified disease-free seeds. Treat seeds with hot water (50°C for 25 minutes) before planting. Practice 2-year rotation."
+      },
+      "urgency": "Moderate — treat within 1-2 weeks",
+      "economic_impact": "Can reduce marketable fruit yield by 20-40%"
+    },
+    {
+      "id": "Pepper,_bell___healthy",
+      "name": "Healthy",
+      "plant": "Pepper",
+      "disease": "Healthy",
+      "severity": "none",
+      "symptoms": ["Deep green glossy leaves with no spots", "Upright healthy stems", "Normal flower and fruit development"],
+      "causes": "No disease present",
+      "treatment": {
+        "chemical": "No treatment needed",
+        "cultural": "Maintain adequate staking, watering and NPK fertilisation",
+        "preventive": "Monitor weekly for whitefly and aphids which spread viruses. Use yellow sticky traps."
+      },
+      "urgency": "None — plant is healthy",
+      "economic_impact": "No negative impact"
+    },
+    {
+      "id": "Potato___Early_blight",
+      "name": "Early Blight",
+      "plant": "Potato",
+      "disease": "Early Blight",
+      "severity": "moderate",
+      "symptoms": ["Circular to angular dark brown spots with concentric rings (target-board pattern)", "Yellow halo around spots", "Spots first appear on older lower leaves", "Premature defoliation in severe cases"],
+      "causes": "Fungal infection — Alternaria solani, favoured by warm days and cool nights",
+      "treatment": {
+        "chemical": "Apply mancozeb (Dithane M-45) at 2g/litre every 7-10 days. Chlorothalonil (Bravo) is also effective. Both available at agro-dealers.",
+        "cultural": "Remove and destroy infected leaves. Avoid overhead watering. Mulch around plant base.",
+        "preventive": "Use certified disease-free seed tubers. Apply balanced NPK fertiliser — stressed plants are more susceptible."
+      },
+      "urgency": "Moderate — treat within 1-2 weeks",
+      "economic_impact": "Can reduce yield by 10-30% if left untreated"
+    },
+    {
+      "id": "Potato___Late_blight",
+      "name": "Late Blight",
+      "plant": "Potato",
+      "disease": "Late Blight",
+      "severity": "high",
+      "symptoms": ["Large, irregular, water-soaked, pale green to brown lesions on leaves", "White cottony fungal growth on underside of leaf in humid conditions", "Dark brown lesions on stems", "Infected tubers show reddish-brown dry rot under skin"],
+      "causes": "Oomycete — Phytophthora infestans. Spreads very rapidly in cool, wet, humid weather.",
+      "treatment": {
+        "chemical": "Apply metalaxyl+mancozeb (Ridomil Gold) immediately at 2.5g/litre. This spreads VERY fast — spray within 24 hours of detection and repeat every 7 days.",
+        "cultural": "Remove and destroy all infected plant material immediately. Do not compost. Harvest tubers early if infection is severe.",
+        "preventive": "Apply preventive fungicide sprays before wet season. Use resistant varieties. Ensure good field drainage."
+      },
+      "urgency": "CRITICAL — treat within 24 hours. Can destroy entire crop in days.",
+      "economic_impact": "Can destroy 100% of crop within 1-2 weeks under favourable conditions. Caused the Irish Potato Famine."
+    },
+    {
+      "id": "Potato___healthy",
+      "name": "Healthy",
+      "plant": "Potato",
+      "disease": "Healthy",
+      "severity": "none",
+      "symptoms": ["Dark green healthy leaves", "Strong upright stems", "No lesions or discolouration"],
+      "causes": "No disease present",
+      "treatment": {
+        "chemical": "No treatment needed",
+        "cultural": "Maintain adequate hilling, irrigation and NPK fertilisation",
+        "preventive": "Apply preventive fungicide before rainy season. Monitor weekly for late blight symptoms."
+      },
+      "urgency": "None — plant is healthy",
+      "economic_impact": "No negative impact"
+    },
+    {
+      "id": "Soybean___healthy",
+      "name": "Healthy",
+      "plant": "Soybean",
+      "disease": "Healthy",
+      "severity": "none",
+      "symptoms": ["Trifoliate dark green leaves with no spots", "Normal nodule formation on roots", "Healthy pod development"],
+      "causes": "No disease present",
+      "treatment": {
+        "chemical": "No treatment needed",
+        "cultural": "Ensure adequate phosphorus and potassium. Inoculate seeds with Bradyrhizobium for nitrogen fixation.",
+        "preventive": "Monitor for soybean rust (Phakopsora pachyrhizi) which is the main threat. Apply preventive fungicide at flowering."
+      },
+      "urgency": "None — plant is healthy",
+      "economic_impact": "No negative impact"
+    },
+    {
+      "id": "Tomato___Bacterial_spot",
+      "name": "Bacterial Spot",
+      "plant": "Tomato",
+      "disease": "Bacterial Spot",
+      "severity": "moderate",
+      "symptoms": ["Small, dark, water-soaked spots on leaves, stems and fruits", "Spots turn brown with yellow halo on leaves", "Raised, scabby brown lesions on fruits", "Severe defoliation in wet conditions"],
+      "causes": "Bacterial infection — Xanthomonas vesicatoria, spread by rain, wind and contaminated tools",
+      "treatment": {
+        "chemical": "Apply copper oxychloride (Cobox) at 3g/litre every 7-10 days. Copper hydroxide (Kocide) is also effective.",
+        "cultural": "Remove infected leaves and fruits. Avoid overhead irrigation. Disinfect tools with bleach solution.",
+        "preventive": "Use certified disease-free seeds. Treat seeds with hot water before planting. Rotate with non-solanaceous crops."
+      },
+      "urgency": "Moderate — treat within 1-2 weeks",
+      "economic_impact": "Can reduce marketable yield by 20-40%"
+    },
+    {
+      "id": "Tomato___Early_blight",
+      "name": "Early Blight",
+      "plant": "Tomato",
+      "disease": "Early Blight",
+      "severity": "moderate",
+      "symptoms": ["Dark brown circular spots with concentric rings on older lower leaves", "Yellow halo around spots", "Stem lesions near soil line (collar rot)", "Premature defoliation from bottom upward"],
+      "causes": "Fungal infection — Alternaria solani, favoured by alternating wet and dry periods",
+      "treatment": {
+        "chemical": "Apply copper-based fungicide or mancozeb (Dithane M-45) at 2g/litre. Spray every 7-10 days. Available at all agro-dealers.",
+        "cultural": "Remove infected leaves immediately. Avoid wetting foliage when watering. Mulch around base of plants.",
+        "preventive": "Stake plants for good air circulation. Apply balanced fertiliser — nitrogen deficiency increases susceptibility."
+      },
+      "urgency": "Moderate — treat within 1-2 weeks",
+      "economic_impact": "Can reduce yield by 15-30% and reduce fruit quality"
+    },
+    {
+      "id": "Tomato___Late_blight",
+      "name": "Late Blight",
+      "plant": "Tomato",
+      "disease": "Late Blight",
+      "severity": "high",
+      "symptoms": ["Large irregular pale green to brown water-soaked lesions on leaves", "White cottony growth on underside of leaf in humid conditions", "Dark brown lesions on stems", "Brown firm rot on green or ripe fruits"],
+      "causes": "Oomycete — Phytophthora infestans. Same pathogen as potato late blight. Spreads extremely fast.",
+      "treatment": {
+        "chemical": "Apply Ridomil Gold (metalaxyl + mancozeb) IMMEDIATELY at 2.5g/litre. Repeat every 7 days. Do not delay.",
+        "cultural": "Remove and destroy ALL infected plant parts. Do not leave on field. Harvest fruits before they become infected.",
+        "preventive": "Apply preventive fungicide before wet season. Ensure good drainage. Avoid dense planting."
+      },
+      "urgency": "CRITICAL — treat within 24 hours. Can destroy crop within 1 week.",
+      "economic_impact": "Can destroy entire tomato crop within 1-2 weeks. Major cause of tomato losses in Nigeria."
+    },
+    {
+      "id": "Tomato___Leaf_Mold",
+      "name": "Leaf Mold",
+      "plant": "Tomato",
+      "disease": "Leaf Mold",
+      "severity": "moderate",
+      "symptoms": ["Pale green to yellow spots on upper leaf surface", "Olive-green to grey-brown velvety growth on underside of leaves", "Infected leaves curl, wither and drop", "Most common in greenhouses or dense plantings"],
+      "causes": "Fungal infection — Passalora fulva, thrives in high humidity (>85%) and moderate temperatures",
+      "treatment": {
+        "chemical": "Apply chlorothalonil (Bravo) or mancozeb. Improve ventilation before spraying. Spray both upper and lower leaf surfaces.",
+        "cultural": "Remove infected leaves. Reduce humidity by improving spacing and ventilation. Avoid overhead irrigation.",
+        "preventive": "Maintain plant spacing of at least 60cm. Stake plants. Use resistant varieties where available."
+      },
+      "urgency": "Moderate — treat within 1-2 weeks",
+      "economic_impact": "Can reduce yield by 10-20% mainly through reduced photosynthesis"
+    },
+    {
+      "id": "Tomato___Septoria_leaf_spot",
+      "name": "Septoria Leaf Spot",
+      "plant": "Tomato",
+      "disease": "Septoria Leaf Spot",
+      "severity": "moderate",
+      "symptoms": ["Many small circular spots with dark brown border and grey-white centre", "Tiny black dots visible in centre of spots (fungal pycnidia)", "Starts on lower leaves and moves upward", "Severe defoliation leaving only top leaves"],
+      "causes": "Fungal infection — Septoria lycopersici, spreads by rain splash and tools",
+      "treatment": {
+        "chemical": "Apply mancozeb (Dithane M-45) or chlorothalonil at first sign. Spray every 7-10 days. Remove lower infected leaves first.",
+        "cultural": "Remove infected leaves. Avoid working in field when wet. Mulch to prevent soil splash.",
+        "preventive": "Rotate with non-solanaceous crops for 2-3 seasons. Destroy crop residue after harvest."
+      },
+      "urgency": "Moderate — treat within 2 weeks",
+      "economic_impact": "Reduces photosynthesis and can cut yield by 15-25%"
+    },
+    {
+      "id": "Tomato___Spider_mites Two-spotted_spider_mite",
+      "name": "Spider Mites (Two-spotted)",
+      "plant": "Tomato",
+      "disease": "Spider Mite Infestation",
+      "severity": "moderate",
+      "symptoms": ["Fine stippling (tiny yellow dots) on upper leaf surface", "Bronzing or yellowing of leaves", "Fine silky webbing on underside of leaves and between stems", "Tiny moving dots (mites) visible on underside of leaves", "Leaves dry up and fall in severe infestation"],
+      "causes": "Pest — Tetranychus urticae. Population explodes in hot dry weather. Spreads by wind and contact.",
+      "treatment": {
+        "chemical": "Apply abamectin (Dynamec/Vertimec) at 1ml/5litres or spiromesifen (Oberon) — these are specific miticides. Regular insecticides do NOT kill mites effectively.",
+        "cultural": "Spray plants with strong water jets to dislodge mites. Remove heavily infested leaves.",
+        "preventive": "Avoid dusty conditions. Maintain adequate irrigation — water-stressed plants are more susceptible. Avoid overuse of broad-spectrum insecticides which kill natural predators."
+      },
+      "urgency": "Moderate — treat within 1 week, populations double every 3-5 days in hot weather",
+      "economic_impact": "Can reduce yield by 20-40% and cause complete crop failure in severe dry season outbreaks"
+    },
+    {
+      "id": "Tomato___Target_Spot",
+      "name": "Target Spot",
+      "plant": "Tomato",
+      "disease": "Target Spot",
+      "severity": "moderate",
+      "symptoms": ["Circular brown spots with concentric rings (similar to early blight but larger)", "Yellow halo around spots", "Affects leaves, stems and fruits", "Dark brown to black sunken lesions on fruits"],
+      "causes": "Fungal infection — Corynespora cassiicola, favoured by warm humid conditions",
+      "treatment": {
+        "chemical": "Apply azoxystrobin (Amistar) or chlorothalonil (Bravo) fungicide every 10-14 days.",
+        "cultural": "Remove infected plant material. Stake plants for better air circulation. Avoid overhead irrigation.",
+        "preventive": "Rotate crops. Destroy crop debris after harvest. Plant with adequate spacing."
+      },
+      "urgency": "Moderate — treat within 2 weeks",
+      "economic_impact": "Can reduce yield by 10-25%"
+    },
+    {
+      "id": "Tomato___Tomato_Yellow_Leaf_Curl_Virus",
+      "name": "Tomato Yellow Leaf Curl Virus (TYLCV)",
+      "plant": "Tomato",
+      "disease": "Yellow Leaf Curl Virus",
+      "severity": "high",
+      "symptoms": ["Upward curling and cupping of young leaves", "Yellow margins on curled leaves", "Stunted plants with reduced internode length", "Very few or no flowers and fruits", "Silverleaf whitefly visible on underside of leaves"],
+      "causes": "Viral disease spread by silverleaf whitefly (Bemisia tabaci). No chemical cure for virus.",
+      "treatment": {
+        "chemical": "No cure for the virus. Control whitefly vector with imidacloprid (Confidor) or spiromesifen (Oberon) every 7 days. Use yellow sticky traps.",
+        "cultural": "Remove and destroy ALL infected plants immediately to prevent spread. Do not leave infected plants in field.",
+        "preventive": "Use TYLCV-resistant varieties (Cobra F1, Lindo F1, Tylka F1 — available from seed companies). Use insect-proof netting in nurseries."
+      },
+      "urgency": "High — remove infected plants immediately, no cure",
+      "economic_impact": "Can cause 100% yield loss. Major cause of tomato crop failure in Nigeria, especially dry season."
+    },
+    {
+      "id": "Tomato___Tomato_mosaic_virus",
+      "name": "Tomato Mosaic Virus (ToMV)",
+      "plant": "Tomato",
+      "disease": "Mosaic Virus",
+      "severity": "high",
+      "symptoms": ["Light and dark green mosaic pattern on leaves", "Leaf distortion, blistering and fern-like appearance", "Stunted plant growth", "Reduced and malformed fruits", "Internal browning of fruits"],
+      "causes": "Viral disease — Tomato mosaic virus. Spread by mechanical contact, contaminated tools, and hands. Very stable virus.",
+      "treatment": {
+        "chemical": "No chemical cure for virus. Disinfect all tools with 10% bleach or 70% alcohol. Wash hands thoroughly before handling plants.",
+        "cultural": "Remove and destroy infected plants. Do not smoke near plants (tobacco mosaic virus can infect tomatoes). Disinfect tools between plants.",
+        "preventive": "Use certified virus-free seeds or seedlings. Wash hands and disinfect tools frequently. Control aphids which may spread other viruses."
+      },
+      "urgency": "High — remove infected plants immediately",
+      "economic_impact": "Can reduce yield by 25-50%. Infected fruits are unmarketable."
+    },
+    {
+      "id": "Tomato___healthy",
+      "name": "Healthy",
+      "plant": "Tomato",
+      "disease": "Healthy",
+      "severity": "none",
+      "symptoms": ["Dark green healthy leaves", "Strong upright stems", "Normal flower and fruit development", "No spots, lesions or discolouration"],
+      "causes": "No disease present",
+      "treatment": {
+        "chemical": "No treatment needed",
+        "cultural": "Maintain staking, adequate NPK fertilisation and consistent irrigation",
+        "preventive": "Monitor weekly for whitefly, aphids and early signs of blight. Apply preventive copper spray before rainy season."
+      },
+      "urgency": "None — plant is healthy",
+      "economic_impact": "No negative impact — maintain good practices for optimal yield"
+    }
+  ],
+  "severity_levels": {
+    "none": {"level": 0, "color": "#16a34a", "description": "No disease — plant is healthy"},
+    "low": {"level": 1, "color": "#ca8a04", "description": "Minor infection — monitor closely"},
+    "moderate": {"level": 2, "color": "#ea580c", "description": "Moderate infection — treatment recommended within 1-2 weeks"},
+    "high": {"level": 3, "color": "#dc2626", "description": "Severe infection — immediate treatment required"},
+    "very_high": {"level": 4, "color": "#7f1d1d", "description": "Critical — aggressive treatment needed urgently"}
+  },
+  "confidence_thresholds": {
+    "high_confidence": 0.70,
+    "medium_confidence": 0.40,
+    "low_confidence": 0.20,
+    "uncertain": 0.10
+  }
+}

main.py

@@ -0,0 +1,237 @@
+"""
+FarmGuard Nigeria - Plant Disease Detection API
+Using Google Gemini Vision for accurate real photo analysis
+"""
+import json
+import logging
+import os
+import base64
+from google import genai
+from google.genai import types
+from dotenv import load_dotenv
+from fastapi import FastAPI, File, UploadFile, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
+
+# Load .env file
+load_dotenv()
+
+# Setup logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+app = FastAPI(
+    title="FarmGuard Nigeria API",
+    description="AI-powered plant disease detection for Nigerian farmers",
+    version="2.0.0"
+)
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# Load disease config
+config_path = os.path.join(os.path.dirname(__file__), "disease_config.json")
+with open(config_path) as f:
+    DISEASE_CONFIG = json.load(f)
+
+# Setup Gemini
+GEMINI_KEY = os.getenv("GEMINI_API_KEY")
+gemini_client = None
+
+if GEMINI_KEY:
+    gemini_client = genai.Client(api_key=GEMINI_KEY)
+    logger.info("Gemini Vision API ready")
+else:
+    logger.warning("No GEMINI_API_KEY found in .env file")
+
+
+@app.get("/")
+def root():
+    return {
+        "message": "FarmGuard Nigeria API is running!",
+        "version": "2.0.0",
+        "gemini_ready": gemini_client is not None,
+        "docs": "/docs"
+    }
+
+
+@app.get("/api/health")
+def health():
+    return {
+        "status": "healthy",
+        "gemini_ready": gemini_client is not None,
+    }
+
+
+@app.post("/api/predict")
+async def predict(file: UploadFile = File(...)):
+    """Detect plant disease from uploaded leaf image using Gemini Vision."""
+
+    if file.content_type not in ["image/jpeg", "image/jpg", "image/png"]:
+        raise HTTPException(
+            status_code=400,
+            detail="Only JPEG and PNG images are supported"
+        )
+
+    image_bytes = await file.read()
+
+    if len(image_bytes) > 8 * 1024 * 1024:
+        raise HTTPException(
+            status_code=400,
+            detail="Image too large. Maximum size is 8MB"
+        )
+
+    if gemini_client is None:
+        raise HTTPException(
+            status_code=503,
+            detail="Gemini API not configured. Add GEMINI_API_KEY to your .env file"
+        )
+
+    try:
+        mime_type = file.content_type
+
+        prompt = """You are an expert Nigerian agricultural plant pathologist.
+
+Analyse this plant leaf image and respond ONLY with a JSON object in this exact format with no extra text:
+{
+  "crop_identified": true,
+  "plant": "crop name e.g. Tomato, Maize, Cassava, Potato, Pepper, Yam, Plantain",
+  "disease": "disease name or Healthy",
+  "is_healthy": false,
+  "confidence": 0.85,
+  "severity": "none or low or moderate or high or very_high",
+  "symptoms": ["symptom 1", "symptom 2", "symptom 3"],
+  "causes": "what causes this disease",
+  "treatment": {
+    "chemical": "specific chemical treatment with Nigerian product names e.g. Dithane M-45, Cobox, Ridomil Gold, Confidor",
+    "cultural": "cultural and physical management practices",
+    "preventive": "prevention measures for Nigerian farmers"
+  },
+  "urgency": "treatment urgency description",
+  "economic_impact": "potential yield and economic impact on Nigerian farmer",
+  "message": ""
+}
+
+Rules:
+- If you cannot see a plant leaf clearly, set crop_identified to false and explain in message
+- Focus on crops grown in Nigeria: Tomato, Maize, Cassava, Potato, Pepper, Yam, Plantain, Rice, Cowpea, Soybean, Orange, Groundnut, Banana
+- Always include locally available Nigerian product names in chemical treatment
+- Be specific and practical for Nigerian smallholder farmers
+- Respond with ONLY the JSON object, nothing else"""
+
+        # Call Gemini with new SDK
+        response = gemini_client.models.generate_content(
+            model="gemini-2.5-flash",
+            contents=[
+                types.Part.from_bytes(data=image_bytes, mime_type=mime_type),
+                prompt
+            ]
+        )
+
+        text = response.text.strip()
+
+        # Remove markdown code blocks if present
+        if "```json" in text:
+            text = text.split("```json")[1].split("```")[0].strip()
+        elif "```" in text:
+            text = text.split("```")[1].split("```")[0].strip()
+
+        result = json.loads(text)
+
+        # Handle unidentified image
+        if not result.get("crop_identified", False):
+            return {
+                "success": False,
+                "crop_identified": False,
+                "message": result.get("message", "Could not identify a plant leaf. Please upload a clear close-up photo of a single leaf."),
+                "top_prediction": None,
+                "all_predictions": [],
+                "is_healthy": False
+            }
+
+        severity = result.get("severity", "moderate")
+        severity_colors = {
+            "none": "#16a34a", "low": "#ca8a04",
+            "moderate": "#ea580c", "high": "#dc2626", "very_high": "#7f1d1d"
+        }
+
+        top_prediction = {
+            "plant": result.get("plant", "Unknown"),
+            "disease": result.get("disease", "Unknown"),
+            "confidence": float(result.get("confidence", 0.8)),
+            "severity": severity,
+            "severity_color": severity_colors.get(severity, "#ea580c"),
+            "symptoms": result.get("symptoms", []),
+            "causes": result.get("causes", ""),
+            "treatment": result.get("treatment", {}),
+            "urgency": result.get("urgency", ""),
+            "economic_impact": result.get("economic_impact", "")
+        }
+
+        conf = float(result.get("confidence", 0.8))
+        confidence_level = "high" if conf >= 0.7 else "medium" if conf >= 0.4 else "low"
+
+        return {
+            "success": True,
+            "crop_identified": True,
+            "is_healthy": result.get("is_healthy", False),
+            "top_prediction": top_prediction,
+            "all_predictions": [top_prediction],
+            "confidence_level": confidence_level,
+            "message": None
+        }
+
+    except json.JSONDecodeError as e:
+        logger.error(f"JSON parse error: {e}")
+        raise HTTPException(
+            status_code=500,
+            detail="Failed to parse AI response. Please try again."
+        )
+    except Exception as e:
+        logger.error(f"Gemini error: {e}")
+        raise HTTPException(
+            status_code=500,
+            detail=f"AI analysis failed: {str(e)}"
+        )
+
+
+@app.get("/api/recommend")
+def recommend(state: str = "Rivers", soil: str = "loamy", season: str = "rainy"):
+    """Get crop recommendations based on location, soil and season."""
+
+    recommendations = {
+        "rainy": {
+            "crops": ["Cassava", "Yam", "Maize", "Plantain", "Cocoyam", "Waterleaf", "Fluted Pumpkin (Ugu)"],
+            "tip": "Rainy season is ideal for root crops and leafy vegetables. Ensure good drainage to prevent waterlogging."
+        },
+        "dry": {
+            "crops": ["Tomato", "Pepper", "Onion", "Carrot", "Cabbage", "Cowpea", "Soybean", "Groundnut"],
+            "tip": "Dry season farming requires irrigation. Vegetables thrive well with consistent water supply."
+        }
+    }
+
+    soil_tips = {
+        "loamy": "Loamy soil is excellent and suitable for almost all Nigerian crops. Maintain organic matter with compost.",
+        "sandy": "Sandy soil drains fast — best for cassava, groundnut and sweet potato. Add organic matter to improve water retention.",
+        "clay": "Clay soil retains water well — suitable for rice and yam. Ensure adequate drainage ridges to avoid waterlogging."
+    }
+
+    season_data = recommendations.get(season.lower(), recommendations["rainy"])
+    soil_tip = soil_tips.get(soil.lower(), "Prepare soil well with compost before planting.")
+
+    return {
+        "state": state,
+        "season": season,
+        "soil_type": soil,
+        "recommended_crops": season_data["crops"],
+        "season_tip": season_data["tip"],
+        "soil_tip": soil_tip
+    }
+
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run("main:app", host="0.0.0.0", port=8000, reload=True)

models/README.txt

@@ -0,0 +1 @@
+placeholder - replace with plant_weights.weights.h5

models/plant_weights.weights.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:805f5c71dda0a81f148ace65abe4912bd777dfc0b9323253f1886a311fcea475
+size 13463744

requirements.txt

@@ -0,0 +1,8 @@
+fastapi==0.104.1
+uvicorn==0.24.0
+python-multipart==0.0.6
+pydantic==2.5.0
+pillow==10.1.0
+numpy==1.26.4
+tensorflow==2.19.0
+python-dotenv==1.0.0