import gradio as gr
import subprocess
import os
import shutil
from PIL import Image

def generate_sketch(concept, temperature):
    save_path = "results/test"
    command = [
        "python", "sketch/gen_sketch.py",
        "--concept_to_draw", concept,
        "--temperature", str(temperature),
        "--path2save", save_path
    ]

    try:
        subprocess.run(command, check=True)

        img_path = os.path.join(save_path, concept, f"{concept}_canvas.png")

        if not os.path.exists(img_path):
            return f"Sketch generation succeeded but image not found: {img_path}", None

        return "Sketch generated successfully!", img_path

    except subprocess.CalledProcessError as e:
        return f"Sketch generation failed: {str(e)}", None


def ui():
    with gr.Blocks() as demo:
        gr.Markdown("## 🎨 Sketch Generator")

        with gr.Row():
            concept = gr.Textbox(label="What Should I sketch", placeholder="e.g. Taj Mahal")
            temperature = gr.Slider(minimum=0.0, maximum=1.0, value=0.3, step=0.05, label="Temperature")
            generate_btn = gr.Button("Generate Sketch")

        status = gr.Textbox(label="Status")
        output_image = gr.Image(label="Generated Sketch", type="filepath")

        generate_btn.click(
            fn=generate_sketch,
            inputs=[concept, temperature],
            outputs=[status, output_image]
        )

    return demo


if __name__ == "__main__":
    ui().launch(share=True)