Update app.py

app.py

@@ -5,7 +5,7 @@ import tempfile
 import datetime
 from huggingface_hub import upload_file
 
-def convert_image_to_dxf(image_file, output_folder=None):
+def convert_image_to_dxf(image_file, output_folder=None, use_lines=False):
     try:
         # Validate input image
         if image_file is None:
@@ -23,9 +23,19 @@ def convert_image_to_dxf(image_file, output_folder=None):
         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")
         
+        # Prepare the command arguments
+        command_args = ["./SimpleImageToDxfHavePass"]
+        
+        # Add --use-lines if the checkbox is checked
+        if use_lines:
+            command_args.append("--use-lines")
+        
+        # Add other arguments
+        command_args.extend([f"--imagePath={image_path}", f"--outputPath={output_path}", f"--debug-output={debug_output_path}"])
+        
         # Execute conversion command
         result = subprocess.run(
-            ["./SimpleImageToDxfHavePass", "--use-lines", f"--imagePath={image_path}", f"--outputPath={output_path}", f"--debug-output={debug_output_path}"],
+            command_args,
             check=True,
             capture_output=True
         )
@@ -94,6 +104,9 @@ def main():
                     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")
                 
+                # Checkbox to decide whether to use --use-lines
+                use_lines_checkbox = gr.Checkbox(label="Use --use-lines for conversion", value=False)  # Default is False
+                
                 # Outputs: Debug image and DXF file download link
                 with gr.Row():
                     debug_output = gr.Image(type="filepath", label="Debug Output Preview")
@@ -103,7 +116,7 @@ def main():
                 convert_btn = gr.Button("Convert to DXF")
                 convert_btn.click(
                     convert_image_to_dxf, 
-                    inputs=[image_input, output_folder], 
+                    inputs=[image_input, output_folder, use_lines_checkbox], 
                     outputs=[dxf_output, debug_output, dxf_output]
                 )