import gradio as gr
import torch
import torch.nn.functional as F
from transformers import ViTImageProcessor, ViTForImageClassification
from PIL import Image
import google.generativeai as genai
import os
import logging
import asyncio
from typing import Tuple, Optional
import time

# Configure logging
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

# Configure Gemini API
GEMINI_API_KEY = os.getenv('GEMINI_API_KEY')
if GEMINI_API_KEY:
    genai.configure(api_key=GEMINI_API_KEY)
    gemini_model = genai.GenerativeModel('gemini-2.0-flash-exp')
else:
    logger.warning("GEMINI_API_KEY not found. Remedy suggestions will be disabled.")
    gemini_model = None

# Global variables for model caching
model = None
processor = None

# Improved crop and disease mapping
CROP_DISEASE_MAPPING = {
    # Tomato diseases
    "tomato___bacterial_spot": ("Tomato", "Bacterial Spot"),
    "tomato___early_blight": ("Tomato", "Early Blight"),
    "tomato___late_blight": ("Tomato", "Late Blight"),
    "tomato___leaf_mold": ("Tomato", "Leaf Mold"),
    "tomato___septoria_leaf_spot": ("Tomato", "Septoria Leaf Spot"),
    "tomato___spider_mites_two_spotted_spider_mite": ("Tomato", "Two-Spotted Spider Mite"),
    "tomato___target_spot": ("Tomato", "Target Spot"),
    "tomato___tomato_yellow_leaf_curl_virus": ("Tomato", "Yellow Leaf Curl Virus"),
    "tomato___tomato_mosaic_virus": ("Tomato", "Mosaic Virus"),
    "tomato___healthy": ("Tomato", "Healthy"),
    
    # Potato diseases
    "potato___early_blight": ("Potato", "Early Blight"),
    "potato___late_blight": ("Potato", "Late Blight"),
    "potato___healthy": ("Potato", "Healthy"),
    
    # Apple diseases
    "apple___apple_scab": ("Apple", "Apple Scab"),
    "apple___black_rot": ("Apple", "Black Rot"),
    "apple___cedar_apple_rust": ("Apple", "Cedar Apple Rust"),
    "apple___healthy": ("Apple", "Healthy"),
    
    # Corn diseases
    "corn_(maize)___cercospora_leaf_spot": ("Corn/Maize", "Cercospora Leaf Spot"),
    "corn_(maize)___common_rust": ("Corn/Maize", "Common Rust"),
    "corn_(maize)___northern_leaf_blight": ("Corn/Maize", "Northern Leaf Blight"),
    "corn_(maize)___healthy": ("Corn/Maize", "Healthy"),
    
    # Grape diseases
    "grape___black_rot": ("Grape", "Black Rot"),
    "grape___esca_(black_measles)": ("Grape", "Esca (Black Measles)"),
    "grape___leaf_blight_(isariopsis_leaf_spot)": ("Grape", "Leaf Blight"),
    "grape___healthy": ("Grape", "Healthy"),
    
    # Bell Pepper diseases
    "pepper,_bell___bacterial_spot": ("Bell Pepper", "Bacterial Spot"),
    "pepper,_bell___healthy": ("Bell Pepper", "Healthy"),
    
    # Cherry diseases
    "cherry_(including_sour)___powdery_mildew": ("Cherry", "Powdery Mildew"),
    "cherry_(including_sour)___healthy": ("Cherry", "Healthy"),
    
    # Peach diseases
    "peach___bacterial_spot": ("Peach", "Bacterial Spot"),
    "peach___healthy": ("Peach", "Healthy"),
    
    # Strawberry diseases
    "strawberry___leaf_scorch": ("Strawberry", "Leaf Scorch"),
    "strawberry___healthy": ("Strawberry", "Healthy"),
    
    # Other crops
    "soybean___healthy": ("Soybean", "Healthy"),
    "squash___powdery_mildew": ("Squash", "Powdery Mildew"),
    "orange___haunglongbing_(citrus_greening)": ("Orange", "Huanglongbing (Citrus Greening)"),
    "raspberry___healthy": ("Raspberry", "Healthy"),
}

def load_model():
    """Load the ViT model for crop disease classification"""
    global model, processor
    try:
        model_name = "wambugu71/crop_leaf_diseases_vit"
        logger.info(f"Loading model: {model_name}")
        
        processor = ViTImageProcessor.from_pretrained(model_name)
        model = ViTForImageClassification.from_pretrained(model_name)
        
        logger.info("Model loaded successfully!")
        return True
    except Exception as e:
        logger.error(f"Error loading model: {str(e)}")
        return False

