app.py

import gradio as gr
import time
import json
from uuid import uuid4
from utils.main_processor import get_statistics

# Sample questions for examples 
SAMPLE_QUESTIONS = {
    "Deforestation Analysis": [
        "What are the main deforestation hotspots in Ecuador?",
        "Show me deforestation trends in the uploaded area",
        "What commodities are driving deforestation in Guatemala?"
    ],
    "EUDR Compliance": [
        "What are the key EUDR requirements for coffee imports?",
        "How do I prove due diligence for my supply chain?",
        "What documentation is needed for EUDR compliance?"
    ],
    "Risk Assessment": [
        "What is the deforestation risk level in this region?",
        "How do I assess supply chain risks?",
        "What are the compliance deadlines?"
    ]
}

def handle_geojson_upload(file):
    """Handle GeoJSON file upload and process with API"""
    if file is not None:
        try:
            # Call the API to get statistics
            stats_result = get_statistics(file)
            
            # Create success message with API results
            success_message = "✅ Document submitted and processed successfully! You can now ask questions about this GeoJSON file."
            
            return (
                success_message,
                gr.update(visible=True),  # upload_status
                stats_result,  # api_results_df
                gr.update(visible=True)   # results_section
            )
        except Exception as e:
            error_message = f"❌ Error processing file: {str(e)}"
            return (
                error_message,
                gr.update(visible=True),  # upload_status
                None,  # api_results_df
                gr.update(visible=False)  # results_section
            )
    else:
        return (
            "❌ Please select a GeoJSON file to upload.",
            gr.update(visible=True),  # upload_status
            None,  # api_results_df
            gr.update(visible=False)  # results_section
        )

def start_chat(query, history):
    """Start a new chat interaction"""
    history = history + [(query, None)]
    return gr.update(interactive=False), gr.update(selected=1), history

def finish_chat():
    """Finish chat and reset input"""
    return gr.update(interactive=True, value="")

async def chat_response(query, history, method, country, uploaded_file):
    """Generate chat response based on method and inputs"""
    
    # Validate inputs based on method
    if method == "Upload GeoJSON":
        if uploaded_file is None:
            warning_message = "⚠️ **No GeoJSON file uploaded.** Please upload a GeoJSON file first."
            history[-1] = (query, warning_message)
            yield history, ""
            return
    else:  # "Talk to Reports"
        if not country:
            warning_message = "⚠️ **No country selected.** Please select a country to analyze reports."
            history[-1] = (query, warning_message)
            yield history, ""
            return
    
    # Simulate processing time
    response = ""
    if method == "Upload GeoJSON":
        full_response = f"Based on your uploaded GeoJSON file, I can help you analyze the deforestation patterns and EUDR compliance aspects in your area of interest. Your question: '{query}' is being processed against the geographic data you provided."
    else:
        full_response = f"Based on EUDR reports for {country}, I can help you understand deforestation patterns and compliance requirements. Your question: '{query}' is being analyzed against our {country} database."
    
    # Simulate streaming response
    words = full_response.split()
    for word in words:
        response += word + " "
        history[-1] = (query, response)
        yield history, "**Sources:** Sample source documents would appear here..."
        await asyncio.sleep(0.05)

def toggle_search_method(method):
    """Toggle between GeoJSON upload and country selection"""
    if method == "Upload GeoJSON":
        return (
            gr.update(visible=True),   # geojson_section
            gr.update(visible=False),  # reports_section
            gr.update(value=None),     # dropdown_country
        )
    else:  # "Talk to Reports"
        return (
            gr.update(visible=False),  # geojson_section 
            gr.update(visible=True),   # reports_section
            gr.update(),               # dropdown_country
        )

def change_sample_questions(key):
    """Update visible examples based on selected category"""
    keys = list(SAMPLE_QUESTIONS.keys())
    index = keys.index(key)
    visible_bools = [False] * len(keys)
    visible_bools[index] = True
    return [gr.update(visible=visible_bools[i]) for i in range(len(keys))]

# Set up Gradio Theme
theme = gr.themes.Base(
    primary_hue="green",
    secondary_hue="blue",
    font=[gr.themes.GoogleFont("Poppins"), "ui-sans-serif", "system-ui", "sans-serif"],
    text_size=gr.themes.utils.sizes.text_sm,
)

init_prompt = """
Hello, I am EUDR Q&A, an AI-powered conversational assistant designed to help you understand EU Deforestation Regulation compliance and analysis. I will answer your questions by using **EUDR reports and uploaded GeoJSON files**.

💡 **How to use (tabs on right)**
- **Data Sources**: Choose to either upload a GeoJSON file for analysis or talk to EUDR reports filtered by country.
- **Examples**: Select from curated example questions across different categories.
- **Sources**: View the content sources used to generate answers for fact-checking.

⚠️ For limitations and data collection information, please check the **Disclaimer** tab.
"""

