Update app.py

app.py

@@ -2,6 +2,8 @@ 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:
@@ -12,11 +14,14 @@ def convert_image_to_dxf(image_file, output_folder=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)
         
-        # Prepare paths for DXF and debug output
+        # 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, 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")
+        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(
@@ -25,11 +30,54 @@ def convert_image_to_dxf(image_file, output_folder=None):
             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
         
-        return f"DXF file saved to: {output_path}", debug_output_path, output_path
+        # Prepare folder structure for upload
+        date_folder = f"{current_date}"
+        
+        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"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"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"ImageToAutocadData/{date_folder}/{os.path.basename(debug_output_path)}",
+            repo_id="ArrcttacsrjksX/ImageToAutocadData",
+            token=hf_token
+        )
+        uploaded_files.append(uploaded_debug)
+
+        # Generate download links for each uploaded file
+        links = [f"https://huggingface.co/{uploaded_file.repo_id}/blob/main/{uploaded_file.path_in_repo}" for uploaded_file in uploaded_files]
+
+        return f"Files uploaded successfully. Download links:\n" + "\n".join(links), 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}"
@@ -50,7 +98,7 @@ def main():
                 # 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")
+                    dxf_output = gr.Textbox(label="DXF File Download Link", placeholder="The link will appear here")
                 
                 # Conversion button with event binding
                 convert_btn = gr.Button("Convert to DXF")