def parse_label(raw_label: str) -> Tuple[str, str]:
    """Parse the raw model output to extract crop and disease information"""
    label = raw_label.lower().strip()
    
    # Check if we have a direct mapping
    if label in CROP_DISEASE_MAPPING:
        return CROP_DISEASE_MAPPING[label]
    
    # Try to parse common patterns
    if "___" in label:
        parts = label.split("___")
        if len(parts) >= 2:
            crop_part = parts[0].replace("_", " ").title()
            disease_part = parts[1].replace("_", " ").title()
            
            # Clean up crop names
            crop_part = crop_part.replace("(", "").replace(")", "")
            if "Corn Maize" in crop_part:
                crop_part = "Corn/Maize"
            elif "Pepper Bell" in crop_part:
                crop_part = "Bell Pepper"
            elif "Cherry Including Sour" in crop_part:
                crop_part = "Cherry"
                
            return crop_part, disease_part
    
    return "Unknown Crop", "Unknown Disease"

def get_detailed_disease_info(crop: str, disease: str) -> str:
    """Get detailed information about the specific crop disease"""
    if disease == "Healthy":
        return f"✅ **Healthy {crop}** - Your plant appears to be in good condition! Continue with proper care practices."
    elif "blight" in disease.lower():
        return f"🍂 **{disease}** detected in **{crop}** - This is a fungal disease that causes tissue death and browning."
    elif "spot" in disease.lower():
        return f"⚫ **{disease}** found on **{crop}** - Characterized by dark lesions on plant tissues."
    elif "rust" in disease.lower():
        return f"🦠 **{disease}** affecting **{crop}** - Shows as orange/reddish pustules on plant surfaces."
    elif "virus" in disease.lower():
        return f"🦠 **{disease}** in **{crop}** - Viral infection causing various symptoms including mottling and distortion."
    else:
        return f"🔍 **{disease}** detected in **{crop}** - This condition requires attention and proper management."

async def get_enhanced_remedy_suggestions(crop: str, disease: str, confidence: float) -> str:
    """Get AI-powered remedy suggestions using Gemini API with crop-specific context"""
    if not gemini_model:
        return get_offline_remedy_suggestions(crop, disease)
    
    try:
        prompt = f"""
        As an expert plant pathologist and agricultural consultant, provide comprehensive treatment recommendations for {disease} affecting {crop} plants. 
        
        The diagnosis confidence is {confidence:.1%}.
        
        Please provide a detailed response with the following sections:
        
        1. **Immediate Actions** (2-3 urgent steps to take right now)
        2. **Treatment Options**:
           - Organic/biological treatments
           - Chemical fungicides/bactericides (if needed)
           - Cultural control methods
        3. **Prevention Strategy** (4-5 preventive measures)
        4. **Monitoring & Follow-up** (what to watch for)
        5. **Expected Timeline** (recovery expectations)
        
        Make it specific to {crop} cultivation and {disease} characteristics. Include product names where appropriate and be practical for farmers.
        Format with clear headings and bullet points for easy reading.
        """
        
        response = gemini_model.generate_content(
            prompt,
            generation_config=genai.types.GenerationConfig(
                temperature=0.7,
                max_output_tokens=1200,
            )
        )
        
        return f"🤖 **AI-Powered Treatment Plan for {crop} - {disease}:**\n\n{response.text}"
        
    except Exception as e:
        logger.error(f"Error getting AI remedy suggestions: {str(e)}")
        return get_offline_remedy_suggestions(crop, disease)

def get_offline_remedy_suggestions(crop: str, disease: str) -> str:
    """Fallback offline remedy suggestions"""
    return f"""
**📋 General Treatment Approach for {crop} - {disease}:**

**🚨 Immediate Actions:**
• Remove affected plant parts immediately
• Improve air circulation around plants
• Apply appropriate fungicide or treatment
• Isolate infected plants if possible

**🧪 Treatment Options:**
• For fungal diseases: Use copper-based fungicides or neem oil
• For bacterial diseases: Apply copper compounds
• For viral diseases: Remove infected plants, control insect vectors
• Organic options: Baking soda spray, compost tea

**🛡️ Prevention Strategy:**
• Choose disease-resistant varieties
• Practice crop rotation (3-4 year cycle)
• Maintain proper plant spacing
• Water at soil level, avoid wetting leaves
• Regular monitoring and early detection

**📞 Recommendation:** Consult local agricultural extension services for region-specific advice and approved treatments.
    """

def classify_disease_enhanced(image: Image.Image) -> Tuple[str, str, float, str]:
    """Enhanced disease classification with better parsing"""
    if model is None or processor is None:
        return "Unknown Crop", "Error", 0.0, "Model not loaded properly."
    
    try:
        inputs = processor(image, return_tensors="pt")
        
        with torch.no_grad():
            outputs = model(**inputs)
            probabilities = F.softmax(outputs.logits, dim=-1)
            confidence = torch.max(probabilities).item()
            predicted_class_id = outputs.logits.argmax().item()
            predicted_label = model.config.id2label[predicted_class_id]
        
        crop, disease = parse_label(predicted_label)
        disease_info = get_detailed_disease_info(crop, disease)
        
        logger.info(f"Raw prediction: {predicted_label}")
        logger.info(f"Parsed: Crop={crop}, Disease={disease}, Confidence={confidence:.2%}")
        
        return crop, disease, confidence, disease_info
        
    except Exception as e:
        logger.error(f"Error in classification: {str(e)}")
        return "Unknown Crop", "Error", 0.0, f"Classification failed: {str(e)}"

