Update app.py

app.py

@@ -4,7 +4,9 @@ import json
 from huggingface_hub import HfApi, create_repo, upload_file
 import random
 import string
-import assembler_docs
+
+# Import documentation from assembler_docs.py
+from assembler_docs import DOCUMENTATION
 
 app = Flask(__name__)
 
@@ -15,12 +17,11 @@ if not HF_TOKEN:
 
 hf_api = HfApi()
 
-# Documentation as a string
-DOCUMENTATION = assembler_docs.DOCUMENTATION
 def generate_space_name():
     """Generate a unique Space name."""
     random_suffix = ''.join(random.choices(string.ascii_lowercase + string.digits, k=6))
     return f"GeneratedSpace-{random_suffix}"
+
 @app.route('/create-space', methods=['POST'])
 def create_hf_space():
     try:
@@ -28,23 +29,24 @@ def create_hf_space():
         data = request.get_json()
         if not data:
             return jsonify({"error": "No JSON data provided"}), 400
+
         # Extract parameters
         space_type = data.get("space_type", "gradio")  # Default to gradio if not specified
         files = data.get("files", {})  # Dictionary of filename: content
         params = data.get("parameters", {})  # Optional parameters
-    
+
         if not files:
             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}"}), 400
-    
-        # Create a unique Space name and repo
+
+        # Create a unique Space name and repo under Space-Share namespace
         space_name = generate_space_name()
-        full_repo_id = f"{hf_api.whoami(HF_TOKEN)['name']}/{space_name}"
-    
+        full_repo_id = f"Space-Share/{space_name}"
+
         create_repo(
             repo_id=space_name,
             repo_type="space",
@@ -52,7 +54,7 @@ def create_hf_space():
             token=HF_TOKEN,
             private=False
         )
-    
+
         # Handle multi-file uploads
         for filename, content in files.items():
             # Write content to a temporary file
@@ -61,7 +63,7 @@ def create_hf_space():
                     # Inject parameters into Python files if present
                     content = f"PARAMS = {json.dumps(params)}\n\n{content}"
                 f.write(content)
-    
+
             # Upload to the new Space
             upload_file(
                 path_or_fileobj=f"temp_{filename}",
@@ -71,7 +73,7 @@ def create_hf_space():
                 token=HF_TOKEN
             )
             os.remove(f"temp_{filename}")
-    
+
         # Add requirements.txt if not provided (basic defaults)
         if "requirements.txt" not in files:
             default_requirements = {
@@ -90,17 +92,17 @@ def create_hf_space():
                 token=HF_TOKEN
             )
             os.remove("temp_requirements.txt")
-    
+
         # Special handling for Docker Spaces
         if space_type == "docker" and "Dockerfile" not in files:
             default_dockerfile = """
-                FROM python:3.10-slim
-                WORKDIR /app
-                COPY . .
-                RUN pip install -r requirements.txt
-                EXPOSE 7860
-                CMD ["python", "app.py"]
-                """
+FROM python:3.10-slim
+WORKDIR /app
+COPY . .
+RUN pip install -r requirements.txt
+EXPOSE 7860
+CMD ["python", "app.py"]
+"""
             with open("temp_Dockerfile", "w") as f:
                 f.write(default_dockerfile)
             upload_file(
@@ -111,7 +113,8 @@ def create_hf_space():
                 token=HF_TOKEN
             )
             os.remove("temp_Dockerfile")
-        space_url = f"https://huggingface.co/spaces/{full_repo_id}"
+
+        space_url = f"https://huggingface.co/spaces/Space-Share/{space_name}"
         return jsonify({
             "message": "New Space created",
             "url": space_url,
@@ -122,9 +125,11 @@ def create_hf_space():
         return jsonify({"error": "Invalid JSON format"}), 400
     except Exception as e:
         return jsonify({"error": str(e)}), 500
+
 @app.route('/docs', methods=['GET'])
 def get_docs():
     """Return the API documentation as plain text."""
     return Response(DOCUMENTATION, mimetype='text/plain'), 200
-if '__name__'== '__main__':
+
+if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)