Update app.py

app.py

@@ -1,31 +1,70 @@
 import gradio as gr
 import subprocess
 import os
+import tempfile
 
-def convert_image_to_dxf(image_path, output_path, debug_output):
+def convert_image_to_dxf(image_file, output_folder=None):
     try:
-        subprocess.run(["./SimpleImageToDxfHavePass", "--use-lines", f"--imagePath={image_path}", f"--outputPath={output_path}", f"--debug-output={debug_output}"], check=True)
-        return f"DXF file saved to: {output_path}", debug_output, output_path
+        # 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()
+        
+        # Prepare paths for DXF and debug output
+        image_path = image_file
+        output_path = os.path.join(output_folder, os.path.splitext(os.path.basename(image_path))[0] + ".dxf")
+        debug_output_path = os.path.join(output_folder, os.path.splitext(os.path.basename(image_path))[0] + "_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
+        )
+        
+        # Check if conversion was successful
+        if not os.path.exists(output_path):
+            return "Conversion failed: DXF file was not created.", None, None
+        
+        return f"DXF file saved to: {output_path}", debug_output_path, output_path
+    
     except subprocess.CalledProcessError as e:
-        return f"Error converting image to DXF: {e}", None, None
+        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(label="Input Image")
-                    dxf_output = gr.File(label="DXF Output")
+                    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(label="Debug Output")
-                    dxf_preview = gr.HTML(label="DXF Preview")
+                    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, dxf_output, debug_output], outputs=[dxf_output, debug_output, dxf_preview])
+                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()
 
 if __name__ == "__main__":
-    main()
+    main()