async def process_image_enhanced(image: Image.Image) -> Tuple[str, str]:
    """Enhanced image processing with better crop and disease identification"""
    if image is None:
        return "⚠️ No image provided", ""
    
    crop, disease, confidence, disease_info = classify_disease_enhanced(image)
    
    if disease == "Error":
        return f"❌ **Classification Error**\n{disease_info}", ""
    
    confidence_emoji = "🎯" if confidence > 0.8 else "📊" if confidence > 0.6 else "⚠️"
    
    main_result = f"""
## 🔬 **Crop Disease Detector Analysis Results**

### 🌱 **Identified Crop:** `{crop}`
### 🦠 **Disease Status:** `{disease}`
### 📊 **Confidence Score:** {confidence_emoji} `{confidence:.2%}`

---

### 📋 **Disease Information:**
{disease_info}

### 🔍 **Analysis Notes:**
- Higher confidence scores (>80%) indicate more reliable identification
- Multiple factors including image quality affect accuracy
- Consider consulting agricultural experts for critical decisions

---
    """.strip()
    
    remedy_text = await get_enhanced_remedy_suggestions(crop, disease, confidence)
    
    return main_result, remedy_text

# Custom CSS for agricultural theme
custom_css = """
@import url('https://fonts.googleapis.com/css2?family=Inter:wght@300;400;500;600;700;800&family=Space+Grotesk:wght@300;400;500;600;700&display=swap');

body, html {
    height: 100% !important;
    margin: 0 !important;
    padding: 0 !important;
    overflow-x: hidden !important;
}

.gradio-container {
    background: linear-gradient(135deg, #1a2e1a 0%, #2d4a2d 15%, #1e3a1e 35%, #3d5a3d 55%, #2a4d2a 75%, #1f361f 100%);
    color: #ffffff;
    font-family: 'Space Grotesk', 'Inter', -apple-system, BlinkMacSystemFont, sans-serif;
    min-height: 100vh !important;
    max-height: 100vh !important;
    overflow-y: auto !important;
    overflow-x: hidden !important;
}

#main-container {
    margin: 0 !important;
    padding: 20px !important;
    margin-bottom: 0 !important;
    padding-bottom: 0 !important;
}

.gr-block {
    margin-bottom: 0 !important;
}

/* Logo styling */
.agri-logo {
    width: 60px;
    height: 60px;
    display: inline-block;
    margin-right: 15px;
    vertical-align: middle;
    border-radius: 8px;
    object-fit: contain;
}

.nav-header-with-logo {
    background: linear-gradient(135deg, #1a2e1a 0%, #2d4a2d 50%, #1e3a1e 100%);
    padding: 20px 0;
    text-align: center;
    border-bottom: 3px solid #7FB069;
    margin-bottom: 30px;
    display: flex;
    align-items: center;
    justify-content: center;
    flex-direction: column;
}

.logo-title-container {
    display: flex;
    align-items: center;
    justify-content: center;
    margin-bottom: 10px;
}

/* Navigation styling */
.nav-header {
    background: linear-gradient(135deg, #1a2e1a 0%, #2d4a2d 50%, #1e3a1e 100%);
    padding: 20px 0;
    text-align: center;
    border-bottom: 3px solid #7FB069;
    margin-bottom: 30px;
}

.nav-title {
    background: linear-gradient(45deg, #B7E4C7, #7FB069);
    -webkit-background-clip: text;
    -webkit-text-fill-color: transparent;
    background-clip: text;
    font-size: 2.5rem;
    font-weight: 700;
    margin: 0;
}

.nav-subtitle {
    color: #B7E4C7;
    font-size: 1.1rem;
    margin: 10px 0 0 0;
}

/* Page content styling */
.page-container {
    max-width: 1200px;
    margin: 0 auto;
    padding: 20px;
}

.feature-grid {
    display: grid;
    grid-template-columns: repeat(auto-fit, minmax(300px, 1fr));
    gap: 20px;
    margin: 30px 0;
}

.feature-card {
    background: rgba(26, 46, 26, 0.85);
    border: 1px solid rgba(127, 176, 105, 0.3);
    border-radius: 15px;
    padding: 25px;
    text-align: center;
    transition: all 0.3s ease;
}

.feature-card:hover {
    transform: translateY(-5px);
    box-shadow: 0 10px 30px rgba(127, 176, 105, 0.2);
}

.feature-icon {
    font-size: 3rem;
    margin-bottom: 15px;
    display: block;
}

/* Pricing card alignment and spacing */
.pricing-container {
    display: flex;
    justify-content: center;
    align-items: stretch;
    gap: 20px;
    margin: 30px 0;
    flex-wrap: wrap;
}

.pricing-card {
    background: rgba(26, 46, 26, 0.9);
    border: 2px solid rgba(127, 176, 105, 0.4);
    border-radius: 20px;
    padding: 30px;
    text-align: center;
    position: relative;
    transition: all 0.3s ease;
    flex: 1;
    min-width: 280px;
    max-width: 350px;
    display: flex;
    flex-direction: column;
    justify-content: space-between;
}

.pricing-card:hover {
    transform: translateY(-3px);
    box-shadow: 0 8px 25px rgba(127, 176, 105, 0.3);
}

.price {
    font-size: 3rem;
    font-weight: bold;
    background: linear-gradient(45deg, #B7E4C7, #7FB069);
    -webkit-background-clip: text;
    -webkit-text-fill-color: transparent;
    background-clip: text;
}

/* Enterprise section alignment */
.enterprise-container {
    display: flex;
    justify-content: center;
    align-items: stretch;
    gap: 30px;
    margin: 30px 0;
    flex-wrap: wrap;
}

.enterprise-container .pricing-card {
    max-width: 400px;
}

/* Upload area styling */
.upload-area {
    border: 3px dashed #7FB069;
    border-radius: 25px;
    background: linear-gradient(135deg, rgba(26, 46, 26, 0.6) 0%, rgba(45, 74, 45, 0.4) 100%);
    transition: all 0.5s ease;
}

.upload-area:hover {
    border-color: #A7C957;
    background: linear-gradient(135deg, rgba(26, 46, 26, 0.8) 0%, rgba(45, 74, 45, 0.6) 100%);
    box-shadow: 0 0 40px rgba(167, 201, 87, 0.3);
}

.disease-result {
    background: linear-gradient(135deg, rgba(127, 176, 105, 0.12) 0%, rgba(183, 228, 199, 0.08) 100%);
    border-radius: 25px;
    padding: 2.5rem;
    margin: 2rem 0;
    border: 1px solid rgba(127, 176, 105, 0.2);
    color: #F8F9F0;
    backdrop-filter: blur(15px);
    position: relative;
}

.disease-result::before {
    content: '';
    position: absolute;
    top: 0;
    left: 0;
    width: 5px;
    height: 100%;
    background: linear-gradient(180deg, #7FB069 0%, #A7C957 100%);
    border-radius: 0 5px 5px 0;
}

.remedy-result {
    background: linear-gradient(135deg, rgba(167, 201, 87, 0.12) 0%, rgba(149, 213, 178, 0.08) 100%);
    border-radius: 25px;
    padding: 2.5rem;
    margin: 2rem 0;
    border: 1px solid rgba(167, 201, 87, 0.2);
    color: #F8F9F0;
    backdrop-filter: blur(15px);
    position: relative;
}

.remedy-result::before {
    content: '';
    position: absolute;
    top: 0;
    left: 0;
    width: 5px;
    height: 100%;
    background: linear-gradient(180deg, #A7C957 0%, #95D5B2 100%);
    border-radius: 0 5px 5px 0;
}

/* Popup styling */
.popup-overlay {
    position: fixed;
    top: 0;
    left: 0;
    width: 100%;
    height: 100%;
    background: rgba(0, 0, 0, 0.8);
    display: flex;
    justify-content: center;
    align-items: center;
    z-index: 1000;
}

.popup-content {
    background: rgba(26, 46, 26, 0.95);
    border: 2px solid #7FB069;
    border-radius: 20px;
    padding: 40px;
    text-align: center;
    max-width: 400px;
    position: relative;
}

.popup-close {
    position: absolute;
    top: 10px;
    right: 15px;
    background: none;
    border: none;
    color: #fff;
    font-size: 24px;
    cursor: pointer;
}

/* Button styling */
.upgrade-btn, .contact-btn, .analyze-btn {
    background: linear-gradient(45deg, #7FB069, #A7C957);
    color: white;
    border: none;
    border-radius: 25px;
    padding: 12px 25px;
    font-weight: 600;
    cursor: pointer;
    transition: all 0.3s ease;
    margin: 10px 5px;
    width: 100%;
}

.analyze-btn {
    padding: 20px 45px;
    font-size: 18px;
    text-transform: uppercase;
    letter-spacing: 1.5px;
    font-family: 'Space Grotesk', sans-serif;
}

.upgrade-btn:hover, .contact-btn:hover, .analyze-btn:hover {
    transform: translateY(-2px);
    box-shadow: 0 8px 25px rgba(127, 176, 105, 0.4);
}

/* Header styling */
.header-text {
    text-align: center;
    color: #F8F9F0;
    background: rgba(26, 46, 26, 0.4);
    backdrop-filter: blur(25px);
    border: 1px solid rgba(183, 228, 199, 0.15);
    border-radius: 30px;
    padding: 2.5rem;
    margin: 2rem auto;
    max-width: 900px;
    box-shadow: 0 20px 60px rgba(106, 153, 78, 0.2);
}

.header-text h1 {
    background: linear-gradient(135deg, #B7E4C7 0%, #7FB069 30%, #6A994E 60%, #95D5B2 100%);
    -webkit-background-clip: text;
    -webkit-text-fill-color: transparent;
    background-clip: text;
    font-size: 3.8rem;
    font-weight: 800;
    margin-bottom: 1rem;
    font-family: 'Space Grotesk', sans-serif;
}

/* Accessibility improvements */
.skip-link {
    position: absolute;
    top: -40px;
    left: 6px;
    background: #7FB069;
    color: #000;
    padding: 8px;
    text-decoration: none;
    border-radius: 4px;
    z-index: 1000;
    transition: top 0.3s;
}

.skip-link:focus {
    top: 6px;
}

a:focus, button:focus, input:focus, textarea:focus, select:focus {
    outline: 2px solid #7FB069;
    outline-offset: 2px;
}

@media (prefers-reduced-motion: reduce) {
    * {
        animation-duration: 0.01ms !important;
        animation-iteration-count: 1 !important;
        transition-duration: 0.01ms !important;
    }
}

/* Mobile responsive */
@media (max-width: 768px) {
    .nav-title {
        font-size: 2rem;
    }
    
    .agri-logo {
        width: 45px;
        height: 45px;
        margin-right: 10px;
    }
    
    .feature-grid {
        grid-template-columns: 1fr;
    }
    
    .pricing-container, .enterprise-container {
        flex-direction: column;
        align-items: center;
    }
    
    .pricing-card {
        max-width: 100%;
        margin: 10px 0;
    }
    
    .header-text h1 {
        font-size: 2.5rem;
    }
}
"""

