Update app.py

app.py

@@ -109,15 +109,15 @@ RECOMMENDATIONS = {
     # BACTERIAL DISEASES (2)
     # =========================================================================
     'bacterial_leaf_blight': {
-        'recommendation': 'Use resistant varieties like PSB Rc18 (ALA), NSIC Rc354, or NSIC Rc302. Apply balanced nutrients, especially nitrogen (avoid excess). Maintain good drainage and field sanitation: weed removal, plow under stubbles and straw. Destroy ratoons and volunteer seedlings. Allow fallow fields to dry to suppress pathogens. Consult your local DA technician for specific advice.',
-        'details': 'Bacterial Leaf Blight (BLB) is caused by Xanthomonas oryzae pv. oryzae. Symptoms include water-soaked stripes that expand to large grayish-white lesions with wavy light brown margins. In susceptible varieties, yield loss can reach 70%. At booting stage, causes poor quality grains and high broken grain percentage. Spread by ooze droplets, strong winds, heavy rains, contaminated stubbles, and infected weeds in tropical and temperate lowlands (25-34°C, >70% relative humidity).',
+        'recommendation': 'Use resistant varieties like PSB Rc18 (ALA), NSIC Rc354, or NSIC Rc302. Apply balanced nutrients, especially nitrogen (avoid excess). Maintain good drainage and field sanitation: weed removal, plow under stubbles and straw. Destroy ratoons and volunteer seedlings. Allow fallow fields to dry to suppress pathogens. Use clean fields, moderate nitrogen use, proper drainage, and destruction of infected plant debris. Consult your local DA technician for specific advice.',
+        'details': 'Bacterial Leaf Blight (BLB) is one of the most serious bacterial diseases of rice caused by Xanthomonas oryzae pv. oryzae. Early symptoms start as tiny water-soaked spots that spread into large grayish-white lesions with wavy light brown margins, causing leaves to wilt and die (known as "Kresek" symptom). In susceptible varieties, yield loss can reach 70%. At booting stage, causes poor quality grains and high broken grain percentage. The disease spreads through rain, wind, contaminated water, tools, ooze droplets, strong winds, heavy rains, contaminated stubbles, and infected weeds. It thrives in high humidity and warm temperatures (25-34°C, >70% relative humidity) in tropical and temperate lowlands.',
         'category': 'Bacterial',
         'severity': 'High'
     },
 
     'bacterial_leaf_streak': {
-        'recommendation': 'Use resistant rice varieties. Apply balanced fertilization and avoid excessive nitrogen. Maintain proper field drainage and sanitation. Remove and destroy infected plant debris. Practice crop rotation where possible. Use copper-based bactericides only if infection is severe, following BPI guidelines. Consult local DA for variety recommendations.',
-        'details': 'Bacterial Leaf Streak (BLS) is caused by Xanthomonas oryzae pv. oryzicola. Symptoms include narrow, dark brown to yellow-orange streaks between leaf veins. Leaves may become yellow and dry. Disease is favored by high humidity, warm temperatures (25-34°C), and heavy rainfall. Can cause 10-30% yield loss in susceptible varieties. Spread through wind-driven rain, contaminated tools, and infected seeds.',
+        'recommendation': 'Use resistant rice varieties. Apply balanced fertilization and avoid excessive nitrogen. Maintain proper field drainage and sanitation. Remove and destroy infected plant debris. Practice crop rotation where possible. Use clean planting materials and balanced fertilization. Use copper-based bactericides only if infection is severe, following BPI guidelines. Consult local DA for variety recommendations.',
+        'details': 'Bacterial Leaf Streak (BLS) is caused by Xanthomonas oryzae pv. oryzicola. Symptoms appear as small water-soaked streaks between leaf veins that turn narrow, dark brown to yellow-orange. In severe cases, leaves become yellow, dry up, and reduce grain weight. Disease is favored by high humidity, warm temperatures (25-34°C), and heavy rainfall. Can cause 10-30% yield loss in susceptible varieties. The disease spreads through infected seeds, stubbles, rain splash, wind-driven rain, and contaminated tools.',
         'category': 'Bacterial',
         'severity': 'Medium'
     },
@@ -134,7 +134,7 @@ RECOMMENDATIONS = {
 
     'brown_spot': {
         'recommendation': 'Plant resistant varieties and ensure balanced fertilization, especially sufficient potassium (use complete or potash fertilizers). Conduct soil test and correct deficiencies. Improve soil health and maintain proper field drainage. Use clean, healthy seeds and practice field sanitation (remove stubbles, weeds). Split nitrogen fertilizer application. Apply calcium silicate & potassium fertilizers. If severe, apply fungicides (triazoles, strobilurins) following BPI registration and safety rules. Consult local DA.',
-        'details': 'Brown Spot is caused by Cochliobolus miyabeanus. Symptoms include small, circular to oval spots with gray centers on leaves, black spots on glumes with dark brown velvety fungal spores, and discolored, shriveled grains. Historically caused the 1943 Great Bengal Famine. Yield loss ranges 5-45% and causes seedling blight mortality of 10-58%. Fungus survives 2-4 years in infected tissues under high humidity (86-100%), 16-36°C, in nutrient-deficient or toxic soils.',
+        'details': 'Brown Spot is caused by Cochliobolus miyabeanus fungus. Historically responsible for the Great Bengal Famine in 1943 due to massive crop loss. It affects leaves, coleoptiles, roots, and grains, forming brown circular to oval spots with gray centers that can merge and wither the leaf. Symptoms also include black spots on glumes with dark brown velvety fungal spores and discolored, shriveled grains. The fungus can survive in infected seeds, stubbles, and soil for 2-4 years. It thrives in high humidity (86-100%), temperatures of 16-36°C, and particularly in nutrient-deficient or toxic soils. Yield loss ranges 5-45% and causes seedling blight mortality of 10-58%.',
         'category': 'Fungal',
         'severity': 'Medium to High'
     },
@@ -148,35 +148,35 @@ RECOMMENDATIONS = {
 
     'rice_blast': {
         'recommendation': 'Use resistant varieties such as PSB Rc18 or IRRI-derived lines. Avoid high nitrogen application and split nitrogen fertilizer application. Maintain proper irrigation (avoid water stress) and field sanitation (remove stubbles, debris). Adjust planting calendar to avoid peak infection period. Apply calcium silicate to strengthen cell walls. Apply fungicides: triazoles and strobilurins at first signs, especially during tillering/booting. Burn or plow under infected straw. Consult local DA.',
-        'details': 'Rice Blast is caused by Magnaporthe oryzae fungus. Symptoms include leaf blast (small, spindle-shaped spots with brown border and gray center), node blast (nodes become black and break easily), and panicle blast (base becomes black, grains remain unfilled). Occurs from seedling to reproductive stage under cool temperature, high relative humidity, continuous rain, and large day-night temperature differences. Can reduce yield significantly by reducing leaf area for grain filling.',
+        'details': 'Rice Blast is one of the most destructive fungal diseases caused by Magnaporthe oryzae. It can infect the plant at almost any stage from seedling to maturity. Symptoms include grayish spindle-shaped spots on leaves with brown borders and gray centers, blackened nodes that break easily, and unfilled panicles with blackened bases. The disease develops under cool and humid conditions with frequent rain or dew, particularly when there are large day-night temperature differences. Can reduce yield significantly by reducing leaf area for grain filling and causing panicle sterility.',
         'category': 'Fungal',
         'severity': 'Very High'
     },
 
     'rice_false_smut': {
         'recommendation': 'Use disease-free seeds and resistant varieties. Apply balanced fertilization, avoid excessive nitrogen at heading stage. Ensure proper field drainage and spacing for air circulation. Apply preventive fungicides (triazoles, strobilurins) at booting to early heading stage. Remove and destroy infected panicles (burn, do not compost). Adjust planting date to avoid high humidity during flowering. Consult DA for timing and fungicide recommendations.',
-        'details': 'Rice False Smut is caused by Ustilaginoidea virens fungus. Individual grains are transformed into large (1-2 cm), velvety, yellow-green to dark green balls of fungal spores. Infected grains are replaced by smut balls that reduce yield and grain quality. Disease is favored by high nitrogen, high humidity (>90%), and temperatures of 25-30°C during flowering. Can produce mycotoxins harmful to humans and animals. Yield loss typically 5-10% but quality loss is significant.',
+        'details': 'Rice False Smut is caused by Ustilaginoidea virens fungus. This disease appears during the flowering to milking stage. Individual grains are transformed into large (1-2 cm), velvety, yellow-green to dark green balls of fungal spores, causing chalky and lighter grains. Infected grains are completely replaced by smut balls that reduce yield and grain quality. The fungus favors low temperatures (around 20°C), high humidity (>90%), and high nitrogen levels during flowering. Can produce mycotoxins harmful to humans and animals. Yield loss typically 5-10% but quality loss is significant.',
         'category': 'Fungal',
         'severity': 'Medium'
     },
 
     'sheath_blight': {
         'recommendation': 'Use moderately resistant varieties. Apply balanced fertilization, avoid excessive nitrogen. Maintain proper plant spacing for air circulation. Ensure good water management and drainage. Remove lower infected leaves when feasible. Apply fungicides (validamycin, azoxystrobin) at early infection stages. Use silicon fertilizers to strengthen plant tissue. Practice crop rotation with non-host crops. Consult local DA for fungicide timing.',
-        'details': 'Sheath Blight is caused by Rhizoctonia solani fungus. Symptoms include oval to irregular greenish-gray lesions on leaf sheaths near water line, later becoming white to gray with brown borders. Lesions expand upward to leaves and panicles. Can cause lodging and reduced grain filling. Disease is favored by high humidity, dense planting, high nitrogen, and warm temperatures (28-32°C). Yield loss can reach 25-50% in susceptible varieties under favorable conditions.',
+        'details': 'Sheath Blight is caused by Rhizoctonia solani fungus and is second only to rice blast in importance. It infects the leaf sheath and lower leaves, producing oval to irregular greenish-gray lesions near the water line that later become white to gray with brown borders. Lesions spread quickly upward to leaves and panicles. The disease develops in warm (28-32°C), humid conditions, especially in fields with dense planting and excessive nitrogen use. It spreads through sclerotia and mycelium floating on water. Can cause lodging and reduced grain filling. Yield loss can reach 25-50% in susceptible varieties under favorable conditions.',
         'category': 'Fungal',
         'severity': 'High'
     },
 
     'sheath_rot': {
         'recommendation': 'Use resistant or tolerant varieties. Apply balanced fertilization (avoid excess nitrogen at late growth stages). Maintain proper water management, especially during heading. Ensure adequate spacing for air circulation. Remove and destroy infected panicles. Apply fungicides at boot leaf stage if disease history exists. Practice field sanitation. Consult DA for variety and fungicide recommendations.',
-        'details': 'Sheath Rot is caused primarily by Sarocladium oryzae (formerly Acrocylindrium oryzae). Symptoms include grayish-brown discoloration of flag leaf sheath, which may have pink to orange spore masses. Panicles emerge with difficulty, are partially exserted, may be discolored or rotted. Grains may be empty, discolored, or show poor filling. Disease is favored by high humidity, injuries (insect or mechanical), and imbalanced nutrition. Yield loss ranges 20-85% in severe cases.',
+        'details': 'Sheath Rot is caused primarily by Sarocladium oryzae (formerly Acrocylindrium oryzae). This disease affects the uppermost leaf sheath enclosing young panicles, leading to rotting and poor panicle emergence. Symptoms include grayish-brown discoloration of flag leaf sheath, which may have pink to orange spore masses. Panicles emerge with difficulty, are partially exserted, may be discolored or rotted. Grains may be empty, discolored, or show poor filling. The disease is common during the wet season with high nitrogen and humidity. Disease is favored by high humidity, injuries from insects or mechanical damage, and imbalanced nutrition. Yield loss ranges 20-85% in severe cases.',
         'category': 'Fungal',
         'severity': 'Medium to High'
     },
 
     'stem_rot': {
         'recommendation': 'Use resistant varieties and disease-free seeds. Maintain proper water management (avoid prolonged flooding). Apply balanced fertilization. Practice good field sanitation (remove stubbles, infected debris). Avoid mechanical injuries to stems. Ensure adequate drainage. Apply silicon fertilizers. Rotate with non-host crops. Remove and destroy infected plants. Consult DA for variety recommendations.',
-        'details': 'Stem Rot is caused by several fungi including Magnaporthe salvinii (formerly Sclerotium oryzae). Small black lesions appear on outer leaf sheaths near water line, expanding to form irregular black patches. Stems become hollow, weak, and lodge easily. Interior of stems shows black fungal growth. Disease is most severe in deep-water and flood-prone areas. Can cause complete lodging and severe yield loss (up to 100% in heavily infected fields). Favored by deep water, high temperatures, and prolonged flooding.',
+        'details': 'Stem Rot is caused by several fungi including Sclerotium oryzae (Magnaporthe salvinii). It infects the stem base near the waterline, forming small black lesions on outer leaf sheaths that expand into irregular black patches. Stems become hollow, weak, and lodge easily. Interior of stems shows black or white sclerotia and fungal growth. The disease is common in irrigated lowlands and spreads through floating sclerotia in water. Disease is most severe in deep-water and flood-prone areas. Can cause complete lodging and severe yield loss (up to 100% in heavily infected fields). Favored by deep water, high temperatures, and prolonged flooding.',
         'category': 'Fungal',
         'severity': 'High to Very High'
     },
@@ -185,22 +185,22 @@ RECOMMENDATIONS = {
     # VIRAL DISEASES (3)
     # =========================================================================
     'grassy_stunt_virus': {
-        'recommendation': 'Use resistant varieties if available. Control brown planthopper (vector) through integrated pest management: encourage natural enemies, avoid excessive nitrogen, maintain proper water depth. Practice synchronous planting with neighbors to break pest cycle. Remove and destroy infected plants immediately. Avoid continuous rice cropping (maintain fallow period). Monitor for planthopper populations. Note: No chemical cure for virus. Consult DA for resistant varieties.',
-        'details': 'Grassy Stunt Virus is transmitted by brown planthopper (Nilaparvata lugens). Symptoms include severe stunting, excessive tillering with thin tillers, narrow and short yellowish leaves, and no panicle formation. Infected plants resemble grass. Disease is most damaging when infection occurs at early growth stages. Can cause 100% yield loss if infected early. Virus cannot be cured once plant is infected. Common in tropical Asia.',
+        'recommendation': 'Use BPH-resistant varieties if available. Control brown planthopper (BPH - the vector) through integrated pest management: encourage natural enemies, avoid excessive nitrogen, maintain proper water depth. Practice synchronized planting with neighbors to break pest cycle and maintain fallow periods. Remove and destroy infected plants immediately. Plow infected stubbles. Avoid continuous rice cropping (maintain fallow period). Monitor for planthopper populations regularly. Note: No chemical cure for virus once infected. Consult DA for resistant varieties.',
+        'details': 'Rice Grassy Stunt Virus (RGSV) is transmitted by the brown planthopper (BPH - Nilaparvata lugens). Symptoms include severe stunting, excessive tillering with thin tillers resembling grass (grassy and rosette-like plants), narrow and short yellowish leaves, and failure to produce panicles or no panicle formation. Infected plants resemble grass. The disease is favored by continuous rice planting and high humidity. Disease is most damaging when infection occurs at early growth stages. Can cause 100% yield loss if infected early. Virus cannot be cured once plant is infected. Common in tropical Asia.',
         'category': 'Viral',
         'severity': 'Very High'
     },
 
     'ragged_stunt_virus': {
-        'recommendation': 'Use resistant or tolerant varieties. Control brown planthopper (vector) populations through IPM: biological control, proper water management, balanced fertilization (avoid excess nitrogen). Practice synchronous planting and maintain fallow periods between crops. Remove and destroy infected plants. Monitor planthopper populations regularly. Use light traps for monitoring. No chemical cure for virus. Consult DA for variety recommendations.',
-        'details': 'Ragged Stunt Virus is transmitted by brown planthopper (Nilaparvata lugens). Symptoms include stunted growth, twisted and ragged leaves with irregular margins, short and twisted panicles, dark green leaves with yellow or orange discoloration, and reduced grain filling. Infected plants produce small, distorted panicles with high sterility. Can cause 20-100% yield loss depending on infection stage. Most damaging when infection occurs before maximum tillering.',
+        'recommendation': 'Use resistant or tolerant varieties. Control brown planthopper (BPH - vector) populations through IPM: biological control, proper water management, balanced fertilization (avoid excess nitrogen). Practice synchronous planting and maintain fallow periods between crops. Remove and destroy infected plants. Monitor planthopper populations regularly. Use light traps for monitoring. Ensure proper crop timing and field sanitation. No chemical cure for virus once infected. Consult DA for variety recommendations.',
+        'details': 'Rice Ragged Stunt Virus (RRSV) is also transmitted by brown planthopper (BPH - Nilaparvata lugens). Symptoms include stunted growth, twisted and ragged leaves with irregular margins, short and twisted panicles, spiral twisted leaves, dark green leaves with yellow or orange discoloration, delayed flowering, unfilled grains, and reduced grain filling. Infected plants produce small, distorted panicles with high sterility. Can cause 20-100% (up to 80%) yield loss depending on infection stage in tropical areas. Most damaging when infection occurs before maximum tillering.',
         'category': 'Viral',
         'severity': 'Very High'
     },
 
     'tungro_virus': {
-        'recommendation': 'Plant tungro- or GLH-resistant varieties like IR64 if available. Practice synchronous planting (sabay-sabay) with neighbors and maintain a fallow period (at least 1 month) between crops to break vector cycles. Remove (rogue) and destroy infected plants immediately. Plow infected stubbles after harvest to destroy inoculum and GLH breeding sites. Maintain balanced nutrient management. Note: Chemical control of GLH is not effective as they move quickly and spread tungro even with short feeding. Consult local DA.',
-        'details': 'Rice Tungro Disease is caused by a combination of two viruses (Rice Tungro Bacilliform Virus + Rice Tungro Spherical Virus) transmitted by green leafhoppers (GLH). Symptoms include young leaves becoming mottled, older leaves turning yellow to orange, stunted growth, reduced tillers, delayed flowering, small panicles, and high sterility. Highly destructive in South & Southeast Asia with yield loss up to 100% in susceptible varieties infected early. Most damaging during vegetative/tillering stage.',
+        'recommendation': 'Grow tungro-resistant or GLH-resistant varieties like IR64 if available. Practice synchronous planting (sabay-sabay) with neighbors and maintain a fallow period (at least 1 month) between crops to break vector cycles. Remove (rogue) and destroy infected plants and stubbles immediately. Plow infected stubbles after harvest to destroy inoculum and GLH breeding sites. Maintain balanced nutrient management. Note: Chemical control of GLH is not effective as they move quickly and spread tungro even with short feeding. Consult local DA.',
+        'details': 'Rice Tungro Disease is caused by a combination of two viruses: Rice Tungro Bacilliform Virus (RTBV) and Rice Tungro Spherical Virus (RTSV), transmitted by the green leafhopper (GLH). Infected plants show symptoms including yellow-orange leaves, young leaves becoming mottled, older leaves turning yellow to orange, stunted growth, fewer tillers, reduced tillers, delayed flowering, small panicles, sterile panicles, and high sterility. Yield loss can reach up to 100% if infection occurs early in the growth stage. Highly destructive in South & Southeast Asia. Most damaging during vegetative and tillering stage.',
         'category': 'Viral',
         'severity': 'Very High'
     },
@@ -419,3 +419,424 @@ if __name__ == "__main__":
     import uvicorn
     # Production settings: no hot reload, proper host binding
     uvicorn.run("app:app", host="0.0.0.0", port=8000, reload=False)
+
+# import io
+# import sys
+# import time
+# import logging
+# import tempfile
+# import torch
+# import torch.nn.functional as F
+# from pydantic import BaseModel
+# from fastapi import FastAPI, UploadFile, File, HTTPException
+# from fastapi.middleware.cors import CORSMiddleware
+# from contextlib import asynccontextmanager  # Add this import
+# from models.rice_model import RiceDiseaseCNN  # Legacy 4-class model
+# from models.resnet_rice_model import create_model  # New 14-class ResNet18 model
+# from rice_leaf_validator import RiceLeafValidator  # New ResNet18 validator (99.84% accuracy)
+# from utils.image_processing import process_image
+# import requests
+# import json
+# import os
+# from typing import Dict, Any
+
+# # Configure logging
+# logging.basicConfig(
+#     level=logging.INFO,
+#     format="%(asctime)s [%(levelname)s] %(message)s",
+#     handlers=[
+#         logging.StreamHandler(sys.stdout)
+#     ]
+# )
+
+# # Initialize model variables
+# model = None
+# rice_validator = None  # New ResNet18 validator instance
+# model_path = "rice_disease_model_final.pth"  # 14-class disease detection model
+# rice_validator_model_path = "rice_leaf_validator_resnet18.pth"  # New ResNet18 validator (99.84% accuracy)
+# model_loaded = False
+# validator_loaded = False
+# model_loading = False
+# last_error = None
+# device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+
+# # Define lifespan context manager for model loading
+# @asynccontextmanager
+# async def lifespan(app):
+#     # Load models on startup
+#     load_model()
+#     yield
+#     # Cleanup operations can be added here for shutdown
+
+# # Create FastAPI app with lifespan
+# app = FastAPI(lifespan=lifespan)
+
+# # Add CORS middleware
+# app.add_middleware(
+#     CORSMiddleware,
+#     allow_origins=["*"],  # Allows all origins
+#     allow_credentials=True,
+#     allow_methods=["*"],  # Allows all methods
+#     allow_headers=["*"],  # Allows all headers
+# )
+
+# # Class names matching your training data (14 classes)
+# # Order MUST match the training script's class order
+# CLASS_NAMES = [
+#     'bacterial_leaf_blight',
+#     'bacterial_leaf_streak',
+#     'bakanae',
+#     'brown_spot',
+#     'grassy_stunt_virus',
+#     'healthy_rice_plant',
+#     'narrow_brown_spot',
+#     'ragged_stunt_virus',
+#     'rice_blast',
+#     'rice_false_smut',
+#     'sheath_blight',
+#     'sheath_rot',
+#     'stem_rot',
+#     'tungro_virus'
+# ]
+
+# # Disease display names for more specific prompts
+# disease_display_names = {
+#     # Bacterial Diseases
+#     'bacterial_leaf_blight': 'Bacterial Leaf Blight (BLB)',
+#     'bacterial_leaf_streak': 'Bacterial Leaf Streak (BLS)',
+
+#     # Fungal Diseases
+#     'bakanae': 'Bakanae (Foolish Seedling Disease)',
+#     'brown_spot': 'Brown Spot Disease',
+#     'narrow_brown_spot': 'Narrow Brown Spot',
+#     'rice_blast': 'Rice Blast Disease',
+#     'rice_false_smut': 'Rice False Smut',
+#     'sheath_blight': 'Sheath Blight',
+#     'sheath_rot': 'Sheath Rot',
+#     'stem_rot': 'Stem Rot',
+
+#     # Viral Diseases
+#     'grassy_stunt_virus': 'Grassy Stunt Virus',
+#     'ragged_stunt_virus': 'Ragged Stunt Virus',
+#     'tungro_virus': 'Rice Tungro Virus',
+
+#     # Healthy
+#     'healthy_rice_plant': 'Healthy Rice Plant'
+# }
+
+# # Disease recommendations for all 14 classes
+# RECOMMENDATIONS = {
+#     # =========================================================================
+#     # BACTERIAL DISEASES (2)
+#     # =========================================================================
+#     'bacterial_leaf_blight': {
+#         'recommendation': 'Use resistant varieties like PSB Rc18 (ALA), NSIC Rc354, or NSIC Rc302. Apply balanced nutrients, especially nitrogen (avoid excess). Maintain good drainage and field sanitation: weed removal, plow under stubbles and straw. Destroy ratoons and volunteer seedlings. Allow fallow fields to dry to suppress pathogens. Consult your local DA technician for specific advice.',
+#         'details': 'Bacterial Leaf Blight (BLB) is caused by Xanthomonas oryzae pv. oryzae. Symptoms include water-soaked stripes that expand to large grayish-white lesions with wavy light brown margins. In susceptible varieties, yield loss can reach 70%. At booting stage, causes poor quality grains and high broken grain percentage. Spread by ooze droplets, strong winds, heavy rains, contaminated stubbles, and infected weeds in tropical and temperate lowlands (25-34°C, >70% relative humidity).',
+#         'category': 'Bacterial',
+#         'severity': 'High'
+#     },
+
+#     'bacterial_leaf_streak': {
+#         'recommendation': 'Use resistant rice varieties. Apply balanced fertilization and avoid excessive nitrogen. Maintain proper field drainage and sanitation. Remove and destroy infected plant debris. Practice crop rotation where possible. Use copper-based bactericides only if infection is severe, following BPI guidelines. Consult local DA for variety recommendations.',
+#         'details': 'Bacterial Leaf Streak (BLS) is caused by Xanthomonas oryzae pv. oryzicola. Symptoms include narrow, dark brown to yellow-orange streaks between leaf veins. Leaves may become yellow and dry. Disease is favored by high humidity, warm temperatures (25-34°C), and heavy rainfall. Can cause 10-30% yield loss in susceptible varieties. Spread through wind-driven rain, contaminated tools, and infected seeds.',
+#         'category': 'Bacterial',
+#         'severity': 'Medium'
+#     },
+
+#     # =========================================================================
+#     # FUNGAL DISEASES (8)
+#     # =========================================================================
+#     'bakanae': {
+#         'recommendation': 'Use certified disease-free seeds. Treat seeds with fungicides (e.g., prochloraz, tricyclazole) before planting. Remove and destroy infected seedlings immediately. Avoid deep sowing and excessive nitrogen fertilization. Maintain proper water management. Practice crop rotation. Do not use infected plants as seedbed. Consult DA for registered fungicides.',
+#         'details': 'Bakanae (Foolish Seedling Disease) is caused by Fusarium fujikuroi fungus. Infected seedlings grow abnormally tall, pale yellow-green, with thin stems and few tillers. Some seedlings may die, others produce empty or partially filled grains. Fungus produces gibberellin (growth hormone) causing abnormal elongation. Seedborne disease that can cause 10-50% stand loss. Favored by contaminated seeds and infected soil.',
+#         'category': 'Fungal',
+#         'severity': 'Medium'
+#     },
+
+#     'brown_spot': {
+#         'recommendation': 'Plant resistant varieties and ensure balanced fertilization, especially sufficient potassium (use complete or potash fertilizers). Conduct soil test and correct deficiencies. Improve soil health and maintain proper field drainage. Use clean, healthy seeds and practice field sanitation (remove stubbles, weeds). Split nitrogen fertilizer application. Apply calcium silicate & potassium fertilizers. If severe, apply fungicides (triazoles, strobilurins) following BPI registration and safety rules. Consult local DA.',
+#         'details': 'Brown Spot is caused by Cochliobolus miyabeanus. Symptoms include small, circular to oval spots with gray centers on leaves, black spots on glumes with dark brown velvety fungal spores, and discolored, shriveled grains. Historically caused the 1943 Great Bengal Famine. Yield loss ranges 5-45% and causes seedling blight mortality of 10-58%. Fungus survives 2-4 years in infected tissues under high humidity (86-100%), 16-36°C, in nutrient-deficient or toxic soils.',
+#         'category': 'Fungal',
+#         'severity': 'Medium to High'
+#     },
+
+#     'narrow_brown_spot': {
+#         'recommendation': 'Use resistant or tolerant varieties. Apply balanced fertilization with adequate potassium and micronutrients. Maintain proper water management and field sanitation. Remove infected plant debris after harvest. Use clean, certified seeds. Apply foliar fungicides if infection is severe. Practice proper crop rotation. Consult DA for variety selection.',
+#         'details': 'Narrow Brown Spot is caused by Cercospora janseana (Sphaerulina oryzina). Symptoms include narrow, elongated brown lesions on leaves, typically 1-5 mm long, running parallel to leaf veins. Lesions may have light centers with dark borders. Disease is favored by nutrient deficiency, especially potassium, and high humidity. Can reduce photosynthetic area and grain filling. Common in nutrient-poor soils.',
+#         'category': 'Fungal',
+#         'severity': 'Low to Medium'
+#     },
+
+#     'rice_blast': {
+#         'recommendation': 'Use resistant varieties such as PSB Rc18 or IRRI-derived lines. Avoid high nitrogen application and split nitrogen fertilizer application. Maintain proper irrigation (avoid water stress) and field sanitation (remove stubbles, debris). Adjust planting calendar to avoid peak infection period. Apply calcium silicate to strengthen cell walls. Apply fungicides: triazoles and strobilurins at first signs, especially during tillering/booting. Burn or plow under infected straw. Consult local DA.',
+#         'details': 'Rice Blast is caused by Magnaporthe oryzae fungus. Symptoms include leaf blast (small, spindle-shaped spots with brown border and gray center), node blast (nodes become black and break easily), and panicle blast (base becomes black, grains remain unfilled). Occurs from seedling to reproductive stage under cool temperature, high relative humidity, continuous rain, and large day-night temperature differences. Can reduce yield significantly by reducing leaf area for grain filling.',
+#         'category': 'Fungal',
+#         'severity': 'Very High'
+#     },
+
+#     'rice_false_smut': {
+#         'recommendation': 'Use disease-free seeds and resistant varieties. Apply balanced fertilization, avoid excessive nitrogen at heading stage. Ensure proper field drainage and spacing for air circulation. Apply preventive fungicides (triazoles, strobilurins) at booting to early heading stage. Remove and destroy infected panicles (burn, do not compost). Adjust planting date to avoid high humidity during flowering. Consult DA for timing and fungicide recommendations.',
+#         'details': 'Rice False Smut is caused by Ustilaginoidea virens fungus. Individual grains are transformed into large (1-2 cm), velvety, yellow-green to dark green balls of fungal spores. Infected grains are replaced by smut balls that reduce yield and grain quality. Disease is favored by high nitrogen, high humidity (>90%), and temperatures of 25-30°C during flowering. Can produce mycotoxins harmful to humans and animals. Yield loss typically 5-10% but quality loss is significant.',
+#         'category': 'Fungal',
+#         'severity': 'Medium'
+#     },
+
+#     'sheath_blight': {
+#         'recommendation': 'Use moderately resistant varieties. Apply balanced fertilization, avoid excessive nitrogen. Maintain proper plant spacing for air circulation. Ensure good water management and drainage. Remove lower infected leaves when feasible. Apply fungicides (validamycin, azoxystrobin) at early infection stages. Use silicon fertilizers to strengthen plant tissue. Practice crop rotation with non-host crops. Consult local DA for fungicide timing.',
+#         'details': 'Sheath Blight is caused by Rhizoctonia solani fungus. Symptoms include oval to irregular greenish-gray lesions on leaf sheaths near water line, later becoming white to gray with brown borders. Lesions expand upward to leaves and panicles. Can cause lodging and reduced grain filling. Disease is favored by high humidity, dense planting, high nitrogen, and warm temperatures (28-32°C). Yield loss can reach 25-50% in susceptible varieties under favorable conditions.',
+#         'category': 'Fungal',
+#         'severity': 'High'
+#     },
+
+#     'sheath_rot': {
+#         'recommendation': 'Use resistant or tolerant varieties. Apply balanced fertilization (avoid excess nitrogen at late growth stages). Maintain proper water management, especially during heading. Ensure adequate spacing for air circulation. Remove and destroy infected panicles. Apply fungicides at boot leaf stage if disease history exists. Practice field sanitation. Consult DA for variety and fungicide recommendations.',
+#         'details': 'Sheath Rot is caused primarily by Sarocladium oryzae (formerly Acrocylindrium oryzae). Symptoms include grayish-brown discoloration of flag leaf sheath, which may have pink to orange spore masses. Panicles emerge with difficulty, are partially exserted, may be discolored or rotted. Grains may be empty, discolored, or show poor filling. Disease is favored by high humidity, injuries (insect or mechanical), and imbalanced nutrition. Yield loss ranges 20-85% in severe cases.',
+#         'category': 'Fungal',
+#         'severity': 'Medium to High'
+#     },
+
+#     'stem_rot': {
+#         'recommendation': 'Use resistant varieties and disease-free seeds. Maintain proper water management (avoid prolonged flooding). Apply balanced fertilization. Practice good field sanitation (remove stubbles, infected debris). Avoid mechanical injuries to stems. Ensure adequate drainage. Apply silicon fertilizers. Rotate with non-host crops. Remove and destroy infected plants. Consult DA for variety recommendations.',
+#         'details': 'Stem Rot is caused by several fungi including Magnaporthe salvinii (formerly Sclerotium oryzae). Small black lesions appear on outer leaf sheaths near water line, expanding to form irregular black patches. Stems become hollow, weak, and lodge easily. Interior of stems shows black fungal growth. Disease is most severe in deep-water and flood-prone areas. Can cause complete lodging and severe yield loss (up to 100% in heavily infected fields). Favored by deep water, high temperatures, and prolonged flooding.',
+#         'category': 'Fungal',
+#         'severity': 'High to Very High'
+#     },
+
+#     # =========================================================================
+#     # VIRAL DISEASES (3)
+#     # =========================================================================
+#     'grassy_stunt_virus': {
+#         'recommendation': 'Use resistant varieties if available. Control brown planthopper (vector) through integrated pest management: encourage natural enemies, avoid excessive nitrogen, maintain proper water depth. Practice synchronous planting with neighbors to break pest cycle. Remove and destroy infected plants immediately. Avoid continuous rice cropping (maintain fallow period). Monitor for planthopper populations. Note: No chemical cure for virus. Consult DA for resistant varieties.',
+#         'details': 'Grassy Stunt Virus is transmitted by brown planthopper (Nilaparvata lugens). Symptoms include severe stunting, excessive tillering with thin tillers, narrow and short yellowish leaves, and no panicle formation. Infected plants resemble grass. Disease is most damaging when infection occurs at early growth stages. Can cause 100% yield loss if infected early. Virus cannot be cured once plant is infected. Common in tropical Asia.',
+#         'category': 'Viral',
+#         'severity': 'Very High'
+#     },
+
+#     'ragged_stunt_virus': {
+#         'recommendation': 'Use resistant or tolerant varieties. Control brown planthopper (vector) populations through IPM: biological control, proper water management, balanced fertilization (avoid excess nitrogen). Practice synchronous planting and maintain fallow periods between crops. Remove and destroy infected plants. Monitor planthopper populations regularly. Use light traps for monitoring. No chemical cure for virus. Consult DA for variety recommendations.',
+#         'details': 'Ragged Stunt Virus is transmitted by brown planthopper (Nilaparvata lugens). Symptoms include stunted growth, twisted and ragged leaves with irregular margins, short and twisted panicles, dark green leaves with yellow or orange discoloration, and reduced grain filling. Infected plants produce small, distorted panicles with high sterility. Can cause 20-100% yield loss depending on infection stage. Most damaging when infection occurs before maximum tillering.',
+#         'category': 'Viral',
+#         'severity': 'Very High'
+#     },
+
+#     'tungro_virus': {
+#         'recommendation': 'Plant tungro- or GLH-resistant varieties like IR64 if available. Practice synchronous planting (sabay-sabay) with neighbors and maintain a fallow period (at least 1 month) between crops to break vector cycles. Remove (rogue) and destroy infected plants immediately. Plow infected stubbles after harvest to destroy inoculum and GLH breeding sites. Maintain balanced nutrient management. Note: Chemical control of GLH is not effective as they move quickly and spread tungro even with short feeding. Consult local DA.',
+#         'details': 'Rice Tungro Disease is caused by a combination of two viruses (Rice Tungro Bacilliform Virus + Rice Tungro Spherical Virus) transmitted by green leafhoppers (GLH). Symptoms include young leaves becoming mottled, older leaves turning yellow to orange, stunted growth, reduced tillers, delayed flowering, small panicles, and high sterility. Highly destructive in South & Southeast Asia with yield loss up to 100% in susceptible varieties infected early. Most damaging during vegetative/tillering stage.',
+#         'category': 'Viral',
+#         'severity': 'Very High'
+#     },
+
+#     # =========================================================================
+#     # HEALTHY
+#     # =========================================================================
+#     'healthy_rice_plant': {
+#         'recommendation': 'Your rice plant appears healthy! Continue with good agricultural practices: maintain balanced fertilization, proper irrigation and drainage, regular field monitoring, integrated pest management, and field sanitation. Keep monitoring for any early signs of diseases or pests. Follow recommended crop calendar and consult your local DA for best practices.',
+#         'details': 'No disease detected. The plant shows normal healthy characteristics including uniform green color, normal growth, no lesions or discoloration, and no signs of stress or infection. Continue preventive measures and regular monitoring to maintain plant health.',
+#         'category': 'Healthy',
+#         'severity': 'None'
+#     }
+# }
+
+# def load_model():
+#     """Load the PyTorch models"""
+#     global model, rice_validator, model_loaded, validator_loaded, model_loading, last_error
+#     try:
+#         logging.info(f"Loading disease model from: {model_path}")
+#         model_loading = True
+
+#         # Initialize the ResNet18 disease model architecture for 14 classes
+#         model = create_model(num_classes=14).to(device)
+
+#         # Load the saved model weights
+#         model.load_state_dict(torch.load(model_path, map_location=device))
+#         model.eval()  # Set to evaluation mode
+#         model_loaded = True
+
+#         # Load rice leaf validator model (new ResNet18 validator - 99.84% accuracy)
+#         logging.info(f"Loading ResNet18 rice leaf validator from: {rice_validator_model_path}")
+#         rice_validator = RiceLeafValidator(rice_validator_model_path, device)
+#         validator_loaded = True
+
+#         model_loading = False
+#         logging.info("Models loaded successfully!")
+#         logging.info(f"Disease model: ResNet18 with {len(CLASS_NAMES)} classes")
+#         logging.info(f"Rice validator: ResNet18 binary classifier (99.84% accuracy)")
+#         logging.info(f"Classes: {CLASS_NAMES}")
+#         return True
+#     except Exception as e:
+#         last_error = str(e)
+#         model_loading = False
+#         logging.error(f"Failed to load models: {e}")
+#         return False
+
+# @app.get("/")
+# async def root():
+#     """API root endpoint"""
+#     return {
+#         "message": "Rice Disease Detection API is running!",
+#         "model_loaded": model_loaded,
+#         "validator_loaded": validator_loaded,
+#         "model_loading": model_loading,
+#         "error": last_error
+#     }
+
+# @app.get("/health")
+# async def health():
+#     """Health check endpoint needed by PHP status checker"""
+#     return {
+#         "status": "running",
+#         "model_status": "loaded" if model_loaded else "loading" if model_loading else "not_loaded",
+#         "validator_status": "loaded" if validator_loaded else "not_loaded",
+#         "model_loaded": model_loaded,
+#         "model_loading": model_loading,
+#         "last_error": last_error,
+#         "device": str(device)
+#     }
+
+# @app.get("/status")
+# async def status():
+#     """Check model loading status"""
+#     return {
+#         "model_loaded": model_loaded,
+#         "validator_loaded": validator_loaded,
+#         "model_loading": model_loading,
+#         "error": last_error,
+#         "device": str(device)
+#     }
+
+# @app.post("/validate-leaf/")
+# async def validate_rice_leaf_image(file: UploadFile = File(...)):
+#     """Check if an image contains a rice leaf"""
+#     if not validator_loaded:
+#         if model_loading:
+#             raise HTTPException(status_code=503, detail="Models are still loading. Please try again later.")
+#         else:
+#             success = load_model()
+#             if not success:
+#                 raise HTTPException(status_code=500, detail=f"Failed to load models: {last_error}")
+
+#     try:
+#         # Save uploaded file temporarily (cross-platform)
+#         with tempfile.NamedTemporaryFile(delete=False, suffix=os.path.splitext(file.filename)[1]) as temp_file:
+#             temp_file_path = temp_file.name
+#             temp_file.write(await file.read())
+
+#         # Check if image contains rice leaf using ResNet18 validator
+#         result = rice_validator.validate_image(temp_file_path, threshold=0.5)
+#         is_rice = result['is_rice_leaf']
+#         confidence = result['confidence']
+
+#         # Remove temporary file
+#         os.remove(temp_file_path)
+
+#         return {
+#             "is_rice_leaf": is_rice,
+#             "confidence": f"{confidence * 100:.2f}%",
+#             "probabilities": result['probabilities']
+#         }
+#     except Exception as e:
+#         logging.error(f"Error during leaf validation: {e}")
+#         raise HTTPException(status_code=500, detail=f"Validation failed: {str(e)}")
+
+# @app.post("/predict/")
+# async def predict_disease(file: UploadFile = File(...), use_ai_recommendation: bool = True):
+#     """Predict rice disease from uploaded image"""
+#     # Check if models are loaded
+#     if not model_loaded or not validator_loaded:
+#         if model_loading:
+#             raise HTTPException(status_code=503, detail="Models are still loading. Please try again later.")
+#         else:
+#             # Try loading the models if they're not already loading
+#             success = load_model()
+#             if not success:
+#                 raise HTTPException(status_code=500, detail=f"Failed to load models: {last_error}")
+
+#     try:
+#         # Read image file
+#         start_time = time.time()
+
+#         # Save uploaded file temporarily for validation (cross-platform)
+#         contents = await file.read()
+#         with tempfile.NamedTemporaryFile(delete=False, suffix=os.path.splitext(file.filename)[1]) as temp_file:
+#             temp_file_path = temp_file.name
+#             temp_file.write(contents)
+
+#         logging.info(f"Image size: {len(contents)} bytes")
+
+#         # First validate if it's a rice leaf using ResNet18 validator
+#         validation_result = rice_validator.validate_image(temp_file_path, threshold=0.5)
+#         is_rice = validation_result['is_rice_leaf']
+#         rice_confidence = validation_result['confidence']
+
+#         # Remove temporary file
+#         os.remove(temp_file_path)
+
+#         if not is_rice:
+#             logging.info(f"Image not recognized as rice leaf: {rice_confidence * 100:.2f}% confidence")
+#             return {
+#                 "is_rice_leaf": False,
+#                 "confidence": f"{rice_confidence * 100:.2f}%",
+#                 "message": "The uploaded image does not appear to be a rice leaf. Please upload a clear image of a rice plant leaf.",
+#                 "probabilities": validation_result['probabilities']
+#             }
+
+#         # Process image for disease detection
+#         input_tensor = process_image(contents).to(device)
+
+#         # Make prediction
+#         with torch.no_grad():
+#             outputs = model(input_tensor)
+#             probabilities = F.softmax(outputs, dim=1)[0]
+
+#             # Get top predictions with probabilities
+#             top_probs, top_classes = torch.topk(probabilities, len(CLASS_NAMES))
+
+#             predictions = []
+#             for i in range(len(top_classes)):
+#                 class_idx = top_classes[i].item()
+#                 predictions.append({
+#                     "label": CLASS_NAMES[class_idx],
+#                     "confidence": f"{top_probs[i].item()*100:.2f}%"
+#                 })
+
+#         # Get primary prediction (highest confidence)
+#         predicted_class = top_classes[0].item()
+#         disease_name = CLASS_NAMES[predicted_class]
+#         confidence = float(top_probs[0].item()) * 100
+
+#         # Get comprehensive disease recommendation from built-in database
+#         recommendation_data = RECOMMENDATIONS.get(disease_name, {
+#             'recommendation': 'Unknown disease. Please consult with agricultural extension services.',
+#             'details': 'Disease information not available in database.',
+#             'category': 'Unknown',
+#             'severity': 'Unknown'
+#         })
+
+#         logging.info(f"Using built-in recommendation for {disease_name}")
+
+#         # Log prediction time
+#         processing_time = time.time() - start_time
+#         logging.info(f"Prediction completed in {processing_time:.3f} seconds")
+
+#         # Return in format expected by PHP code
+#         return {
+#             "is_rice_leaf": True,
+#             "leaf_confidence": f"{rice_confidence * 100:.2f}%",
+#             "predictions": predictions,
+#             "disease": disease_name,
+#             "confidence": f"{confidence:.2f}%",
+#             "recommendation": recommendation_data.get("recommendation", ""),
+#             "details": recommendation_data.get("details", ""),
+#             "category": recommendation_data.get("category", "Unknown"),
+#             "severity": recommendation_data.get("severity", "Unknown"),
+#             "inference_time_seconds": round(processing_time, 3)
+#         }
+
+#     except Exception as e:
+#         logging.error(f"Error during prediction: {e}")
+#         raise HTTPException(status_code=500, detail=f"Prediction failed: {str(e)}")
+
+# if __name__ == "__main__":
+#     import uvicorn
+#     # Production settings: no hot reload, proper host binding
+#     uvicorn.run("app:app", host="0.0.0.0", port=8000, reload=False)