Spaces:
Running
Running
File size: 3,138 Bytes
13d2927 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 |
import gradio as gr
import subprocess
import os
import tempfile
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()
# 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:
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()
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()
|