# Crop Disease Detector URL (you can replace with your actual logo)
AGRI_LOGO_URL = "🌱"  # Using emoji as placeholder - replace with your actual logo URL

def create_home_page():
    """Create the home page for Crop Disease Detector"""
    with gr.Column():
        gr.HTML(f"""
            <div class="nav-header-with-logo">
                <div class="logo-title-container">
                    <div style="font-size: 60px; margin-right: 15px;">🌱</div>
                    <h1 class="nav-title">Crop Disease Detector</h1>
                </div>
                <p class="nav-subtitle">Your Crop's AI Doctor | Instant Plant Disease Detection</p>
            </div>
        """)
        
        gr.HTML("""
            <div class="page-container">
                <h2 style="text-align: center; color: #7FB069; font-size: 2.5rem; margin-bottom: 20px;">
                    Transforming Agriculture with AI
                </h2>
                
                <div style="text-align: center; margin: 40px 0;">
                    <p style="font-size: 1.3rem; color: #B7E4C7; max-width: 800px; margin: 0 auto; line-height: 1.6;">
                        Empowering farmers worldwide with instant plant disease detection using advanced AI. 
                        Upload a photo of your crop and get expert diagnosis in seconds.
                    </p>
                </div>
            </div>
        """)
        
        with gr.Row():
            with gr.Column():
                gr.HTML("""
                    <div class="feature-card">
                        <span class="feature-icon">⚡</span>
                        <h3 style="color: #7FB069;">Instant Diagnosis</h3>
                        <p>Photo → Disease identification in 30 seconds. No laboratory testing required.</p>
                    </div>
                """)
            with gr.Column():
                gr.HTML("""
                    <div class="feature-card">
                        <span class="feature-icon">🧠</span>
                        <h3 style="color: #7FB069;">AI-Powered Treatment</h3>
                        <p>Get detailed, actionable treatment recommendations powered by Gemini AI.</p>
                    </div>
                """)
            with gr.Column():
                gr.HTML("""
                    <div class="feature-card">
                        <span class="feature-icon">📱</span>
                        <h3 style="color: #7FB069;">Smartphone Accessible</h3>
                        <p>Works on any smartphone with camera. No special hardware needed.</p>
                    </div>
                """)
        
        with gr.Row():
            with gr.Column():
                gr.HTML("""
                    <div class="feature-card">
                        <span class="feature-icon">🌾</span>
                        <h3 style="color: #A7C957;">Multi-Crop Support</h3>
                        <p>Detects diseases in tomatoes, potatoes, corn, apples, grapes, and many more crops.</p>
                    </div>
                """)
            with gr.Column():
                gr.HTML("""
                    <div class="feature-card">
                        <span class="feature-icon">💚</span>
                        <h3 style="color: #A7C957;">Organic Solutions</h3>
                        <p>Prioritizes eco-friendly and organic treatment options for sustainable farming.</p>
                    </div>
                """)
            with gr.Column():
                gr.HTML("""
                    <div class="feature-card">
                        <span class="feature-icon">🆓</span>
                        <h3 style="color: #A7C957;">Free to Use</h3>
                        <p>Basic disease detection is completely free. Premium features available for advanced users.</p>
                    </div>
                """)
        
        gr.HTML("""
            <div style="text-align: center; margin: 50px 0;">
                <h3 style="color: #95D5B2; font-size: 1.5rem;">Our Mission</h3>
                <p style="font-size: 1.1rem; color: #ffffff; max-width: 700px; margin: 20px auto;">
                    To democratize agricultural expertise and prevent crop losses through accessible AI technology, 
                    ensuring food security and supporting farmer livelihoods worldwide.
                </p>
            </div>
        """)
        
        gr.HTML("""
            <div style="background: rgba(127, 176, 105, 0.15); border: 2px solid #7FB069; border-radius: 15px; padding: 20px; margin: 30px 0;">
                <p style="color: #7FB069; text-align: center; font-size: 1.1rem; margin: 0;">
                    <strong>🎯 Impact:</strong> Addressing ₹1,40,000 crores annual crop losses in India • 
                    Supporting SDG 2 (Zero Hunger) • Empowering 146 million agricultural households
                </p>
            </div>
        """)

