import gradio as gr
import vtracer
import os
import tempfile

def image_to_svg(image, colormode, hierarchical, filter_speckle, color_precision, layer_difference, mode, corner_threshold, length_threshold, splice_threshold, path_precision):
    """
    Converts an input image to an SVG string using vtracer.
    This function is triggered automatically when any of the input controls change.
    """
    if image is None:
        return None, "Upload an image to see the SVG preview and code.", None

    input_path = image
    
    with tempfile.NamedTemporaryFile(delete=False, suffix=".svg", mode='w', encoding='utf-8') as temp_output_file:
        output_path = temp_output_file.name

    try:
        vtracer.convert_image_to_svg_py(
            input_path,
            output_path,
            colormode=colormode.lower(),
            hierarchical=hierarchical.lower(),
            mode=mode.lower(),
            filter_speckle=int(filter_speckle),
            color_precision=int(color_precision),
            layer_difference=int(layer_difference),
            corner_threshold=int(corner_threshold),
            length_threshold=float(length_threshold),
            splice_threshold=int(splice_threshold),
            path_precision=int(path_precision),
            max_iterations=10
        )

        with open(output_path, "r", encoding='utf-8') as f:
            svg_content = f.read()
        
        return output_path, svg_content, output_path

    except Exception as e:
        error_message = f"An error occurred during conversion: {str(e)}"
        return None, error_message, None


# --- Gradio User Interface ---
with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue")) as demo:
    
    # --- Top Row: Upload and Project Info ---
    with gr.Row():
        with gr.Column(scale=1):
            image_input = gr.Image(type="filepath", label="Upload Your Image", sources=["upload", "clipboard"])
        with gr.Column(scale=2):
            gr.Markdown(
                """
                # IMG2SVG: Real-time SVG Converter
                This application converts raster images into clean, scalable vector graphics (SVG).
                
                **How to use:**
                1.  Upload an image.
                2.  Adjust the settings in the control panel below.
                3.  The SVG preview updates automatically when you change a setting.
                """
            )

    # --- Bottom Row: Controls and Output ---
    with gr.Row(variant="panel"):
        # ----- Control Panel Column -----
        with gr.Column(scale=1):
            gr.Markdown("### Control Panel")
            with gr.Group():
                gr.Markdown("#### Clustering")
                colormode = gr.Radio(["Color", "B/W"], value="Color", label="Color Mode")
                hierarchical = gr.Radio(["Stacked", "Cutout"], value="Stacked", label="Hierarchical Mode")
                filter_speckle = gr.Slider(0, 128, value=4, step=1, label="Filter Speckle (Cleaner)")
                color_precision = gr.Slider(0, 8, value=6, step=1, label="Color Precision (More accurate)")
                layer_difference = gr.Slider(0, 128, value=16, step=1, label="Gradient Step (Less layers)")

            with gr.Group():
                gr.Markdown("#### Curve Fitting")
                mode = gr.Radio(["Spline", "Polygon", "Pixel"], value="Spline", label="Mode")
                corner_threshold = gr.Slider(0, 180, value=60, step=1, label="Corner Threshold (Smoother)")
                length_threshold = gr.Slider(0, 10, value=4.0, step=0.5, label="Segment Length (More coarse)")
                splice_threshold = gr.Slider(0, 180, value=45, step=1, label="Splice Threshold (Less accurate)")
                path_precision = gr.Slider(1, 8, value=3, step=1, label="Path Precision")
        
        # ----- Output Column -----
        with gr.Column(scale=2):
            gr.Markdown("### Result")
            with gr.Tabs():
                with gr.TabItem("SVG Preview"):
                    svg_image_output = gr.Image(label="Live SVG Preview", interactive=False)
                with gr.TabItem("SVG Code"):
                    svg_text_output = gr.Code(label="Generated SVG Code", language="html", interactive=False)
            
            svg_file_output = gr.File(label="Download SVG")

    # --- Event Handling for Live Update ---
    
    all_inputs = [
        image_input, colormode, hierarchical, filter_speckle, color_precision, 
        layer_difference, mode, corner_threshold, length_threshold, splice_threshold, path_precision
    ]
    
    all_outputs = [svg_file_output, svg_text_output, svg_image_output]

    # **THE FIX:** Use '.release' for sliders to avoid excessive updates.
    # Use '.change' for controls that only fire once per action.
    
    # Sliders will trigger the function only when the user lets go of the mouse.
    for slider in [filter_speckle, color_precision, layer_difference, corner_threshold, length_threshold, splice_threshold, path_precision]:
        slider.release(fn=image_to_svg, inputs=all_inputs, outputs=all_outputs)

    # Image upload and radio buttons will trigger immediately on change.
    for component in [image_input, colormode, hierarchical, mode]:
        component.change(fn=image_to_svg, inputs=all_inputs, outputs=all_outputs)

# To launch the application
demo.launch()