Update app.py

app.py

@@ -4,7 +4,7 @@ import os
 import tempfile
 import datetime
 import shutil
-from huggingface_hub import upload_file
+from huggingface_hub import upload_file, hf_hub_download
 
 def convert_image_to_dxf(image_file, output_folder=None, use_lines=False):
     try:
@@ -50,10 +50,6 @@ def convert_image_to_dxf(image_file, output_folder=None, use_lines=False):
         if not os.path.exists(output_path):
             return "Conversion failed: DXF file was not created.", None, None
         
-        # Check if the output DXF file exists
-        if not os.path.isfile(output_path):
-            return f"Error: The DXF file was not found at {output_path}.", None, None
-        
         # Clean up temporary folder
         if output_folder and os.path.isdir(output_folder):
             shutil.rmtree(output_folder)
@@ -107,6 +103,19 @@ def convert_image_to_dxf(image_file, output_folder=None, use_lines=False):
         error_msg = f"Error converting image to DXF: {e.stderr.decode('utf-8') if e.stderr else e}"
         return error_msg, None, None
 
+# Function to download the DXF file from Hugging Face
+def download_from_huggingface(file_name, folder="datasets/ArrcttacsrjksX/ImageToAutocadData"):
+    try:
+        # Path on Hugging Face Repo
+        file_path = hf_hub_download(
+            repo_id="ArrcttacsrjksX/ImageToAutocadData",
+            filename=f"{folder}/{file_name}",
+            use_auth_token=os.getenv("HF_TOKEN")  # Ensure token is available in the environment
+        )
+        return file_path
+    except Exception as e:
+        return str(e)
+
 def main():
     with gr.Blocks() as demo:
         with gr.Tabs():
@@ -135,9 +144,28 @@ def main():
                     outputs=[dxf_output, debug_output, dxf_output]
                 )
             
+            # Download tab for downloading from Hugging Face
+            with gr.TabItem("Download DXF"):
+                gr.Markdown("# Download DXF File from Hugging Face")
+                
+                # Input for filename
+                file_name_input = gr.Textbox(label="Enter DXF file name to download", placeholder="e.g., 2024-11-12_15-10-30_output.dxf")
+                
+                # Button to trigger download
+                download_btn = gr.Button("Download DXF")
+                
+                # Output for download link
+                download_link = gr.File(label="Download DXF")
+                
+                download_btn.click(
+                    download_from_huggingface,
+                    inputs=[file_name_input],
+                    outputs=[download_link]
+                )
+
             # 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.")
+                gr.Markdown("This Gradio app allows users to convert an image to a DXF file using the SimpleImageToDxfHavePass command-line tool and download DXF files from Hugging Face.")
 
     demo.launch(share=True)