Update app.py

app.py

@@ -18,6 +18,7 @@ from g4f.client import Client
 
 import tempfile
 from huggingface_hub import HfApi, login
+import threading
 
 client = Client()
 
@@ -426,6 +427,18 @@ def make_text():
         os.unlink(temp_file_path)
 
 
+def delete_file_after_delay(file_path, delay_seconds):
+    def delete_file():
+        time.sleep(delay_seconds)
+        try:
+            os.unlink(file_path)
+            print(f"Temporary file {file_path} deleted after {delay_seconds} seconds.")
+        except Exception as e:
+            print(f"Error deleting temporary file {file_path}: {str(e)}")
+
+    thread = threading.Thread(target=delete_file)
+    thread.start()
+
 @app.route('/upload-image-dataset', methods=['POST'])
 def upload_image_dataset():
     data = request.json
@@ -433,6 +446,7 @@ def upload_image_dataset():
     file_name = data.get('fileName')
     repo_id = data.get('repoId')
     repo_type = data.get('repoType', 'dataset')
+    is_forwarded = data.get('is_forwarded', False)
 
     if not url or not file_name or not repo_id:
         return jsonify({"error": "Parameters 'url', 'fileName', and 'repoId' are required"}), 400
@@ -442,7 +456,11 @@ def upload_image_dataset():
 
     try:
         # Download the image
-        response = requests.get(url)
+        headers = {}
+        if is_forwarded:
+            headers['Authorization'] = f'Bearer {HF_TOKEN}'
+        
+        response = requests.get(url, headers=headers)
         response.raise_for_status()
 
         # Create a temporary file
@@ -450,6 +468,9 @@ def upload_image_dataset():
             temp_file.write(response.content)
             temp_file_path = temp_file.name
 
+        # Schedule file deletion after 1 minute
+        delete_file_after_delay(temp_file_path, 60)
+
         # Upload the file to Hugging Face
         api.upload_file(
             path_or_fileobj=temp_file_path,
@@ -458,9 +479,6 @@ def upload_image_dataset():
             repo_type=repo_type,
         )
 
-        # Clean up the temporary file
-        os.unlink(temp_file_path)
-
         return jsonify({
             "message": f"File '{file_name}' uploaded successfully to {repo_id} ({repo_type})",
             "size": len(response.content)