app.py

"""
VrukshaVed — Ayurvedic Plant Intelligence
FastAPI Backend with ConvNeXt-Small model + TTA inference
"""

import io
import json
import os
import sys
from pathlib import Path

import numpy as np
import torch
import torch.nn as nn
import torchvision.models as models
import torchvision.transforms as T
from fastapi import FastAPI, File, HTTPException, UploadFile
from fastapi.middleware.cors import CORSMiddleware
from fastapi.responses import JSONResponse
from fastapi.staticfiles import StaticFiles
from fastapi.templating import Jinja2Templates
from fastapi import Request
from PIL import Image

# ─── Paths ────────────────────────────────────────────────────────────────────
BASE_DIR = Path(__file__).parent
MODEL_PATH = BASE_DIR / "model" / "leaf80_best_convnext_small.pth"
LABELS_PATH = BASE_DIR / "model" / "leaf80_label_to_class.json"

# ─── Model Config ─────────────────────────────────────────────────────────────
DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
MODEL_NAME = "convnext_small"
NUM_CLASSES = 80
IMG_SIZE = 224
DROPOUT_P = 0.4
CONFIDENCE_THRESH = 0.50
MEAN = [0.485, 0.456, 0.406]
STD  = [0.229, 0.224, 0.225]

# ─── TTA Transforms ───────────────────────────────────────────────────────────
_norm = T.Normalize(MEAN, STD)
TTA_TRANSFORMS = [
    T.Compose([T.Resize((IMG_SIZE, IMG_SIZE)), T.ToTensor(), _norm]),
    T.Compose([T.Resize((IMG_SIZE, IMG_SIZE)), T.RandomHorizontalFlip(p=1.0), T.ToTensor(), _norm]),
    T.Compose([T.Resize((256, 256)), T.CenterCrop(IMG_SIZE), T.ToTensor(), _norm]),
    T.Compose([T.Resize((256, 256)), T.RandomCrop(IMG_SIZE), T.ToTensor(), _norm]),
    T.Compose([T.Resize((IMG_SIZE, IMG_SIZE)),
               T.ColorJitter(brightness=0.2, contrast=0.2, saturation=0.1),
               T.ToTensor(), _norm]),
]

