Update app.py

app.py

@@ -7,11 +7,6 @@ from huggingface_hub import upload_file
 
 def convert_image_to_dxf(image_file, output_folder=None, use_lines=False):
     try:
-        # Ensure HF_TOKEN environment variable is set
-        hf_token = os.getenv("HF_TOKEN")
-        if not hf_token:
-            return "Hugging Face token not found", None, None
-        
         # Validate input image
         if image_file is None:
             return "No image provided", None, None
@@ -54,7 +49,12 @@ def convert_image_to_dxf(image_file, output_folder=None, use_lines=False):
         
         # Prepare folder structure for upload
         date_folder = f"{current_date}"
-
+        
+        # Hugging Face token
+        hf_token = os.getenv("HF_TOKEN")
+        if not hf_token:
+            return "Hugging Face token not found", None, None
+        
         # Upload input image, output DXF, and optionally debug image to Hugging Face in a date-based folder
         uploaded_files = []
         
@@ -62,8 +62,8 @@ def convert_image_to_dxf(image_file, output_folder=None, use_lines=False):
         uploaded_input = upload_file(
             path_or_fileobj=image_path,
             path_in_repo=f"datasets/ArrcttacsrjksX/ImageToAutocadData/{date_folder}/{os.path.basename(image_path)}",
-            repo_id="ArrcttacsrjksX/ImageToAutocadData",  # Repository name
-            token=hf_token  # Use the Hugging Face token for authorization
+            repo_id="ArrcttacsrjksX/ImageToAutocadData",
+            token=hf_token
         )
         uploaded_files.append(uploaded_input)
 
@@ -71,8 +71,8 @@ def convert_image_to_dxf(image_file, output_folder=None, use_lines=False):
         uploaded_dxf = upload_file(
             path_or_fileobj=output_path,
             path_in_repo=f"datasets/ArrcttacsrjksX/ImageToAutocadData/{date_folder}/{os.path.basename(output_path)}",
-            repo_id="ArrcttacsrjksX/ImageToAutocadData",  # Repository name
-            token=hf_token  # Use the Hugging Face token for authorization
+            repo_id="ArrcttacsrjksX/ImageToAutocadData",
+            token=hf_token
         )
         uploaded_files.append(uploaded_dxf)
 
@@ -81,8 +81,8 @@ def convert_image_to_dxf(image_file, output_folder=None, use_lines=False):
             uploaded_debug = upload_file(
                 path_or_fileobj=debug_output_path,
                 path_in_repo=f"datasets/ArrcttacsrjksX/ImageToAutocadData/{date_folder}/{os.path.basename(debug_output_path)}",
-                repo_id="ArrcttacsrjksX/ImageToAutocadData",  # Repository name
-                token=hf_token  # Use the Hugging Face token for authorization
+                repo_id="ArrcttacsrjksX/ImageToAutocadData",
+                token=hf_token
             )
             uploaded_files.append(uploaded_debug)
 
@@ -93,7 +93,6 @@ 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
 
-
 def main():
     with gr.Blocks() as demo:
         with gr.Tabs():
@@ -128,7 +127,6 @@ def main():
 
     demo.launch(share=True)
 
-
 if __name__ == "__main__":
     try:
         subprocess.run(['chmod', '+x', './SimpleImageToDxfHavePass'], check=True)