Update app.py

app.py

@@ -47,13 +47,18 @@ def create_hf_space():
         space_name = generate_space_name()
         full_repo_id = f"Space-Share/{space_name}"
 
-        create_repo(
-            repo_id=space_name,
-            repo_type="space",
-            space_sdk=space_type,
-            token=HF_TOKEN,
-            private=False
-        )
+        try:
+            # Attempt to create the repository
+            repo_info = create_repo(
+                repo_id=space_name,
+                repo_type="space",
+                space_sdk=space_type,
+                token=HF_TOKEN,
+                private=False,
+                exist_ok=True  # Allow creation even if the repo might exist (rare case)
+            )
+        except Exception as e:
+            return jsonify({"error": f"Failed to create repository: {str(e)}"}), 500
 
         # Handle multi-file uploads
         for filename, content in files.items():
@@ -65,13 +70,18 @@ def create_hf_space():
                 f.write(content)
 
             # Upload to the new Space
-            upload_file(
-                path_or_fileobj=f"temp_{filename}",
-                path_in_repo=filename,
-                repo_id=full_repo_id,
-                repo_type="space",
-                token=HF_TOKEN
-            )
+            try:
+                upload_file(
+                    path_or_fileobj=f"temp_{filename}",
+                    path_in_repo=filename,
+                    repo_id=full_repo_id,
+                    repo_type="space",
+                    token=HF_TOKEN
+                )
+            except Exception as e:
+                os.remove(f"temp_{filename}")
+                return jsonify({"error": f"Failed to upload file {filename}: {str(e)}"}), 500
+
             os.remove(f"temp_{filename}")
 
         # Add requirements.txt if not provided (basic defaults)
@@ -84,13 +94,17 @@ def create_hf_space():
             }.get(space_type, "")
             with open("temp_requirements.txt", "w") as f:
                 f.write(default_requirements)
-            upload_file(
-                path_or_fileobj="temp_requirements.txt",
-                path_in_repo="requirements.txt",
-                repo_id=full_repo_id,
-                repo_type="space",
-                token=HF_TOKEN
-            )
+            try:
+                upload_file(
+                    path_or_fileobj="temp_requirements.txt",
+                    path_in_repo="requirements.txt",
+                    repo_id=full_repo_id,
+                    repo_type="space",
+                    token=HF_TOKEN
+                )
+            except Exception as e:
+                os.remove("temp_requirements.txt")
+                return jsonify({"error": f"Failed to upload requirements.txt: {str(e)}"}), 500
             os.remove("temp_requirements.txt")
 
         # Special handling for Docker Spaces
@@ -105,13 +119,17 @@ CMD ["python", "app.py"]
 """
             with open("temp_Dockerfile", "w") as f:
                 f.write(default_dockerfile)
-            upload_file(
-                path_or_fileobj="temp_Dockerfile",
-                path_in_repo="Dockerfile",
-                repo_id=full_repo_id,
-                repo_type="space",
-                token=HF_TOKEN
-            )
+            try:
+                upload_file(
+                    path_or_fileobj="temp_Dockerfile",
+                    path_in_repo="Dockerfile",
+                    repo_id=full_repo_id,
+                    repo_type="space",
+                    token=HF_TOKEN
+                )
+            except Exception as e:
+                os.remove("temp_Dockerfile")
+                return jsonify({"error": f"Failed to upload Dockerfile: {str(e)}"}), 500
             os.remove("temp_Dockerfile")
 
         space_url = f"https://huggingface.co/spaces/Space-Share/{space_name}"