# ─── Plant Info Database ───────────────────────────────────────────────────────
PLANT_DB = {
    "Aloevera": {
        "botanical_name": "Aloe barbadensis miller",
        "ayurvedic_name": "Kumari",
        "family": "Asphodelaceae",
        "habitat": "Native to the Arabian Peninsula; grown worldwide in tropical and subtropical climates.",
        "medicinal_uses": ["Soothes burns and wounds", "Treats digestive disorders", "Laxative properties", "Skin moisturiser and anti-ageing", "Reduces blood sugar levels"],
        "active_compounds": ["Aloin", "Acemannan", "Anthraquinones", "Barbaloin", "Emodin"],
        "rasa_guna": {"Rasa (Taste)": "Tikta (Bitter), Kashaya", "Guna (Quality)": "Guru, Snigdha", "Virya (Potency)": "Sheet (Cold)", "Vipaka (Post-digestion)": "Katu (Pungent)"},
        "dosage_form": "Fresh gel: 10–15 ml twice daily. Juice: 20–30 ml before meals.",
        "parts_used": ["Leaf gel", "Latex", "Whole leaf"],
        "precautions": "Avoid during pregnancy. Can cause diarrhoea in excess. Not recommended for children under 12."
    },
    "Amla": {
        "botanical_name": "Phyllanthus emblica",
        "ayurvedic_name": "Amalaki",
        "family": "Phyllanthaceae",
        "habitat": "Tropical and subtropical Asia; widely cultivated in India.",
        "medicinal_uses": ["Potent antioxidant", "Improves immunity", "Hair nourishment and growth", "Manages diabetes", "Improves digestion and liver function"],
        "active_compounds": ["Emblicanin A & B", "Punigluconin", "Vitamin C", "Tannins", "Gallic acid"],
        "rasa_guna": {"Rasa (Taste)": "All six tastes (mainly sour)", "Guna (Quality)": "Laghu, Ruksha", "Virya (Potency)": "Sheet (Cold)", "Vipaka (Post-digestion)": "Madhura (Sweet)"},
        "dosage_form": "Powder: 3–6g with water or honey. Fresh fruit juice: 10–20 ml twice daily.",
        "parts_used": ["Fruit", "Seeds", "Bark"],
        "precautions": "Consult physician if on anticoagulant medications. Excess may cause dryness."
    },
    "Neem": {
        "botanical_name": "Azadirachta indica",
        "ayurvedic_name": "Nimba",
        "family": "Meliaceae",
        "habitat": "Native to the Indian subcontinent; grows in tropical and semi-arid regions.",
        "medicinal_uses": ["Powerful antibacterial and antifungal", "Treats skin diseases", "Blood purifier", "Dental hygiene", "Antidiabetic properties", "Insect repellent"],
        "active_compounds": ["Nimbin", "Nimbidin", "Azadirachtin", "Quercetin", "Limonoids"],
        "rasa_guna": {"Rasa (Taste)": "Tikta (Bitter)", "Guna (Quality)": "Laghu, Ruksha", "Virya (Potency)": "Sheet (Cold)", "Vipaka (Post-digestion)": "Katu (Pungent)"},
        "dosage_form": "Leaf juice: 10–20 ml once daily. Powder: 2–4g with warm water.",
        "parts_used": ["Leaves", "Bark", "Seeds", "Twigs", "Flowers"],
        "precautions": "Not recommended during pregnancy or while trying to conceive. Avoid in very young children."
    },
    "Tulsi": {
        "botanical_name": "Ocimum tenuiflorum",
        "ayurvedic_name": "Tulasi",
        "family": "Lamiaceae",
        "habitat": "Native to tropical Asia; widely cultivated across India as a sacred plant.",
        "medicinal_uses": ["Adaptogen (stress relief)", "Treats respiratory disorders", "Anti-inflammatory", "Improves immunity", "Antimicrobial properties", "Reduces fever"],
        "active_compounds": ["Eugenol", "Ursolic acid", "Rosmarinic acid", "Caryophyllene", "Apigenin"],
        "rasa_guna": {"Rasa (Taste)": "Katu (Pungent), Tikta (Bitter)", "Guna (Quality)": "Laghu, Ruksha, Tikshna", "Virya (Potency)": "Ushna (Hot)", "Vipaka (Post-digestion)": "Katu (Pungent)"},
        "dosage_form": "Leaf juice: 10–20 ml with honey. Tea: 5–10 fresh leaves boiled in water twice daily.",
        "parts_used": ["Leaves", "Seeds", "Roots"],
        "precautions": "Avoid excessive use during pregnancy. May interact with blood-thinning medications."
    },
    "Turmeric": {
        "botanical_name": "Curcuma longa",
        "ayurvedic_name": "Haridra",
        "family": "Zingiberaceae",
        "habitat": "Native to tropical South Asia; widely cultivated throughout India.",
        "medicinal_uses": ["Powerful anti-inflammatory", "Antioxidant properties", "Wound healing", "Supports liver function", "Improves digestion", "Antimicrobial"],
        "active_compounds": ["Curcumin", "Bisdemethoxycurcumin", "Turmerone", "Ar-turmerone", "Curdione"],
        "rasa_guna": {"Rasa (Taste)": "Tikta (Bitter), Katu (Pungent)", "Guna (Quality)": "Laghu, Ruksha", "Virya (Potency)": "Ushna (Hot)", "Vipaka (Post-digestion)": "Katu (Pungent)"},
        "dosage_form": "Powder: 1–3g with warm milk or water. Fresh juice: 10–20 ml daily.",
        "parts_used": ["Rhizome", "Leaves"],
        "precautions": "High doses may cause stomach upset. Avoid in gallbladder disease. Consult before use in pregnancy."
    },
    "Ginger": {
        "botanical_name": "Zingiber officinale",
        "ayurvedic_name": "Shunti / Ardraka",
        "family": "Zingiberaceae",
        "habitat": "Tropical Asia; cultivated throughout India, especially in Kerala and Karnataka.",
        "medicinal_uses": ["Relieves nausea and vomiting", "Anti-inflammatory", "Digestive stimulant", "Reduces pain", "Improves circulation"],
        "active_compounds": ["Gingerol", "Shogaol", "Zingerone", "Zingiberene", "Paradols"],
        "rasa_guna": {"Rasa (Taste)": "Katu (Pungent)", "Guna (Quality)": "Laghu, Snigdha, Tikshna", "Virya (Potency)": "Ushna (Hot)", "Vipaka (Post-digestion)": "Madhura (Sweet)"},
        "dosage_form": "Fresh juice: 5–10 ml with honey. Dry powder: 1–3g with warm water.",
        "parts_used": ["Rhizome"],
        "precautions": "Avoid in high pitta conditions. Not recommended in large doses during pregnancy."
    },
    "Mint": {
        "botanical_name": "Mentha spicata / Mentha piperita",
        "ayurvedic_name": "Pudina",
        "family": "Lamiaceae",
        "habitat": "Temperate regions worldwide; cultivated across India.",
        "medicinal_uses": ["Relieves indigestion and bloating", "Treats headaches", "Cooling effect on body", "Antispasmodic", "Improves breath and oral health"],
        "active_compounds": ["Menthol", "Menthone", "Menthyl acetate", "Rosmarinic acid", "Flavonoids"],
        "rasa_guna": {"Rasa (Taste)": "Katu (Pungent), Tikta (Bitter)", "Guna (Quality)": "Laghu, Ruksha", "Virya (Potency)": "Sheet (Cold)", "Vipaka (Post-digestion)": "Katu (Pungent)"},
        "dosage_form": "Fresh juice: 10–15 ml with water. Tea: 5–10 leaves brewed in hot water.",
        "parts_used": ["Leaves", "Stems"],
        "precautions": "Not suitable for infants. Menthol may cause breathing difficulties in very young children."
    },
    "Curry": {
        "botanical_name": "Murraya koenigii",
        "ayurvedic_name": "Surabhi",
        "family": "Rutaceae",
        "habitat": "Native to India and Sri Lanka; grows in tropical and subtropical regions.",
        "medicinal_uses": ["Treats digestive disorders", "Anti-diabetic properties", "Hair loss prevention", "Antioxidant", "Reduces nausea"],
        "active_compounds": ["Carbazole alkaloids", "Mahanimbine", "Girinimbine", "Murrayamine", "Linalool"],
        "rasa_guna": {"Rasa (Taste)": "Tikta (Bitter), Katu (Pungent)", "Guna (Quality)": "Laghu, Ruksha", "Virya (Potency)": "Ushna (Hot)", "Vipaka (Post-digestion)": "Katu (Pungent)"},
        "dosage_form": "Fresh leaves: 8–10 leaves chewed in morning. Powder: 2–3g with buttermilk.",
        "parts_used": ["Leaves", "Bark", "Roots"],
        "precautions": "Generally safe as a spice. Medicinal doses should be discussed with a practitioner."
    },
    "Guava": {
        "botanical_name": "Psidium guajava",
        "ayurvedic_name": "Amrud / Peru",
        "family": "Myrtaceae",
        "habitat": "Tropical and subtropical regions; widely grown throughout India.",
        "medicinal_uses": ["Antidiarrheal properties", "Rich in Vitamin C", "Controls blood sugar", "Reduces cholesterol", "Anti-inflammatory"],
        "active_compounds": ["Quercetin", "Guajaverin", "Lycopene", "Vitamin C", "Tannins"],
        "rasa_guna": {"Rasa (Taste)": "Kashaya (Astringent), Madhura (Sweet)", "Guna (Quality)": "Guru, Ruksha", "Virya (Potency)": "Sheet (Cold)", "Vipaka (Post-digestion)": "Madhura (Sweet)"},
        "dosage_form": "Leaf decoction: 50 ml twice daily for diarrhoea. Fruit: eaten fresh.",
        "parts_used": ["Leaves", "Fruit", "Bark"],
        "precautions": "Seeds may cause constipation if eaten in excess. Avoid in irritable bowel syndrome."
    },
    "Hibiscus": {
        "botanical_name": "Hibiscus rosa-sinensis",
        "ayurvedic_name": "Japa / China Rose",
        "family": "Malvaceae",
        "habitat": "Tropical and subtropical Asia; widely cultivated across India.",
        "medicinal_uses": ["Lowers blood pressure", "Promotes hair growth", "Liver protection", "Antioxidant", "Treats urinary tract infections"],
        "active_compounds": ["Anthocyanins", "Quercetin", "Chlorogenic acid", "Hibiscetin", "Protocatechuic acid"],
        "rasa_guna": {"Rasa (Taste)": "Madhura (Sweet), Tikta (Bitter)", "Guna (Quality)": "Guru, Snigdha", "Virya (Potency)": "Sheet (Cold)", "Vipaka (Post-digestion)": "Madhura (Sweet)"},
        "dosage_form": "Flower decoction: 50 ml twice daily. Flower paste applied topically for hair care.",
        "parts_used": ["Flowers", "Leaves", "Roots"],
        "precautions": "May lower blood pressure significantly. Avoid concurrent use with antihypertensive drugs."
    },
    "Lemon": {
        "botanical_name": "Citrus limon",
        "ayurvedic_name": "Nimbuka",
        "family": "Rutaceae",
        "habitat": "Native to South Asia; cultivated throughout India.",
        "medicinal_uses": ["Rich source of Vitamin C", "Aids digestion", "Alkalises blood pH", "Prevents kidney stones", "Antibacterial"],
        "active_compounds": ["Vitamin C", "Citric acid", "Limonene", "Flavonoids", "Pectin"],
        "rasa_guna": {"Rasa (Taste)": "Amla (Sour)", "Guna (Quality)": "Laghu, Snigdha", "Virya (Potency)": "Ushna (Hot)", "Vipaka (Post-digestion)": "Amla (Sour)"},
        "dosage_form": "Fresh juice: 20–30 ml with warm water in the morning.",
        "parts_used": ["Fruit", "Juice", "Peel"],
        "precautions": "Avoid on empty stomach if prone to acidity. Excess can erode tooth enamel."
    },
    "Mango": {
        "botanical_name": "Mangifera indica",
        "ayurvedic_name": "Amra",
        "family": "Anacardiaceae",
        "habitat": "Native to India; grown throughout tropical regions.",
        "medicinal_uses": ["Digestive tonic", "Rich in vitamins A, C, E", "Antioxidant", "Boosts immunity", "Supports eye health"],
        "active_compounds": ["Mangiferin", "Quercetin", "Beta-carotene", "Gallic acid", "Vitamin C"],
        "rasa_guna": {"Rasa (Taste)": "Madhura (Sweet), Amla (Sour)", "Guna (Quality)": "Guru, Snigdha", "Virya (Potency)": "Ushna (Hot)", "Vipaka (Post-digestion)": "Madhura (Sweet)"},
        "dosage_form": "Leaf powder: 2–3g with water. Bark decoction: 50 ml for diarrhoea.",
        "parts_used": ["Fruit", "Leaves", "Bark", "Seeds"],
        "precautions": "Unripe mango in excess may cause throat irritation and indigestion."
    },
    "Papaya": {
        "botanical_name": "Carica papaya",
        "ayurvedic_name": "Papita / Erand Karkati",
        "family": "Caricaceae",
        "habitat": "Tropical regions; widely grown throughout India.",
        "medicinal_uses": ["Digestive enzyme source", "Treats dengue fever (leaf extract)", "Anti-inflammatory", "Wound healing", "Anthelmintic"],
        "active_compounds": ["Papain", "Chymopapain", "Lycopene", "Beta-carotene", "Carpaine"],
        "rasa_guna": {"Rasa (Taste)": "Madhura (Sweet), Tikta (Bitter)", "Guna (Quality)": "Laghu, Ruksha", "Virya (Potency)": "Ushna (Hot)", "Vipaka (Post-digestion)": "Katu (Pungent)"},
        "dosage_form": "Leaf juice: 10–20 ml twice daily. Ripe fruit: eaten fresh.",
        "parts_used": ["Fruit", "Leaves", "Seeds", "Latex"],
        "precautions": "Avoid raw papaya and seeds during pregnancy (abortifacient). Latex may cause allergic reactions."
    },
    "Jasmine": {
        "botanical_name": "Jasminum officinale",
        "ayurvedic_name": "Jati / Mallika",
        "family": "Oleaceae",
        "habitat": "Native to South and West Asia; widely cultivated in India.",
        "medicinal_uses": ["Antidepressant (aromatherapy)", "Improves skin health", "Antiseptic", "Reduces anxiety", "Enhances libido"],
        "active_compounds": ["Benzyl acetate", "Linalool", "Methyl jasmonate", "Jasmonates", "Indole"],
        "rasa_guna": {"Rasa (Taste)": "Tikta (Bitter), Katu (Pungent)", "Guna (Quality)": "Laghu, Ruksha", "Virya (Potency)": "Ushna (Hot)", "Vipaka (Post-digestion)": "Katu (Pungent)"},
        "dosage_form": "Flower infusion: as tea or aromatherapy. Leaf paste applied topically.",
        "parts_used": ["Flowers", "Leaves"],
        "precautions": "Essential oil should be diluted before topical use. Not recommended in large quantities internally."
    },
    "Drumstick": {
        "botanical_name": "Moringa oleifera",
        "ayurvedic_name": "Shigru / Sahijana",
        "family": "Moringaceae",
        "habitat": "Native to the sub-Himalayan tracts; cultivated across India.",
        "medicinal_uses": ["Highly nutritious superfood", "Lowers blood sugar", "Reduces inflammation", "Lowers cholesterol", "Boosts immunity"],
        "active_compounds": ["Isothiocyanates", "Moringin", "Beta-carotene", "Quercetin", "Chlorogenic acid"],
        "rasa_guna": {"Rasa (Taste)": "Katu (Pungent), Tikta (Bitter)", "Guna (Quality)": "Laghu, Ruksha, Tikshna", "Virya (Potency)": "Ushna (Hot)", "Vipaka (Post-digestion)": "Katu (Pungent)"},
        "dosage_form": "Leaf powder: 2–5g daily. Fresh leaves in food. Pod vegetable: eaten cooked.",
        "parts_used": ["Leaves", "Pods", "Seeds", "Roots"],
        "precautions": "Root bark may be toxic in large amounts. Avoid during pregnancy (root/bark extracts)."
    },
    "Tamarind": {
        "botanical_name": "Tamarindus indica",
        "ayurvedic_name": "Amli / Chincha",
        "family": "Fabaceae",
        "habitat": "Native to tropical Africa; naturalised throughout India.",
        "medicinal_uses": ["Treats constipation", "Bile stimulant", "Antioxidant", "Cooling in fever", "Anti-inflammatory"],
        "active_compounds": ["Tartaric acid", "Malic acid", "Potassium bitartrate", "Lupeol", "Catechins"],
        "rasa_guna": {"Rasa (Taste)": "Amla (Sour)", "Guna (Quality)": "Guru, Snigdha", "Virya (Potency)": "Ushna (Hot)", "Vipaka (Post-digestion)": "Amla (Sour)"},
        "dosage_form": "Pulp: 10–15g soaked in water. Leaf decoction: 50 ml for fever.",
        "parts_used": ["Fruit pulp", "Leaves", "Seeds", "Bark"],
        "precautions": "Avoid in excess if prone to acidity. May interfere with aspirin absorption."
    },
    "Tomato": {
        "botanical_name": "Solanum lycopersicum",
        "ayurvedic_name": "Tamatar",
        "family": "Solanaceae",
        "habitat": "Originally from South America; widely cultivated throughout India.",
        "medicinal_uses": ["Rich in lycopene (anti-cancer)", "Cardiovascular health", "Antioxidant", "Improves skin", "Bone health"],
        "active_compounds": ["Lycopene", "Beta-carotene", "Vitamin C", "Naringenin", "Chlorogenic acid"],
        "rasa_guna": {"Rasa (Taste)": "Amla (Sour), Madhura (Sweet)", "Guna (Quality)": "Laghu, Ruksha", "Virya (Potency)": "Ushna (Hot)", "Vipaka (Post-digestion)": "Amla (Sour)"},
        "dosage_form": "Fresh fruit: consumed in diet. Juice: 100–200 ml daily.",
        "parts_used": ["Fruit"],
        "precautions": "Avoid in excess if you have arthritis (nightshade family). Unripe fruit contains solanine."
    },
    "Coriender": {
        "botanical_name": "Coriandrum sativum",
        "ayurvedic_name": "Dhanyaka / Dhaniya",
        "family": "Apiaceae",
        "habitat": "Cultivated throughout India as a culinary herb.",
        "medicinal_uses": ["Digestive stimulant", "Reduces blood sugar", "Anti-inflammatory", "Antibacterial", "Lowers cholesterol"],
        "active_compounds": ["Linalool", "Borneol", "Geraniol", "Quercetin", "Rutin"],
        "rasa_guna": {"Rasa (Taste)": "Tikta (Bitter), Katu (Pungent)", "Guna (Quality)": "Laghu, Snigdha", "Virya (Potency)": "Sheet (Cold)", "Vipaka (Post-digestion)": "Madhura (Sweet)"},
        "dosage_form": "Seed decoction: 50 ml twice daily. Fresh leaf juice: 10–15 ml.",
        "parts_used": ["Leaves", "Seeds"],
        "precautions": "Allergy rare but possible. Excessive use may lower blood sugar too much in diabetics on medication."
    },
    "Lemongrass": {
        "botanical_name": "Cymbopogon citratus",
        "ayurvedic_name": "Bhustrina",
        "family": "Poaceae",
        "habitat": "Tropical and subtropical regions; cultivated in India for essential oil.",
        "medicinal_uses": ["Reduces anxiety", "Lowers blood pressure", "Antimicrobial", "Analgesic", "Digestive tonic"],
        "active_compounds": ["Citral", "Geraniol", "Limonene", "Myrcene", "Linalool"],
        "rasa_guna": {"Rasa (Taste)": "Katu (Pungent), Tikta (Bitter)", "Guna (Quality)": "Laghu, Ruksha", "Virya (Potency)": "Ushna (Hot)", "Vipaka (Post-digestion)": "Katu (Pungent)"},
        "dosage_form": "Tea: 2–3 stalks brewed in hot water. Essential oil: diluted for topical use.",
        "parts_used": ["Stems", "Leaves"],
        "precautions": "Essential oil should not be taken internally. Avoid during pregnancy in medicinal doses."
    },
    "Eucalyptus": {
        "botanical_name": "Eucalyptus globulus",
        "ayurvedic_name": "Tailaparna",
        "family": "Myrtaceae",
        "habitat": "Native to Australia; widely planted in India for timber and essential oil.",
        "medicinal_uses": ["Treats respiratory conditions", "Decongestant", "Antiseptic", "Insect repellent", "Pain relief"],
        "active_compounds": ["1,8-Cineole (Eucalyptol)", "Alpha-pinene", "Limonene", "Terpineol", "Globulol"],
        "rasa_guna": {"Rasa (Taste)": "Katu (Pungent), Tikta (Bitter)", "Guna (Quality)": "Laghu, Ruksha, Tikshna", "Virya (Potency)": "Ushna (Hot)", "Vipaka (Post-digestion)": "Katu (Pungent)"},
        "dosage_form": "Steam inhalation: 3–5 drops oil in hot water. Topical oil: diluted with carrier oil.",
        "parts_used": ["Leaves", "Essential oil"],
        "precautions": "Internal use of essential oil is toxic. Not suitable for young children. Avoid near face of infants."
    },
    "Rose": {
        "botanical_name": "Rosa damascena / Rosa indica",
        "ayurvedic_name": "Taruni / Shatapatra",
        "family": "Rosaceae",
        "habitat": "Temperate regions; cultivated throughout India.",
        "medicinal_uses": ["Treats skin conditions", "Anti-inflammatory", "Stress reduction", "Digestive tonic", "Antimicrobial"],
        "active_compounds": ["Citronellol", "Geraniol", "Nerol", "Kaempferol", "Quercetin"],
        "rasa_guna": {"Rasa (Taste)": "Madhura (Sweet), Kashaya (Astringent)", "Guna (Quality)": "Guru, Snigdha", "Virya (Potency)": "Sheet (Cold)", "Vipaka (Post-digestion)": "Madhura (Sweet)"},
        "dosage_form": "Rose water: used topically. Petal jam (Gulkand): 5–10g daily. Infusion: as tea.",
        "parts_used": ["Petals", "Hips", "Leaves"],
        "precautions": "Allergy possible in sensitive individuals. Avoid wilted or chemically treated flowers."
    },
    "Jackfruit": {
        "botanical_name": "Artocarpus heterophyllus",
        "ayurvedic_name": "Panasa",
        "family": "Moraceae",
        "habitat": "Native to the Western Ghats; grown throughout tropical India.",
        "medicinal_uses": ["Boosts immunity", "Improves digestion", "Anti-ulcer properties", "Antioxidant", "Controls blood pressure"],
        "active_compounds": ["Artocarpin", "Morusin", "Norartocarpin", "Vitamin C", "Flavonoids"],
        "rasa_guna": {"Rasa (Taste)": "Madhura (Sweet)", "Guna (Quality)": "Guru, Snigdha", "Virya (Potency)": "Ushna (Hot)", "Vipaka (Post-digestion)": "Madhura (Sweet)"},
        "dosage_form": "Leaf decoction: 50 ml twice daily. Fruit: eaten fresh or cooked.",
        "parts_used": ["Fruit", "Seeds", "Leaves", "Bark"],
        "precautions": "Excess consumption may cause digestive discomfort. Avoid if allergic to latex (cross-reactivity possible)."
    },
    "Castor": {
        "botanical_name": "Ricinus communis",
        "ayurvedic_name": "Eranda",
        "family": "Euphorbiaceae",
        "habitat": "Tropical and subtropical regions; widely grown in India.",
        "medicinal_uses": ["Laxative (castor oil)", "Anti-inflammatory", "Skin conditions", "Arthritis pain relief", "Induces labour (under supervision)"],
        "active_compounds": ["Ricinoleic acid", "Ricin (toxic in seeds)", "Undecylenic acid", "Tocopherols", "Flavonoids"],
        "rasa_guna": {"Rasa (Taste)": "Madhura (Sweet), Katu (Pungent)", "Guna (Quality)": "Guru, Snigdha, Tikshna", "Virya (Potency)": "Ushna (Hot)", "Vipaka (Post-digestion)": "Madhura (Sweet)"},
        "dosage_form": "Castor oil: 10–15 ml at bedtime as laxative. Leaf poultice for topical use.",
        "parts_used": ["Seeds (oil)", "Leaves", "Roots"],
        "precautions": "Seeds are extremely toxic (contain ricin). Only use processed castor oil. Not for internal use during pregnancy."
    },
    "Betel": {
        "botanical_name": "Piper betle",
        "ayurvedic_name": "Nagavalli / Tambula",
        "family": "Piperaceae",
        "habitat": "Tropical Asia; cultivated throughout India.",
        "medicinal_uses": ["Digestive stimulant", "Antiseptic", "Treats mouth ulcers", "Anti-inflammatory", "Stimulates salivation"],
        "active_compounds": ["Chavicol", "Eugenol", "Beta-sitosterol", "Tannins", "Safrole"],
        "rasa_guna": {"Rasa (Taste)": "Katu (Pungent), Tikta (Bitter)", "Guna (Quality)": "Laghu, Ruksha, Tikshna", "Virya (Potency)": "Ushna (Hot)", "Vipaka (Post-digestion)": "Katu (Pungent)"},
        "dosage_form": "Fresh leaf paste: applied externally. Leaf juice: 5–10 ml for mouth conditions.",
        "parts_used": ["Leaves"],
        "precautions": "Do NOT chew with tobacco or areca nut — carcinogenic combination. Leaf alone is medicinal in limited quantities."
    },
    "Catharanthus": {
        "botanical_name": "Catharanthus roseus",
        "ayurvedic_name": "Sadabahar / Nityakalyani",
        "family": "Apocynaceae",
        "habitat": "Native to Madagascar; naturalised throughout India.",
        "medicinal_uses": ["Anti-diabetic properties", "Cancer treatment (alkaloids)", "Lowers blood pressure", "Wound healing", "Antibacterial"],
        "active_compounds": ["Vincristine", "Vinblastine", "Catharanthine", "Vindoline", "Ajmalicine"],
        "rasa_guna": {"Rasa (Taste)": "Tikta (Bitter), Kashaya (Astringent)", "Guna (Quality)": "Laghu, Ruksha", "Virya (Potency)": "Ushna (Hot)", "Vipaka (Post-digestion)": "Katu (Pungent)"},
        "dosage_form": "Leaf extract: only under medical supervision. Folk use: 5–10 flowers in water for diabetes.",
        "parts_used": ["Leaves", "Flowers", "Roots"],
        "precautions": "Highly toxic in large doses. Pharmaceutical alkaloids (vincristine/vinblastine) only used medically. Do NOT self-medicate."
    },
    "Henna": {
        "botanical_name": "Lawsonia inermis",
        "ayurvedic_name": "Mehndi / Madayantika",
        "family": "Lythraceae",
        "habitat": "Arid and semi-arid regions; cultivated throughout India.",
        "medicinal_uses": ["Hair conditioning and colouring", "Treats skin infections", "Reduces fever", "Anti-inflammatory", "Headache relief"],
        "active_compounds": ["Lawsone", "Gallic acid", "Glucose", "Mannitol", "Tannins"],
        "rasa_guna": {"Rasa (Taste)": "Tikta (Bitter), Kashaya (Astringent)", "Guna (Quality)": "Laghu, Ruksha", "Virya (Potency)": "Sheet (Cold)", "Vipaka (Post-digestion)": "Katu (Pungent)"},
        "dosage_form": "Leaf paste applied topically. Powder in hair conditioning packs.",
        "parts_used": ["Leaves", "Seeds", "Bark"],
        "precautions": "Black henna may contain PPD (para-phenylenediamine) — causes severe allergic reactions. Use only natural henna."
    },
    "Marigold": {
        "botanical_name": "Tagetes erecta / Calendula officinalis",
        "ayurvedic_name": "Sthulapushpa / Gendha",
        "family": "Asteraceae",
        "habitat": "Tropical America; widely cultivated throughout India.",
        "medicinal_uses": ["Wound healing", "Anti-inflammatory", "Antifungal", "Eye health", "Antiseptic"],
        "active_compounds": ["Lutein", "Zeaxanthin", "Quercetin", "Isorhamnetin", "Terpenoids"],
        "rasa_guna": {"Rasa (Taste)": "Tikta (Bitter), Katu (Pungent)", "Guna (Quality)": "Laghu, Ruksha", "Virya (Potency)": "Ushna (Hot)", "Vipaka (Post-digestion)": "Katu (Pungent)"},
        "dosage_form": "Flower petal infusion: 50 ml daily. Petal cream applied topically.",
        "parts_used": ["Flowers", "Leaves"],
        "precautions": "May cause allergic contact dermatitis in Asteraceae-sensitive individuals."
    },
    "Bamboo": {
        "botanical_name": "Bambusa vulgaris / Dendrocalamus sp.",
        "ayurvedic_name": "Vanshalochan / Tvak",
        "family": "Poaceae",
        "habitat": "Tropical and subtropical Asia; widely found in India.",
        "medicinal_uses": ["Respiratory tonic", "Treats bleeding disorders", "Rich in silica for bone health", "Anti-ulcer", "Cooling properties"],
        "active_compounds": ["Silica", "Bamboo silica (Tabasheer)", "Flavonoids", "Chlorophyll", "Lignin"],
        "rasa_guna": {"Rasa (Taste)": "Madhura (Sweet), Kashaya (Astringent)", "Guna (Quality)": "Laghu, Ruksha", "Virya (Potency)": "Sheet (Cold)", "Vipaka (Post-digestion)": "Madhura (Sweet)"},
        "dosage_form": "Bamboo manna (Tabasheer): 1–2g with honey. Young shoots: eaten as vegetable.",
        "parts_used": ["Young shoots", "Internodal silica (Tabasheer)", "Leaves"],
        "precautions": "Some species contain taxiphyllin (cyanogenic glycoside) in young shoots — must be cooked before eating."
    },
    "Pepper": {
        "botanical_name": "Piper nigrum",
        "ayurvedic_name": "Maricha",
        "family": "Piperaceae",
        "habitat": "Native to Kerala; cultivated in tropical India.",
        "medicinal_uses": ["Digestive stimulant", "Improves bioavailability of nutrients", "Anti-inflammatory", "Antibacterial", "Expectorant"],
        "active_compounds": ["Piperine", "Piperic acid", "Safrole", "Beta-caryophyllene", "Linalool"],
        "rasa_guna": {"Rasa (Taste)": "Katu (Pungent)", "Guna (Quality)": "Laghu, Ruksha, Tikshna", "Virya (Potency)": "Ushna (Hot)", "Vipaka (Post-digestion)": "Katu (Pungent)"},
        "dosage_form": "Powder: 500mg–1g with honey or milk. Added to food as spice.",
        "parts_used": ["Fruit (peppercorns)", "Leaves"],
        "precautions": "Avoid large doses in gastritis or ulcer patients. May interact with certain drugs by increasing their absorption."
    },
    "Coffee": {
        "botanical_name": "Coffea arabica",
        "ayurvedic_name": "Kapi",
        "family": "Rubiaceae",
        "habitat": "Native to Ethiopia; major cultivation in Karnataka, Kerala, Tamil Nadu.",
        "medicinal_uses": ["CNS stimulant", "Reduces fatigue", "Antioxidant properties", "Improves cognitive function", "Reduces risk of Parkinson's disease"],
        "active_compounds": ["Caffeine", "Chlorogenic acid", "Cafestol", "Kahweol", "Diterpenes"],
        "rasa_guna": {"Rasa (Taste)": "Tikta (Bitter)", "Guna (Quality)": "Laghu, Ruksha, Tikshna", "Virya (Potency)": "Ushna (Hot)", "Vipaka (Post-digestion)": "Katu (Pungent)"},
        "dosage_form": "Moderate consumption: 1–3 cups daily. Green coffee extract: 200–400 mg daily.",
        "parts_used": ["Seeds (beans)", "Leaves"],
        "precautions": "Excessive consumption causes anxiety, insomnia, palpitations. Avoid in pregnancy and hypertension."
    },
    "Pumpkin": {
        "botanical_name": "Cucurbita pepo",
        "ayurvedic_name": "Kushmanda",
        "family": "Cucurbitaceae",
        "habitat": "Widely cultivated throughout India as a vegetable.",
        "medicinal_uses": ["Antiulcer properties", "Diuretic", "Antidepressant (seeds)", "Anthelmintic", "Nutritive tonic"],
        "active_compounds": ["Cucurbitacins", "Beta-carotene", "Zinc", "Vitamin E", "Lignans"],
        "rasa_guna": {"Rasa (Taste)": "Madhura (Sweet)", "Guna (Quality)": "Guru, Snigdha", "Virya (Potency)": "Sheet (Cold)", "Vipaka (Post-digestion)": "Madhura (Sweet)"},
        "dosage_form": "Fresh juice: 50–100 ml daily. Seeds: 30g for deworming. Vegetable cooked.",
        "parts_used": ["Fruit", "Seeds", "Leaves", "Flowers"],
        "precautions": "Excessive use of seeds may cause stomach discomfort. Cucurbitacin content may be toxic in wild varieties."
    },
    "Onion": {
        "botanical_name": "Allium cepa",
        "ayurvedic_name": "Palandu",
        "family": "Amaryllidaceae",
        "habitat": "Cultivated throughout India as a staple vegetable.",
        "medicinal_uses": ["Antibacterial", "Reduces blood sugar", "Improves heart health", "Antiparasitic", "Treats cough and cold"],
        "active_compounds": ["Quercetin", "Allicin", "Diallyl disulfide", "Fisetin", "Chromium"],
        "rasa_guna": {"Rasa (Taste)": "Katu (Pungent)", "Guna (Quality)": "Guru, Snigdha, Tikshna", "Virya (Potency)": "Ushna (Hot)", "Vipaka (Post-digestion)": "Katu (Pungent)"},
        "dosage_form": "Raw juice: 10–20 ml. Fresh onion in daily diet. Roasted onion for earache.",
        "parts_used": ["Bulb", "Leaves"],
        "precautions": "Raw onion may cause heartburn. May enhance the effect of blood-thinning medications."
    },
    "Pomegranate": {
        "botanical_name": "Punica granatum",
        "ayurvedic_name": "Dadima",
        "family": "Lythraceae",
        "habitat": "Native to Iran and northern India; widely cultivated in Rajasthan, Maharashtra, Gujarat.",
        "medicinal_uses": ["Powerful antioxidant", "Heart health", "Anti-inflammatory", "Anti-cancer properties", "Treats anaemia"],
        "active_compounds": ["Punicalagin", "Ellagic acid", "Anthocyanins", "Punicic acid", "Quercetin"],
        "rasa_guna": {"Rasa (Taste)": "Madhura (Sweet), Amla (Sour), Kashaya (Astringent)", "Guna (Quality)": "Laghu, Snigdha", "Virya (Potency)": "Sheet (Cold)", "Vipaka (Post-digestion)": "Madhura (Sweet)"},
        "dosage_form": "Fresh juice: 100–200 ml daily. Fruit: eaten fresh. Bark decoction: for parasites.",
        "parts_used": ["Fruit", "Peel", "Seeds", "Bark"],
        "precautions": "Pomegranate juice may interact with certain medications (similar to grapefruit). Consult doctor if on prescription drugs."
    },
    "Pea": {
        "botanical_name": "Pisum sativum",
        "ayurvedic_name": "Harit Shimbhi",
        "family": "Fabaceae",
        "habitat": "Temperate regions; cultivated throughout India.",
        "medicinal_uses": ["Rich in protein and fibre", "Supports digestion", "Lowers cholesterol", "Manages blood sugar", "Promotes bone health"],
        "active_compounds": ["Pisumsaponins", "Coumestrol", "Ferulic acid", "Carotenoids", "Vitamin K"],
        "rasa_guna": {"Rasa (Taste)": "Madhura (Sweet)", "Guna (Quality)": "Guru, Snigdha", "Virya (Potency)": "Sheet (Cold)", "Vipaka (Post-digestion)": "Madhura (Sweet)"},
        "dosage_form": "Fresh peas in diet. Dried pea flour: 20–30g in preparations.",
        "parts_used": ["Seeds", "Pods", "Leaves"],
        "precautions": "May cause bloating and gas in some individuals. Avoid excess in those with uric acid/gout issues."
    },
    "Insulin": {
        "botanical_name": "Costus igneus",
        "ayurvedic_name": "Insulin plant / Keukand",
        "family": "Costaceae",
        "habitat": "Native to tropical America; cultivated in South India.",
        "medicinal_uses": ["Manages type 2 diabetes", "Lowers blood glucose", "Antioxidant", "Kidney protection", "Anti-inflammatory"],
        "active_compounds": ["Diosgenin", "Corosolic acid", "Quercetin", "Kaempferol", "Luteolin"],
        "rasa_guna": {"Rasa (Taste)": "Tikta (Bitter), Kashaya (Astringent)", "Guna (Quality)": "Laghu, Ruksha", "Virya (Potency)": "Sheet (Cold)", "Vipaka (Post-digestion)": "Katu (Pungent)"},
        "dosage_form": "Fresh leaf: chew 1–2 leaves before meals. Leaf juice: 10–20 ml twice daily.",
        "parts_used": ["Leaves"],
        "precautions": "Do not stop prescribed diabetes medications without doctor's advice. Monitor blood sugar carefully."
    },
    "Malabar_Nut": {
        "botanical_name": "Justicia adhatoda",
        "ayurvedic_name": "Vasa / Adusa",
        "family": "Acanthaceae",
        "habitat": "Sub-Himalayan tracts and throughout India's plains.",
        "medicinal_uses": ["Treats respiratory diseases (asthma, bronchitis)", "Expectorant", "Antispasmodic", "Antitubercular", "Hemostatic"],
        "active_compounds": ["Vasicine", "Vasicinone", "Adhatodic acid", "Beta-sitosterol", "Quinazoline alkaloids"],
        "rasa_guna": {"Rasa (Taste)": "Tikta (Bitter), Kashaya (Astringent)", "Guna (Quality)": "Laghu, Ruksha", "Virya (Potency)": "Sheet (Cold)", "Vipaka (Post-digestion)": "Katu (Pungent)"},
        "dosage_form": "Leaf decoction: 50 ml 2–3 times daily. Leaf juice: 10–15 ml with honey.",
        "parts_used": ["Leaves", "Roots", "Flowers"],
        "precautions": "Can cause abortion — STRICTLY avoid during pregnancy. Vasicine is a known uterine stimulant."
    },
}

