from __future__ import annotations

import gradio as gr

from drawing_intelligence.config import MODEL_CHOICES, SETTINGS
from drawing_intelligence.exporting import TABLE_COLUMNS
from drawing_intelligence.ui import (
    APP_CSS,
    export_excel_ui,
    export_json_ui,
    extract_ui,
    load_sample_ui,
    page_ui,
    preview_file_ui,
    select_row_ui,
    summary_ui,
)


def build_app() -> gr.Blocks:
    theme = gr.themes.Base(
        primary_hue=gr.themes.colors.red,
        neutral_hue=gr.themes.colors.slate,
        font=["Inter", "Segoe UI", "Arial", "sans-serif"],
        font_mono=["Roboto Mono", "Consolas", "monospace"],
    ).set(
        body_background_fill="#ffffff",
        body_text_color="#171d26",
        button_primary_background_fill="#b5121b",
        button_primary_background_fill_hover="#8f0d14",
        button_primary_border_color="#b5121b",
        block_border_color="#dce1e8",
        block_shadow="none",
        input_border_color="#d8dde5",
    )

    with gr.Blocks(
        title="Drawing Intelligence",
        theme=theme,
        css=APP_CSS,
        fill_width=True,
    ) as app:
        session = gr.State(value=None)

        gr.HTML(
            """
            <header class="app-header">
              <h1>Drawing Intelligence</h1>
              <p>Ballooned drawing parameter extraction</p>
            </header>
            """
        )

        with gr.Group(elem_classes=["control-strip"]):
            with gr.Row(equal_height=True):
                drawing = gr.File(
                    label="Upload ballooned drawing",
                    file_types=[".pdf", ".png", ".jpg", ".jpeg", ".tif", ".tiff"],
                    type="filepath",
                    scale=5,
                )
                model = gr.Dropdown(
                    choices=list(MODEL_CHOICES),
                    value=list(MODEL_CHOICES)[0],
                    label="Hugging Face model",
                    interactive=True,
                    scale=3,
                )
                dpi = gr.Dropdown(
                    choices=[150, 200, 300, 450, 600],
                    value=SETTINGS.default_dpi,
                    label="Render DPI",
                    scale=2,
                )
                extract = gr.Button(
                    "Extract parameters",
                    variant="primary",
                    elem_classes=["primary-action"],
                    scale=3,
                )
            with gr.Row():
                sample = gr.Button("Load demonstration drawing", size="sm")
                readiness = gr.Markdown(
                    "Upload a ballooned PDF or image to begin.",
                    show_label=False,
                    container=False,
                )

        with gr.Accordion("Advanced processing settings", open=False):
            with gr.Row():
                max_pages = gr.Slider(
                    1,
                    SETTINGS.max_pages,
                    value=min(3, SETTINGS.max_pages),
                    step=1,
                    label="Maximum pages",
                )
                tile_size = gr.Slider(
                    1024,
                    2560,
                    value=1800,
                    step=128,
                    label="AI tile size (pixels)",
                )
                overlap = gr.Slider(
                    0,
                    35,
                    value=18,
                    step=1,
                    label="Tile overlap (%)",
                )

        with gr.Group(elem_classes=["workspace-shell"]):
            with gr.Row(equal_height=True):
                with gr.Column(scale=11, min_width=480, elem_classes=["workspace-column"]):
                    with gr.Row():
                        page = gr.Slider(
                            1, 1, value=1, step=1, label="Page", visible=False
                        )
                    viewer = gr.Image(
                        label="Drawing review",
                        show_label=False,
                        type="pil",
                        height=685,
                        elem_id="drawing-viewer",
                    )
                with gr.Column(scale=9, min_width=460, elem_classes=["workspace-column"]):
                    gr.HTML('<div class="review-heading">Characteristics</div>')
                    characteristics = gr.Dataframe(
                        headers=TABLE_COLUMNS,
                        value=[],
                        interactive=True,
                        wrap=True,
                        show_label=False,
                        row_count=(10, "dynamic"),
                        col_count=(len(TABLE_COLUMNS), "fixed"),
                        pinned_columns=2,
                        column_widths=[
                            78,
                            62,
                            72,
                            120,
                            130,
                            240,
                            90,
                            95,
                            95,
                            95,
                            95,
                            78,
                            65,
                            115,
                            115,
                            115,
                            90,
                            105,
                            95,
                            160,
                            140,
                            220,
                            95,
                            115,
                            260,
                            120,
                        ],
                        elem_id="characteristics-table",
                    )
                    gr.HTML(
                        """
                        <div class="assurance-note">
                          AI output is never approved automatically. Confirm every requirement
                          against the controlled drawing before changing Status to Verified.
                        </div>
                        """
                    )

        with gr.Tabs():
            with gr.Tab("Drawing information"):
                metadata = gr.Markdown("No extraction has been run.")
            with gr.Tab("General notes"):
                notes = gr.Markdown("No extraction has been run.")
            with gr.Tab("Raw model output"):
                raw_output = gr.Code(language="json", value="[]", lines=18)

        with gr.Group(elem_classes=["bottom-strip"]):
            with gr.Row(equal_height=True):
                summary = gr.HTML(
                    '<div class="status-summary"><span class="status-count">0 extracted</span>'
                    '<span class="status-separator">·</span>'
                    '<span class="status-review">0 need review</span></div>'
                )
                excel_button = gr.Button("Export Excel", variant="secondary")
                json_button = gr.Button("Export JSON", variant="secondary")
            with gr.Row():
                excel_file = gr.File(label="Excel export", interactive=False)
                json_file = gr.File(label="JSON export", interactive=False)

        sample.click(
            load_sample_ui,
            outputs=[drawing, viewer, readiness],
            api_name=False,
        )
        drawing.change(
            preview_file_ui,
            inputs=[drawing],
            outputs=[viewer, readiness],
            api_name=False,
        )
        extract.click(
            extract_ui,
            inputs=[drawing, model, dpi, max_pages, tile_size, overlap],
            outputs=[
                viewer,
                characteristics,
                summary,
                metadata,
                notes,
                raw_output,
                session,
                page,
            ],
            api_name="extract",
        )
        page.change(
            page_ui,
            inputs=[page, session],
            outputs=[viewer],
            api_name=False,
        )
        characteristics.select(
            select_row_ui,
            inputs=[session, characteristics, page],
            outputs=[viewer, page],
            api_name=False,
        )
        characteristics.change(
            summary_ui,
            inputs=[characteristics, session],
            outputs=[summary],
            api_name=False,
        )
        excel_button.click(
            export_excel_ui,
            inputs=[characteristics, session],
            outputs=[excel_file],
            api_name="export_excel",
        )
        json_button.click(
            export_json_ui,
            inputs=[characteristics, session],
            outputs=[json_file],
            api_name="export_json",
        )

    return app.queue(default_concurrency_limit=1, max_size=12)


demo = build_app()


if __name__ == "__main__":
    demo.launch(show_api=False)