import gradio as gr
import subprocess
import os
import tempfile
import datetime
from huggingface_hub import upload_file

def convert_image_to_dxf(image_file, output_folder=None):
    try:
        # Validate input image
        if image_file is None:
            return "No image provided", None, None
        
        # Define output folder, using a temp directory if none is specified
        if not output_folder:
            output_folder = tempfile.mkdtemp()
        else:
            os.makedirs(output_folder, exist_ok=True)
        
        # Generate the date-based output file name
        current_date = datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
        image_path = image_file
        output_path = os.path.join(output_folder, f"{current_date}_output.dxf")
        debug_output_path = os.path.join(output_folder, f"{current_date}_debug.png")
        
        # Execute conversion command
        result = subprocess.run(
            ["./SimpleImageToDxfHavePass", "--use-lines", f"--imagePath={image_path}", f"--outputPath={output_path}", f"--debug-output={debug_output_path}"],
            check=True,
            capture_output=True
        )
        
        # Log stdout for debugging
        print(result.stdout.decode('utf-8'))
        
        # Check if conversion was successful
        if not os.path.exists(output_path):
            return "Conversion failed: DXF file was not created.", None, None
        
        # Prepare folder structure for upload
        date_folder = f"{current_date}"
        
        # Hugging Face token
        hf_token = os.getenv("HF_TOKEN")
        if not hf_token:
            return "Hugging Face token not found", None, None
        
        # Upload input image, output DXF, and debug image to Hugging Face in a date-based folder
        uploaded_files = []
        
        # Upload input image
        uploaded_input = upload_file(
            path_or_fileobj=image_path,
            path_in_repo=f"datasets/ArrcttacsrjksX/ImageToAutocadData/{date_folder}/{os.path.basename(image_path)}",
            repo_id="ArrcttacsrjksX/ImageToAutocadData",
            token=hf_token
        )
        uploaded_files.append(uploaded_input)

        # Upload DXF output
        uploaded_dxf = upload_file(
            path_or_fileobj=output_path,
            path_in_repo=f"datasets/ArrcttacsrjksX/ImageToAutocadData/{date_folder}/{os.path.basename(output_path)}",
            repo_id="ArrcttacsrjksX/ImageToAutocadData",
            token=hf_token
        )
        uploaded_files.append(uploaded_dxf)

        # Upload debug image
        uploaded_debug = upload_file(
            path_or_fileobj=debug_output_path,
            path_in_repo=f"datasets/ArrcttacsrjksX/ImageToAutocadData/{date_folder}/{os.path.basename(debug_output_path)}",
            repo_id="ArrcttacsrjksX/ImageToAutocadData",
            token=hf_token
        )
        uploaded_files.append(uploaded_debug)

        # Return files directly for download in Gradio interface
        return output_path, debug_output_path, uploaded_files
    
    except subprocess.CalledProcessError as e:
        error_msg = f"Error converting image to DXF: {e.stderr.decode('utf-8') if e.stderr else e}"
        return error_msg, None, None

def main():
    with gr.Blocks() as demo:
        with gr.Tabs():
            # Tab for conversion
            with gr.TabItem("Image to DXF"):
                gr.Markdown("# SimpleImageToDxfHavePass")
                
                # Input row for image and optional output folder
                with gr.Row():
                    image_input = gr.Image(type="filepath", label="Input Image (PNG/JPG)")
                    output_folder = gr.Textbox(label="Output Folder (optional)", placeholder="Leave blank to use a temporary folder")
                
                # Outputs: Debug image and DXF file download link
                with gr.Row():
                    debug_output = gr.Image(type="filepath", label="Debug Output Preview")
                    dxf_output = gr.File(label="DXF File Download")
                
                # Conversion button with event binding
                convert_btn = gr.Button("Convert to DXF")
                convert_btn.click(
                    convert_image_to_dxf, 
                    inputs=[image_input, output_folder], 
                    outputs=[dxf_output, debug_output, dxf_output]
                )
            
            # About tab
            with gr.TabItem("About"):
                gr.Markdown("This Gradio app allows users to convert an image to a DXF file using the SimpleImageToDxfHavePass command-line tool.")

    demo.launch(share=True)

if __name__ == "__main__":
    try:
        subprocess.run(['chmod', '+x', './SimpleImageToDxfHavePass'], check=True)
    except subprocess.CalledProcessError as e:
        print(f"Error setting permissions: {e}")
        exit(1)
    main()