# Generate generic data for plants not in the detailed DB
def _generic_plant_info(name: str) -> dict:
    clean = name.replace("_", " ").replace("1", "")
    return {
        "botanical_name": f"{clean} sp.",
        "ayurvedic_name": clean,
        "family": "Medicinal Plant",
        "habitat": f"{clean} is found in tropical and subtropical regions of India, commonly growing in gardens, forests, and cultivated fields.",
        "medicinal_uses": [
            f"Traditional Ayurvedic medicine uses {clean} as a healing herb",
            "Anti-inflammatory and antioxidant properties",
            "Supports digestive health",
            "Boosts immune function",
            "Used in traditional formulations for general wellness"
        ],
        "active_compounds": ["Flavonoids", "Tannins", "Alkaloids", "Terpenoids", "Phenolic compounds"],
        "rasa_guna": {
            "Rasa (Taste)": "Tikta (Bitter), Kashaya (Astringent)",
            "Guna (Quality)": "Laghu, Ruksha",
            "Virya (Potency)": "Ushna (Hot)",
            "Vipaka (Post-digestion)": "Katu (Pungent)"
        },
        "dosage_form": f"Traditional preparations of {clean} include decoctions (50 ml twice daily), powders (2–5g), and fresh juice (10–20 ml). Consult an Ayurvedic practitioner for personalised dosage.",
        "parts_used": ["Leaves", "Roots", "Bark"],
        "precautions": f"Consult a qualified Ayurvedic practitioner before using {clean} medicinally. Keep out of reach of children. Avoid during pregnancy unless advised by a physician."
    }