def create_crop_detection_demo():
    """Create the main crop disease detection page"""
    with gr.Column():
        gr.HTML('<a href="#main-app" class="skip-link">Skip to main application</a>')
        
        # Header with logo
        gr.HTML("""
            <div class="header-text">
                <h1>🌱 Crop Disease Detector</h1>
                <p style="font-size: 1.3em; margin: 15px 0; background: linear-gradient(135deg, #B7E4C7 0%, #7FB069 100%); -webkit-background-clip: text; -webkit-text-fill-color: transparent; font-weight: 600;">
                    Advanced Crop Disease Detection & Treatment System
                </p>
                <p style="font-size: 1em; opacity: 0.9; color: rgba(255,255,255,0.8);">
                    Enhanced AI • Precise Crop Identification • Expert Treatment Plans
                </p>
            </div>
        """)
        
        if not model_loaded:
            gr.Warning("⚠️ Model failed to load. Please refresh the page or contact support.")
            return
        
        with gr.Row():
            # Input Column
            with gr.Column(scale=1, elem_classes="input-container"):
                gr.HTML("""
                <h3 style="margin-bottom: 20px; color: #B7E4C7;">📤 Upload Plant Image</h3>
                """)
                
                image_input = gr.Image(
                    type="pil",
                    label="Drop your plant image here or click to upload",
                    elem_classes="upload-area",
                    height=300,
                    elem_id="main-app"
                )
                
                analyze_btn = gr.Button(
                    "🔬 Analyze Plant Disease",
                    variant="primary",
                    size="lg",
                    elem_classes="analyze-btn"
                )
        
        # Output Section
        gr.HTML("""
        <h3 style="text-align: center; color: #B7E4C7; font-size: 2em; margin: 30px 0;">📊 AI Analysis Results</h3>
        """)
        
        with gr.Row():
            with gr.Column(scale=1):
                disease_output = gr.Markdown(
                    label="Disease Classification",
                    elem_classes="disease-result"
                )
            
            with gr.Column(scale=1):
                remedy_output = gr.Markdown(
                    label="Treatment Recommendations",
                    elem_classes="remedy-result"
                )
        
        # Event handlers
        async def handle_analysis(image):
            if image is None:
                return "⚠️ Please upload an image first.", ""
            
            try:
                disease_result, remedy_result = await process_image_enhanced(image)
                return disease_result, remedy_result
            except Exception as e:
                error_msg = f"❌ **Analysis Error**: {str(e)}"
                return error_msg, ""
        
        analyze_btn.click(
            fn=handle_analysis,
            inputs=[image_input],
            outputs=[disease_output, remedy_output]
        )
        
        # Auto-analyze on image upload
        image_input.change(
            fn=handle_analysis,
            inputs=[image_input],
            outputs=[disease_output, remedy_output]
        )
        
        # Footer information
        gr.HTML("""
        <div style="background: rgba(26, 46, 26, 0.4); backdrop-filter: blur(20px); border-radius: 25px; margin: 3rem auto; padding: 2rem; border: 1px solid rgba(183, 228, 199, 0.1); box-shadow: 0 10px 40px rgba(106, 153, 78, 0.1);">
            <div style="display: flex; justify-content: center; gap: 30px; flex-wrap: wrap; margin-bottom: 20px;">
                <div style="text-align: center;">
                    <div style="font-size: 2em; margin-bottom: 5px;">🧠</div>
                    <div style="color: #B7E4C7; font-weight: 600;">Enhanced Vision AI</div>
                    <div style="color: rgba(255,255,255,0.7); font-size: 0.9em;">Advanced Plant Analysis</div>
                </div>
                <div style="text-align: center;">
                    <div style="font-size: 2em; margin-bottom: 5px;">🌱</div>
                    <div style="color: #7FB069; font-weight: 600;">Crop-Specific Detection</div>
                    <div style="color: rgba(255,255,255,0.7); font-size: 0.9em;">Precise Identification</div>
                </div>
                <div style="text-align: center;">
                    <div style="font-size: 2em; margin-bottom: 5px;">⚡</div>
                    <div style="color: #A7C957; font-weight: 600;">Instant Treatment Plans</div>
                    <div style="color: rgba(255,255,255,0.7); font-size: 0.9em;">Expert Recommendations</div>
                </div>
            </div>
            <p style="color: rgba(255,255,255,0.8); margin: 10px 0; text-align: center;">
                🚀 Revolutionizing Agriculture with Enhanced AI • Built with Gradio • Powered by Transformers
            </p>
            <p style="color: rgba(255,255,255,0.6); font-size: 0.9em; margin: 0; text-align: center;">
                Empowering farmers worldwide with precise crop disease detection
            </p>
        </div>
        """)

