import gradio as gr
import os
import subprocess
from gradio_app.project_info import (
    CONTENT_DESCRIPTION,
    CONTENT_IN_1, CONTENT_IN_2,
    CONTENT_OUT_1, CONTENT_OUT_2
)
from gradio_app.inference import run_inference
from gradio_app.examples import load_examples, select_example

def run_setup_script():
    setup_script = os.path.join(os.path.dirname(__file__), "gradio_app", "setup_scripts.py")
    try:
        result = subprocess.run(["python", setup_script], capture_output=True, text=True, check=True)
        return result.stdout
    except subprocess.CalledProcessError as e:
        print(f"Setup script failed with error: {e.stderr}")
        return f"Setup script failed: {e.stderr}"

def create_gui():
    # Load CSS and JavaScript
    custom_css = open(os.path.join(os.path.dirname(__file__), "gradio_app", "static", "styles.css")).read()
    outer_scale_warning = open(os.path.join(os.path.dirname(__file__), "gradio_app", "static", "outer_scale_warning.js")).read()

    with gr.Blocks(css=custom_css) as demo:
        gr.Markdown("# Anime Super Resolution 🖼️")
        gr.Markdown(CONTENT_DESCRIPTION)
        gr.Markdown(CONTENT_IN_1)
        gr.HTML(CONTENT_IN_2)
        with gr.Row():
            with gr.Column(scale=2):
                input_image = gr.Image(
                    type="filepath",
                    label="Input Image",
                    elem_classes="input-image"
                )
                model_id = gr.Textbox(
                    label="Model ID",
                    value="danhtran2mind/Real-ESRGAN-Anime-finetuning"
                )
                outer_scale = gr.Slider(
                    minimum=1,
                    maximum=8,
                    step=1,
                    value=2,
                    label="Outer Scale",
                    elem_id="outer-scale-slider"
                )
                warning_text = gr.HTML(elem_id="warning-text")
                gr.Markdown(
                    "**Note:** For optimal output quality, set `Outer Scale` to a value between 1 and 4. "
                    "**Values greater than 4 are not recommended**. "
                    "Please ensure `Outer Scale` is greater than or equal to `Inner Scale` (default: 4)."
                )
                examples_data = load_examples()
                if not examples_data:
                    gr.Warning("No examples found. Check the 'apps/assets/examples' directory.")
                submit_button = gr.Button("Run Inference")

            with gr.Column(scale=3):
                output_image = gr.Image(
                    label="Output Image",
                    elem_classes="output-image"
                )
                output_text = gr.Textbox(label="Status")

        outer_scale.change(
            fn=lambda x: x,
            inputs=outer_scale,
            outputs=outer_scale,
            js=outer_scale_warning
        )

        gr.Examples(
            examples=[[input_img, output_img, outer_scale] for input_img, output_img, outer_scale in examples_data],
            inputs=[input_image, outer_scale],
            label="Example Inputs",
            examples_per_page=4,
            fn=select_example,
            outputs=[input_image, outer_scale, output_image, output_text]
        )

        submit_button.click(
            fn=run_inference,
            inputs=[input_image, model_id, outer_scale],
            outputs=[output_image, output_text]
        )
        gr.HTML(CONTENT_OUT_1)
        gr.Markdown(CONTENT_OUT_2)

    return demo

if __name__ == "__main__":
    run_setup_script()
    demo = create_gui()
    demo.launch(debug=True)