def get_plant_info(name: str) -> dict:
    """Return plant info from DB or generate generic entry."""
    # Normalise lookup
    for key in PLANT_DB:
        if key.lower() == name.lower().replace(" ", "_"):
            return PLANT_DB[key]
    # Also try direct match
    if name in PLANT_DB:
        return PLANT_DB[name]
    return _generic_plant_info(name)


# ─── Load Model + Labels ───────────────────────────────────────────────────────
print(f"[VrukshaVed] Loading model from {MODEL_PATH} on {DEVICE}...")

try:
    from huggingface_hub import hf_hub_download
    HAS_HF_HUB = True
except ImportError:
    HAS_HF_HUB = False

DEFAULT_HF_REPOS = [
    os.getenv("HF_MODEL_REPO", ""),
    "omkhk/vrukshaved-convnext",
    "omkhk/vrukshaved-ayurvedic-plant-convnext",
]
DEFAULT_HF_REPOS = [r for r in DEFAULT_HF_REPOS if r]

def is_valid_weight_file(path: Path) -> bool:
    return path.exists() and path.stat().st_size > 1_000_000

resolved_model_path = MODEL_PATH
resolved_labels_path = LABELS_PATH

if not is_valid_weight_file(MODEL_PATH) and HAS_HF_HUB:
    print("[VrukshaVed] NOTICE: Local model missing or LFS pointer. Attempting HF Hub download...", file=sys.stderr)
    for repo_id in DEFAULT_HF_REPOS:
        try:
            dl_path = hf_hub_download(repo_id=repo_id, filename="leaf80_best_convnext_small.pth")
            resolved_model_path = Path(dl_path)
            print(f"[VrukshaVed] SUCCESS: Downloaded model from HF Hub: {repo_id}")
            break
        except Exception as err:
            print(f"[VrukshaVed] Could not download model from {repo_id}: {err}", file=sys.stderr)

