Update app.py

app.py

@@ -4,6 +4,11 @@ import json
 from huggingface_hub import HfApi, create_repo, upload_file, login
 import random
 import string
+import logging
+
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 # Import documentation from assembler_docs.py
 from assembler_docs import DOCUMENTATION
@@ -18,6 +23,7 @@ if not HF_TOKEN:
 # Log in to Hugging Face Hub with the token
 try:
     login(token=HF_TOKEN)
+    logger.info("Successfully logged in to Hugging Face Hub")
 except Exception as e:
     raise ValueError(f"Failed to log in to Hugging Face Hub: {str(e)}")
 
@@ -28,6 +34,17 @@ def generate_space_name():
     random_suffix = ''.join(random.choices(string.ascii_lowercase + string.digits, k=6))
     return f"GeneratedSpace-{random_suffix}"
 
+def verify_organization_access(organization: str, token: str) -> bool:
+    """Verify if the token has write access to the organization."""
+    try:
+        # Try to list repositories or check organization membership
+        repos = hf_api.list_repos(author=organization, token=token)
+        logger.info(f"Verified access to organization {organization}")
+        return True
+    except Exception as e:
+        logger.error(f"No write access to organization {organization}: {str(e)}")
+        return False
+
 @app.route('/create-space', methods=['POST'])
 def create_hf_space():
     try:
@@ -49,21 +66,27 @@ def create_hf_space():
         if space_type not in valid_space_types:
             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
+        # Organization namespace for new Spaces
+        ORGANIZATION = "Space-Share"
         space_name = generate_space_name()
-        full_repo_id = f"Space-Share/{space_name}"
+        full_repo_id = f"{ORGANIZATION}/{space_name}"
+
+        # Verify token has access to Space-Share organization
+        if not verify_organization_access(ORGANIZATION, HF_TOKEN):
+            return jsonify({"error": f"No write access to organization {ORGANIZATION}. Check your HF_TOKEN permissions."}), 403
 
         try:
-            # Attempt to create the repository with explicit verification
+            # Attempt to create the repository with explicit organization context
             repo_info = create_repo(
                 repo_id=space_name,
+                organization=ORGANIZATION,  # Explicitly specify the organization
                 repo_type="space",
                 space_sdk=space_type,
                 token=HF_TOKEN,
                 private=False,
                 exist_ok=True  # Allow creation even if the repo might exist
             )
-            print(f"Repository created: {full_repo_id}")
+            logger.info(f"Repository created: {full_repo_id}")
         except Exception as e:
             return jsonify({"error": f"Failed to create repository {full_repo_id}: {str(e)}"}), 500
 
@@ -72,6 +95,7 @@ def create_hf_space():
             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
+            logger.info(f"Verified repository exists: {full_repo_id}")
         except Exception as e:
             return jsonify({"error": f"Failed to verify repository {full_repo_id}: {str(e)}"}), 500
 
@@ -93,7 +117,7 @@ def create_hf_space():
                     repo_type="space",
                     token=HF_TOKEN
                 )
-                print(f"Uploaded file: {filename}")
+                logger.info(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
@@ -118,7 +142,7 @@ def create_hf_space():
                     repo_type="space",
                     token=HF_TOKEN
                 )
-                print("Uploaded requirements.txt")
+                logger.info("Uploaded requirements.txt")
             except Exception as e:
                 os.remove("temp_requirements.txt")
                 return jsonify({"error": f"Failed to upload requirements.txt: {str(e)}"}), 500
@@ -144,13 +168,13 @@ CMD ["python", "app.py"]
                     repo_type="space",
                     token=HF_TOKEN
                 )
-                print("Uploaded Dockerfile")
+                logger.info("Uploaded Dockerfile")
             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}"
+        space_url = f"https://huggingface.co/spaces/{full_repo_id}"
         return jsonify({
             "message": "New Space created",
             "url": space_url,