def create_pricing_page():
    """Create the pricing page for Crop Disease Detector"""
    with gr.Column():
        # State variable to control popup visibility
        popup_state = gr.State(False)
        popup = gr.HTML(visible=False, elem_id="popup")
        
        gr.HTML("""
            <div class="nav-header">
                <h1 class="nav-title">💰 Crop Disease Detector Pricing</h1>
                <p class="nav-subtitle">Choose the right plan for your agricultural needs</p>
            </div>
        """)
        
        # Individual plans section
        gr.HTML('<div class="pricing-container">')
        
        with gr.Row():
            with gr.Column():
                gr.HTML("""
                    <div class="pricing-card">
                        <h3 style="color: #7FB069;">Always Free</h3>
                        <div class="price">₹0</div>
                        <p style="color: #B7E4C7;">per month</p>
                        <hr style="border-color: rgba(255,255,255,0.1);">
                        <ul style="text-align: left; color: #ffffff; margin: 20px 0;">
                            <li>✅ 10 disease scans per day</li>
                            <li>✅ Basic crop identification</li>
                            <li>✅ Standard treatment advice</li>
                            <li>✅ Mobile-friendly interface</li>
                            <li>✅ Community support</li>
                            <li>❌ Advanced AI insights</li>
                            <li>❌ Priority support</li>
                        </ul>
                    </div>
                """)
                free_btn = gr.Button("Get Started Free", elem_classes="upgrade-btn")
                
            with gr.Column():
                gr.HTML("""
                    <div class="pricing-card">
                        <h3 style="color: #7FB069;">Farmer Pro</h3>
                        <div class="price">₹199</div>
                        <p style="color: #B7E4C7;">per month</p>
                        <hr style="border-color: rgba(255,255,255,0.1);">
                        <ul style="text-align: left; color: #ffffff; margin: 20px 0;">
                            <li>✅ Unlimited disease scans</li>
                            <li>✅ Advanced AI diagnosis</li>
                            <li>✅ Detailed treatment plans</li>
                            <li>✅ Crop health monitoring</li>
                            <li>✅ Seasonal recommendations</li>
                            <li>✅ Multi-language support</li>
                            <li>❌ API access</li>
                        </ul>
                    </div>
                """)
                pro_btn = gr.Button("Upgrade to Pro", elem_classes="upgrade-btn")
                
            with gr.Column():
                gr.HTML("""
                    <div class="pricing-card">
                        <h3 style="color: #7FB069;">Agricultural Expert</h3>
                        <div class="price">₹499</div>
                        <p style="color: #B7E4C7;">per month</p>
                        <hr style="border-color: rgba(255,255,255,0.1);">
                        <ul style="text-align: left; color: #ffffff; margin: 20px 0;">
                            <li>✅ Everything in Farmer Pro</li>
                            <li>✅ Gemini Pro AI model</li>
                            <li>✅ Pest detection (coming soon)</li>
                            <li>✅ Soil health analysis</li>
                            <li>✅ Weather integration</li>
                            <li>✅ API access for integration</li>
                            <li>✅ Priority expert support</li>
                        </ul>
                    </div>
                """)
                expert_btn = gr.Button("Upgrade to Expert", elem_classes="upgrade-btn")
        
        gr.HTML('</div>')  # Close pricing-container
        
        gr.HTML("""
            <div style="text-align: center; margin: 50px 0;">
                <h3 style="color: #95D5B2; font-size: 1.8rem;">Enterprise & Cooperative Solutions</h3>
            </div>
        """)
        
        # Enterprise section
        gr.HTML('<div class="enterprise-container">')
        
        with gr.Row():
            with gr.Column():
                gr.HTML("""
                    <div class="pricing-card">
                        <h3 style="color: #7FB069;">Cooperative Basic</h3>
                        <div class="price">₹15,000</div>
                        <p style="color: #B7E4C7;">per year (100+ farmers)</p>
                        <hr style="border-color: rgba(255,255,255,0.1);">
                        <ul style="text-align: left; color: #ffffff; margin: 20px 0;">
                            <li>✅ Bulk farmer licensing</li>
                            <li>✅ Admin dashboard</li>
                            <li>✅ Usage analytics</li>
                            <li>✅ Training materials</li>
                            <li>✅ Custom branding</li>
                            <li>✅ WhatsApp integration</li>
                        </ul>
                    </div>
                """)
                coop_btn = gr.Button("Contact for Cooperatives", elem_classes="contact-btn")
                
            with gr.Column():
                gr.HTML("""
                    <div class="pricing-card">
                        <h3 style="color: #7FB069;">Enterprise</h3>
                        <div class="price">Custom</div>
                        <p style="color: #B7E4C7;">pricing</p>
                        <hr style="border-color: rgba(255,255,255,0.1);">
                        <ul style="text-align: left; color: #ffffff; margin: 20px 0;">
                            <li>✅ Custom AI model training</li>
                            <li>✅ On-premise deployment</li>
                            <li>✅ API integration</li>
                            <li>✅ White-label solution</li>
                            <li>✅ SLA guarantees</li>
                            <li>✅ Dedicated support</li>
                            <li>✅ Custom development</li>
                        </ul>
                    </div>
                """)
                enterprise_btn = gr.Button("Contact Enterprise Sales", elem_classes="contact-btn")
        
        gr.HTML('</div>')  # Close enterprise-container
        
        # Popup mechanism
        def toggle_popup(current_state):
            """Toggle popup visibility"""
            new_state = not current_state
            if new_state:
                popup_content = """
                    <div class="popup-overlay" onclick="this.style.display='none'">
                        <div class="popup-content" onclick="event.stopPropagation()">
                            <button class="popup-close" onclick="this.parentElement.parentElement.style.display='none'">&times;</button>
                            <h2 style="color: #7FB069;">Coming Soon! 🚀</h2>
                            <p style="color: #ffffff; margin: 20px 0;">
                                We're working hard to launch premium features. 
                                Stay tuned for updates!
                            </p>
                            <p style="color: #B7E4C7; font-size: 14px;">
                                Join our waiting list for early access and special pricing.
                            </p>
                        </div>
                    </div>
                """
                return new_state, gr.update(visible=True, value=popup_content)
            else:
                return new_state, gr.update(visible=False, value="")
        
        # Connect buttons with popup
        pro_btn.click(
            fn=toggle_popup,
            inputs=[popup_state],
            outputs=[popup_state, popup]
        )
        
        expert_btn.click(
            fn=toggle_popup,
            inputs=[popup_state],
            outputs=[popup_state, popup]
        )
        
        coop_btn.click(
            fn=toggle_popup,
            inputs=[popup_state],
            outputs=[popup_state, popup]
        )
        
        enterprise_btn.click(
            fn=toggle_popup,
            inputs=[popup_state],
            outputs=[popup_state, popup]
        )
        
        return free_btn