if not LABELS_PATH.exists() and HAS_HF_HUB:
    for repo_id in DEFAULT_HF_REPOS:
        try:
            dl_path = hf_hub_download(repo_id=repo_id, filename="leaf80_label_to_class.json")
            resolved_labels_path = Path(dl_path)
            print(f"[VrukshaVed] SUCCESS: Downloaded labels from HF Hub: {repo_id}")
            break
        except Exception:
            pass

if resolved_labels_path.exists():
    try:
        with open(resolved_labels_path, encoding="utf-8") as f:
            label_to_class: dict = json.load(f)
        CLASS_NAMES = [label_to_class[str(i)] for i in range(NUM_CLASSES)]
    except Exception as e:
        print(f"[ERROR] Could not parse labels JSON: {e}", file=sys.stderr)
        CLASS_NAMES = [f"Class_{i}" for i in range(NUM_CLASSES)]
else:
    CLASS_NAMES = [f"Class_{i}" for i in range(NUM_CLASSES)]

model = None
MODEL_LOADED = False

try:
    if not is_valid_weight_file(resolved_model_path):
        raise FileNotFoundError(f"Model file invalid or missing at {resolved_model_path}")

    _model = models.convnext_small(weights=None)
    in_f = _model.classifier[2].in_features
    _model.classifier[2] = nn.Sequential(
        nn.Dropout(p=DROPOUT_P),
        nn.Linear(in_f, NUM_CLASSES)
    )
    _model.load_state_dict(torch.load(resolved_model_path, map_location=DEVICE))
    _model = _model.to(DEVICE).eval()
    model = _model
    MODEL_LOADED = True