with gr.Blocks(title="EUDR Q&A", theme=theme, css="style.css") as demo:
    
    # Main Chat Interface
    with gr.Tab("EUDR Q&A"):
        with gr.Row():
            # Left column - Chat interface (2/3 width)
            with gr.Column(scale=2):
                chatbot = gr.Chatbot(
                    value=[(None, init_prompt)],
                    show_copy_button=True,
                    show_label=False,
                    layout="panel",
                    avatar_images=(None, "🌳"),
                    height=500
                )
                
                # Feedback UI
                with gr.Column():
                    with gr.Row(visible=False) as feedback_row:
                        gr.Markdown("Was this response helpful?")
                        with gr.Row():
                            okay_btn = gr.Button("👍 Okay", size="sm")
                            not_okay_btn = gr.Button("👎 Not to expectations", size="sm")
                    feedback_thanks = gr.Markdown("Thanks for the feedback!", visible=False)
                
                # Input textbox
                with gr.Row():
                    textbox = gr.Textbox(
                        placeholder="Ask me anything about EUDR compliance or upload your GeoJSON for analysis!",
                        show_label=False,
                        scale=7,
                        lines=1,
                        interactive=True
                    )
            
            # Right column - Controls and tabs (1/3 width)
            with gr.Column(scale=1, variant="panel"):
                with gr.Tabs() as tabs:
                    
                    # Data Sources Tab
                    with gr.Tab("Data Sources", id=2):
                        search_method = gr.Radio(
                            choices=["Upload GeoJSON", "Talk to Reports"],
                            label="Choose data source",
                            info="Upload a GeoJSON file for analysis or select country-specific EUDR reports",
                            value="Upload GeoJSON",
                        )
                        
                        # GeoJSON Upload Section
                        with gr.Group(visible=True) as geojson_section:
                            uploaded_file = gr.File(
                                label="Upload GeoJSON File",
                                file_types=[".geojson", ".json"],
                                file_count="single"
                            )
                            submit_file_btn = gr.Button("Submit GeoJSON", variant="primary")
                            upload_status = gr.Markdown("", visible=False)
                            
                            # API Results Section
                            with gr.Group(visible=False) as results_section:
                                gr.Markdown("### Analysis Results")
                                api_results_df = gr.Dataframe(
                                    label="Statistics from API",
                                    interactive=False,
                                    wrap=True
                                )
                        
                        # Talk to Reports Section
                        with gr.Group(visible=False) as reports_section:
                            dropdown_country = gr.Dropdown(
                                ["Ecuador", "Guatemala"],
                                label="Select Country",
                                value=None,
                                interactive=True,
                            )
                    
                    # Examples Tab
                    with gr.Tab("Examples", id=0):
                        examples_hidden = gr.Textbox(visible=False)
                        
                        first_key = list(SAMPLE_QUESTIONS.keys())[0]
                        dropdown_samples = gr.Dropdown(
                            SAMPLE_QUESTIONS.keys(),
                            value=first_key,
                            interactive=True,
                            show_label=True,
                            label="Select a category of sample questions"
                        )
                        
                        # Create example sections
                        sample_groups = []
                        for i, (key, questions) in enumerate(SAMPLE_QUESTIONS.items()):
                            examples_visible = True if i == 0 else False
                            with gr.Row(visible=examples_visible) as group_examples:
                                gr.Examples(
                                    questions,
                                    [examples_hidden],
                                    examples_per_page=8,
                                    run_on_click=False,
                                )
                            sample_groups.append(group_examples)
                    
                    # Sources Tab
                    with gr.Tab("Sources", id=1):
                        sources_textbox = gr.HTML(
                            show_label=False,
                            value="Source documents will appear here after you ask a question..."
                        )
    
    # Guidelines Tab
    with gr.Tab("Guidelines"):
        gr.Markdown("""
        #### Welcome to EUDR Q&A! 
        
        This AI-powered assistant helps you understand EU Deforestation Regulation compliance and analyze geographic data.
        
        ## 💬 How to Ask Effective Questions
        
        | ❌ Less Effective | ✅ More Effective |
        |------------------|-------------------|
        | "What is deforestation?" | "What are the main deforestation hotspots in Ecuador?" |
        | "Tell me about compliance" | "What EUDR requirements apply to coffee imports from Guatemala?" |
        | "Show me data" | "What is the deforestation rate in the uploaded region?" |
        
        ## 🔍 Using Data Sources
        
        **Upload GeoJSON:** Upload your geographic data files for specific area analysis  
        **Talk to Reports:** Select Ecuador or Guatemala for country-specific EUDR analysis
        
        ## ⭐ Best Practices
        
        - Be specific about regions, commodities, or time periods
        - Ask one question at a time for clearer answers
        - Use follow-up questions to explore topics deeper
        - Provide context when possible
        
        ## 📊 API Analysis
        
        When you upload a GeoJSON file, our system automatically processes it through our analysis API to provide:
        - Statistical analysis of the geographic area
        - Deforestation risk assessment
        - Compliance indicators
        - Area calculations and boundaries
        """)
    
    # About Tab
    with gr.Tab("About"):
        gr.Markdown("""
        ## About EUDR Q&A
        
        The **EU Deforestation Regulation (EUDR)** requires companies to ensure that specific commodities 
        placed on the EU market are deforestation-free and legally produced.
        
        This AI-powered tool helps stakeholders:
        - Understand EUDR compliance requirements
        - Analyze geographic deforestation data
        - Assess supply chain risks
        - Navigate complex regulatory landscapes
        
        **Developed by GIZ** to enhance accessibility and understanding of EUDR requirements 
        through advanced AI and geographic data processing capabilities.
        
        ### Key Features:
        - Interactive analysis of uploaded GeoJSON files
        - Automated statistical analysis via API integration
        - Country-specific EUDR compliance guidance
        - Real-time question answering with source citations
        - User-friendly interface for complex regulatory information
        """)
    
    # Disclaimer Tab
    with gr.Tab("Disclaimer"):
        gr.Markdown("""
        ## Important Disclaimers
        
        ⚠️ **Scope & Limitations:**
        - This tool is designed for EUDR compliance assistance and geographic data analysis
        - Responses should not be considered official legal or compliance advice
        - Always consult qualified professionals for official compliance decisions
        
        ⚠️ **Data & Privacy:**
        - Uploaded GeoJSON files are processed temporarily for analysis
        - Files are sent to our analysis API for statistical processing
        - We collect usage statistics to improve the tool
        - No sensitive geographic data is permanently stored
        
        ⚠️ **AI Limitations:**
        - Responses are AI-generated and may contain inaccuracies
        - The tool is a prototype under continuous development
        - Always verify important information with authoritative sources
        
        **Data Collection:** We collect questions, answers, feedback, and anonymized usage statistics 
        to improve tool performance based on legitimate interest in service enhancement.
        
        By using this tool, you acknowledge these limitations and agree to use responses responsibly.
        """)
    
    # Event Handlers
    
    # Toggle search method
    search_method.change(
        fn=toggle_search_method,
        inputs=[search_method],
        outputs=[geojson_section, reports_section, dropdown_country]
    )
    
    # File upload with API integration
    submit_file_btn.click(
        fn=handle_geojson_upload,
        inputs=[uploaded_file],
        outputs=[upload_status, upload_status, api_results_df, results_section]
    )
    
    # Chat functionality
    textbox.submit(
        start_chat,
        [textbox, chatbot],
        [textbox, tabs, chatbot],
        queue=False
    ).then(
        chat_response,
        [textbox, chatbot, search_method, dropdown_country, uploaded_file],
        [chatbot, sources_textbox]
    ).then(
        lambda: gr.update(visible=True),
        outputs=[feedback_row]
    ).then(
        finish_chat,
        outputs=[textbox]
    )
    
    # Examples functionality
    examples_hidden.change(
        start_chat,
        [examples_hidden, chatbot],
        [textbox, tabs, chatbot],
        queue=False
    ).then(
        chat_response,
        [examples_hidden, chatbot, search_method, dropdown_country, uploaded_file],
        [chatbot, sources_textbox]
    ).then(
        lambda: gr.update(visible=True),
        outputs=[feedback_row]
    ).then(
        finish_chat,
        outputs=[textbox]
    )
    
    # Sample questions dropdown
    dropdown_samples.change(
        change_sample_questions,
        [dropdown_samples],
        sample_groups
    )
    
    # Feedback buttons
    okay_btn.click(
        lambda: (gr.update(visible=False), gr.update(visible=True)),
        outputs=[feedback_row, feedback_thanks]
    )
    
    not_okay_btn.click(
        lambda: (gr.update(visible=False), gr.update(visible=True)),
        outputs=[feedback_row, feedback_thanks]
    )

# Launch the app
if __name__ == "__main__":
    demo.launch()