Update app.py

app.py

@@ -1,39 +1,27 @@
 from flask import Flask, request, jsonify, Response
 import os
 import json
-from huggingface_hub import HfApi, create_repo, upload_file
+from huggingface_hub import HfApi, create_repo, upload_file, login
 import random
 import string
-from flask_httpauth import HTTPTokenAuth
 
 # Import documentation from assembler_docs.py
 from assembler_docs import DOCUMENTATION
 
 app = Flask(__name__)
 
-# Set up HTTP Token Authentication
-auth = HTTPTokenAuth(scheme='Bearer')
-
-# Hugging Face API token (set in Space settings or provided in request)
+# Hugging Face API token (set in Space settings)
 HF_TOKEN = os.getenv("HF_TOKEN")
 if not HF_TOKEN:
     raise ValueError("HF_TOKEN not set. Add it in Space settings.")
 
-hf_api = HfApi()
+# Log in to Hugging Face Hub with the token
+try:
+    login(token=HF_TOKEN)
+except Exception as e:
+    raise ValueError(f"Failed to log in to Hugging Face Hub: {str(e)}")
 
-# Authentication function to validate the token
-@auth.verify_token
-def verify_token(token):
-    if not token:
-        return False
-    try:
-        # Attempt to verify the token by checking if it can access the user's profile or Space-Share
-        user_info = hf_api.whoami(token=token)
-        # Optionally, check if the token has write access to Space-Share
-        # This is a basic check; you might want to add more specific permission checks
-        return bool(user_info and "Space-Share" in user_info.get('orgs', [])) or token == HF_TOKEN
-    except Exception:
-        return False
+hf_api = HfApi()
 
 def generate_space_name():
     """Generate a unique Space name."""
@@ -41,17 +29,12 @@ def generate_space_name():
     return f"GeneratedSpace-{random_suffix}"
 
 @app.route('/create-space', methods=['POST'])
-@auth.login_required
 def create_hf_space():
     try:
-        # Use the authenticated token for Hugging Face operations
-        token = auth.current_user()
-        hf_api = HfApi(token=token)
-
         # Parse JSON input
         data = request.get_json()
         if not data:
-            return jsonify({"error": "No JSON data provided"}), 401
+            return jsonify({"error": "No JSON data provided"}), 400
 
         # Extract parameters
         space_type = data.get("space_type", "gradio")  # Default to gradio if not specified
@@ -59,29 +42,38 @@ def create_hf_space():
         params = data.get("parameters", {})  # Optional parameters
 
         if not files:
-            return jsonify({"error": "No files provided in JSON"}), 401
+            return jsonify({"error": "No files provided in JSON"}), 400
 
         # Validate space_type
         valid_space_types = ["gradio", "static", "docker", "streamlit"]
         if space_type not in valid_space_types:
-            return jsonify({"error": f"Invalid space_type. Must be one of {valid_space_types}"}), 401
+            return jsonify({"error": f"Invalid space_type. Must be one of {valid_space_types}"}), 400
 
         # Create a unique Space name and repo under Space-Share namespace
         space_name = generate_space_name()
         full_repo_id = f"Space-Share/{space_name}"
 
         try:
-            # Attempt to create the repository
+            # Attempt to create the repository with explicit verification
             repo_info = create_repo(
                 repo_id=space_name,
                 repo_type="space",
                 space_sdk=space_type,
-                token=token,
+                token=HF_TOKEN,
                 private=False,
-                exist_ok=True  # Allow creation even if the repo might exist (rare case)
+                exist_ok=True  # Allow creation even if the repo might exist
             )
+            print(f"Repository created: {full_repo_id}")
+        except Exception as e:
+            return jsonify({"error": f"Failed to create repository {full_repo_id}: {str(e)}"}), 500
+
+        # Verify repository existence before uploading
+        try:
+            repo_exists = hf_api.repo_exists(repo_id=full_repo_id, repo_type="space")
+            if not repo_exists:
+                return jsonify({"error": f"Repository {full_repo_id} does not exist after creation attempt."}), 500
         except Exception as e:
-            return jsonify({"error": f"Failed to create repository: {str(e)}"}), 500
+            return jsonify({"error": f"Failed to verify repository {full_repo_id}: {str(e)}"}), 500
 
         # Handle multi-file uploads
         for filename, content in files.items():
@@ -99,8 +91,9 @@ def create_hf_space():
                     path_in_repo=filename,
                     repo_id=full_repo_id,
                     repo_type="space",
-                    token=token
+                    token=HF_TOKEN
                 )
+                print(f"Uploaded file: {filename}")
             except Exception as e:
                 os.remove(f"temp_{filename}")
                 return jsonify({"error": f"Failed to upload file {filename}: {str(e)}"}), 500
@@ -123,8 +116,9 @@ def create_hf_space():
                     path_in_repo="requirements.txt",
                     repo_id=full_repo_id,
                     repo_type="space",
-                    token=token
+                    token=HF_TOKEN
                 )
+                print("Uploaded requirements.txt")
             except Exception as e:
                 os.remove("temp_requirements.txt")
                 return jsonify({"error": f"Failed to upload requirements.txt: {str(e)}"}), 500
@@ -148,8 +142,9 @@ CMD ["python", "app.py"]
                     path_in_repo="Dockerfile",
                     repo_id=full_repo_id,
                     repo_type="space",
-                    token=token
+                    token=HF_TOKEN
                 )
+                print("Uploaded Dockerfile")
             except Exception as e:
                 os.remove("temp_Dockerfile")
                 return jsonify({"error": f"Failed to upload Dockerfile: {str(e)}"}), 500
@@ -163,12 +158,11 @@ CMD ["python", "app.py"]
         }), 200
 
     except json.JSONDecodeError:
-        return jsonify({"error": "Invalid JSON format"}), 401
+        return jsonify({"error": "Invalid JSON format"}), 400
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
 @app.route('/docs', methods=['GET'])
-@auth.login_required
 def get_docs():
     """Return the API documentation as plain text."""
     return Response(DOCUMENTATION, mimetype='text/plain'), 200