except Exception as e:
    print(f"[VrukshaVed] WARNING: Model NOT loaded: {e}", file=sys.stderr)
    print("[VrukshaVed] Server will start in DEMO mode.", file=sys.stderr)

if MODEL_LOADED:
    print(f"[VrukshaVed] SUCCESS: Model loaded. Running on {DEVICE}.")




# ─── Predict Function ──────────────────────────────────────────────────────────
def predict_leaf(pil_image: Image.Image, threshold: float = CONFIDENCE_THRESH, top_k: int = 5) -> dict:
    if not MODEL_LOADED or model is None:
        # Demo mode — return plausible fake results
        import random
        random.seed(42)
        picks = random.sample(range(NUM_CLASSES), top_k)
        weights = sorted([random.uniform(0.05, 0.85) for _ in range(top_k)], reverse=True)
        total = sum(weights)
        weights = [w / total for w in weights]
        top_k_list = [(CLASS_NAMES[picks[i]], weights[i]) for i in range(top_k)]
        best_cls, best_conf = top_k_list[0]
        return {
            "predicted": best_cls if best_conf >= threshold else "Unknown",
            "confidence": best_conf,
            "top_k": top_k_list,
            "status": "ok" if best_conf >= threshold else "low_confidence",
            "demo_mode": True,
            "message": f"DEMO MODE — place real model in model/ folder"
        }

    img = pil_image.convert("RGB")
    prob_sum = np.zeros(NUM_CLASSES)

    with torch.no_grad():
        for tfm in TTA_TRANSFORMS:
            t = tfm(img).unsqueeze(0).to(DEVICE)
            prob_sum += torch.softmax(model(t), dim=1).cpu().numpy()[0]

    avg = prob_sum / len(TTA_TRANSFORMS)
    top_i = avg.argsort()[::-1][:top_k]
    top_k_list = [(CLASS_NAMES[i], float(avg[i])) for i in top_i]
    best_cls, best_conf = top_k_list[0]

    if best_conf < threshold:
        return {
            "predicted": "Unknown",
            "confidence": best_conf,
            "top_k": top_k_list,
            "status": "low_confidence",
            "demo_mode": False,
            "message": f"Best='{best_cls}' ({best_conf*100:.1f}%) < {threshold*100:.0f}% threshold"
        }

    return {
        "predicted": best_cls,
        "confidence": best_conf,
        "top_k": top_k_list,
        "status": "ok",
        "demo_mode": False,
        "message": f"{best_cls} ({best_conf*100:.1f}%)"
    }


