Update app.py

app.py

@@ -7,7 +7,6 @@ import string
 import logging
 import black
 
-user_="broadfield-dev"
 # Configure logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
@@ -18,17 +17,16 @@ from assembler_docs import DOCUMENTATION
 app = Flask(__name__)
 
 # Hugging Face API token (set in Space settings)
-#HF_TOKEN = os.getenv("HF_TOKEN")
-HF_TOKEN = ""
-#if not HF_TOKEN:
-#    raise ValueError("HF_TOKEN not set. Add it in Space settings.")
+HF_TOKEN = os.getenv("HF_TOKEN")
+if not HF_TOKEN:
+    raise ValueError("HF_TOKEN not set. Add it in Space settings.")
 
 # Log in to Hugging Face Hub with the token
-#try:
-    #login(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)}")
+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)}")
 
 hf_api = HfApi()
 
@@ -47,7 +45,7 @@ def verify_organization_access(organization: str, token: str) -> bool:
     except Exception as e:
         logger.error(f"No write access to organization {organization}: {str(e)}")
         return False
-'''def build_space(repo_name,file_name,file_content,access_token=""):
+def build_space(repo_name,file_name,file_content,access_token=""):
     try:
         repo_path=user_+str(repo_)
         access_token=os.environ['HF_TOKEN']
@@ -77,7 +75,7 @@ def verify_organization_access(organization: str, token: str) -> bool:
         return [{'role':'assistant','content': commit_message+'\nCommit Success' }]
     except Exception as e:
         print("ERROR ",e)
-        return [{'role':'assistant','content': 'There was an Error: ' + str(e)}]'''
+        return [{'role':'assistant','content': 'There was an Error: ' + str(e)}]
 
 @app.route('/create-space', methods=['POST'])
 def create_hf_space():
@@ -101,7 +99,7 @@ def create_hf_space():
             return jsonify({"error": f"Invalid space_type. Must be one of {valid_space_types}"}), 400
 
         # Organization namespace for new Spaces
-        ORGANIZATION = user_
+        ORGANIZATION = "Space-Share"
         space_name = generate_space_name()
         full_repo_id = f"{ORGANIZATION}/{space_name}"
 
@@ -124,7 +122,7 @@ def create_hf_space():
 
         # Verify repository existence before uploading
         try:
-            repo_exists = api.repo_exists(repo_id=full_repo_id, repo_type="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}")