# Initialize the model
logger.info("Initializing  Crop Disease Detector...")
model_loaded = load_model()

# Create the multi-page application
with gr.Blocks(css=custom_css, theme=gr.themes.Soft(), title="Crop Disease Detector - Your Crop's AI Doctor") as app:
    gr.HTML('<a href="#main-content" class="skip-link">Skip to main content</a>')
    
    with gr.Tabs(elem_id="main-content") as tabs:
        with gr.Tab("🏠 Home", elem_id="home-tab"):
            create_home_page()
        
        with gr.Tab("🔬 Crop Disease Detection", elem_id="detection-tab"):
            create_crop_detection_demo()
        
        with gr.Tab("💰 Pricing", elem_id="pricing-tab"):
            free_button = create_pricing_page()
    
    # JavaScript functionality for tab switching
    free_button.click(
        lambda: None,
        outputs=[],
        js="""
        function() {
            // Find the detection tab and click it
            const tabs = document.querySelectorAll('button[role="tab"]');
            for (let tab of tabs) {
                if (tab.textContent.includes('Crop Disease Detection')) {
                    tab.click();
                    break;
                }
            }
        }
        """
    )

# Launch configuration
if __name__ == "__main__":
    app.launch(
        share=True,
        server_name="0.0.0.0",
        server_port=7860,
        show_error=True,
        show_api=False,
        favicon_path=None,
        app_kwargs={"docs_url": None}
    )