# ─── FastAPI App ───────────────────────────────────────────────────────────────
app = FastAPI(title="VrukshaVed — Ayurvedic Plant Intelligence", version="3.0")

app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_methods=["*"],
    allow_headers=["*"],
)

app.mount("/static", StaticFiles(directory=BASE_DIR / "static"), name="static")
templates = Jinja2Templates(directory=BASE_DIR / "templates")


# ─── Routes ───────────────────────────────────────────────────────────────────
@app.get("/")
async def index(request: Request):
    return templates.TemplateResponse(request, "index.html")


@app.post("/api/predict")
async def predict(file: UploadFile = File(...)):
    """Main prediction endpoint. Accepts image file, returns prediction + plant info."""
    if not file.content_type or not file.content_type.startswith("image/"):
        raise HTTPException(status_code=400, detail="File must be an image (JPG, PNG, WEBP).")

    contents = await file.read()
    if len(contents) > 15 * 1024 * 1024:
        raise HTTPException(status_code=413, detail="Image too large. Max 15 MB.")

    try:
        img = Image.open(io.BytesIO(contents)).convert("RGB")
    except Exception:
        raise HTTPException(status_code=400, detail="Could not open image. Please upload a valid image file.")

    result = predict_leaf(img)

    plant_name = result["predicted"]
    plant_info = get_plant_info(plant_name) if plant_name != "Unknown" else {}

    return JSONResponse({
        "status": result["status"],
        "demo_mode": result.get("demo_mode", False),
        "top_prediction": {
            "plant_name": plant_name,
            "confidence": result["confidence"],
            "confidence_pct": f"{result['confidence']*100:.1f}%",
        },
        "all_predictions": [
            {"plant": name, "confidence": conf}
            for name, conf in result["top_k"]
        ],
        "plant_info": plant_info,
        "tta_passes": len(TTA_TRANSFORMS),
        "threshold": CONFIDENCE_THRESH,
        "model": MODEL_NAME,
    })


@app.get("/api/plants")
async def list_plants():
    """List all 80 supported plant species."""
    return JSONResponse({
        "total": len(CLASS_NAMES),
        "plants": sorted(CLASS_NAMES),
        "model_loaded": MODEL_LOADED,
    })


@app.get("/api/plant/{name}")
async def plant_detail(name: str):
    """Get detailed information for a specific plant."""
    info = get_plant_info(name)
    return JSONResponse({
        "found": True,
        "name": name,
        "info": info,
    })


@app.get("/api/health")
async def health():
    """Health check endpoint."""
    return JSONResponse({
        "status": "ok",
        "model_loaded": MODEL_LOADED,
        "device": str(DEVICE),
        "num_classes": NUM_CLASSES,
        "tta_passes": len(TTA_TRANSFORMS),
        "confidence_threshold": CONFIDENCE_THRESH,
    })


# ─── Entry Point ───────────────────────────────────────────────────────────────
if __name__ == "__main__":
    import uvicorn
    port = int(os.getenv("PORT", "7860"))
    uvicorn.run("app:app", host="0.0.0.0", port=port, reload=False)