Update app.py

app.py

@@ -1,274 +1,123 @@
 import gradio as gr
-import os
-import re
-import tempfile
-import shutil
-import git
-from huggingface_hub import (
-    create_repo,
-    upload_folder,
-    list_repo_files,
-    delete_file,
-    Repository,
-    whoami,
+from app_logic import (
+    create_space,
+    view_space_files,
+    update_space_file,
 )
-import logging
-from pathlib import Path
-
-# Configure logging
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
-
-# Function to parse markdown input
-def parse_markdown(markdown_input):
-    """Parse markdown input to extract space details and file structure."""
-    space_info = {"repo_name": "", "owner": "", "sdk": "gradio", "files": []}
-    current_file = None
-    file_content = []
-    in_file_content = False
-
-    lines = markdown_input.strip().split("\n")
-    for line in lines:
-        line = line.strip()
-        # Extract space name
-        if line.startswith("# Space:"):
-            space_info["repo_name"] = line.replace("# Space:", "").strip()
-            if "/" in space_info["repo_name"]:
-                space_info["owner"], space_info["repo_name"] = space_info["repo_name"].split("/", 1)
-        # Detect file structure section
-        elif line.startswith("## File Structure"):
-            continue
-        # Detect file in structure
-        elif line.startswith("📄") or line.startswith("📁"):
-            if current_file and file_content:
-                space_info["files"].append({"path": current_file, "content": "\n".join(file_content)})
-                file_content = []
-            current_file = line[2:].strip()
-            in_file_content = False
-        # Detect file content section
-        elif line.startswith("### File:"):
-            if current_file and file_content:
-                space_info["files"].append({"path": current_file, "content": "\n".join(file_content)})
-                file_content = []
-            current_file = line.replace("### File:", "").strip()
-            in_file_content = True
-        # Handle file content
-        elif in_file_content and line.startswith("```"):
-            if file_content:
-                space_info["files"].append({"path": current_file, "content": "\n".join(file_content)})
-                file_content = []
-                in_file_content = False
-            else:
-                in_file_content = True
-        elif in_file_content:
-            file_content.append(line)
-
-    # Append the last file
-    if current_file and file_content:
-        space_info["files"].append({"path": current_file, "content": "\n".join(file_content)})
-
-    return space_info
-
-# Function to create and populate a Space
-def create_space(api_token, space_name, owner, sdk, markdown_input):
-    """Create a Hugging Face Space and populate it with files from markdown input."""
-    try:
-        # Authenticate with Hugging Face Hub
-        if not api_token:
-            return "Error: Please provide a valid Hugging Face API token."
-
-        # Get user info
-        user_info = whoami(token=api_token)
-        if not owner:
-            owner = user_info["name"]
-        repo_id = f"{owner}/{space_name}"
-
-        # Create temporary directory
-        with tempfile.TemporaryDirectory() as temp_dir:
-            repo_path = os.path.join(temp_dir, space_name)
-            os.makedirs(repo_path, exist_ok=True)
-
-            # Parse markdown input
-            space_info = parse_markdown(markdown_input)
-            if space_info["repo_name"] != f"{owner}/{space_name}":
-                return "Error: Space name in markdown does not match provided owner/space_name."
-
-            # Write files to temporary directory
-            for file_info in space_info["files"]:
-                file_path = os.path.join(repo_path, file_info["path"])
-                os.makedirs(os.path.dirname(file_path), exist_ok=True)
-                with open(file_path, "w", encoding="utf-8") as f:
-                    f.write(file_info["content"])
-                logger.info(f"Wrote file: {file_path}")
-
-            # Create repository on Hugging Face
-            try:
-                create_repo(
-                    repo_id=repo_id,
-                    token=api_token,
-                    repo_type="space",
-                    space_sdk=sdk,
-                    private=False,
-                )
-                logger.info(f"Created Space: {repo_id}")
-            except Exception as e:
-                if "already exists" in str(e):
-                    logger.info(f"Space {repo_id} already exists, proceeding to update.")
-                else:
-                    return f"Error creating Space: {str(e)}"
-
-            # Initialize Git repository
-            repo = git.Repo.init(repo_path)
-            repo.git.add(all=True)
-            repo.index.commit("Initial commit from Gradio app")
-
-            # Push to Hugging Face Space
-            upload_folder(
-                repo_id=repo_id,
-                folder_path=repo_path,
-                path_in_repo=".",
-                token=api_token,
-                commit_message="Initial Space setup",
-            )
-
-            return f"Successfully created Space: https://huggingface.co/spaces/{repo_id}"
-
-    except Exception as e:
-        logger.error(f"Error: {str(e)}")
-        return f"Error: {str(e)}"
-
-# Function to view Space files
-def view_space_files(api_token, space_name, owner):
-    """List files in a Hugging Face Space."""
-    try:
-        if not api_token:
-            return "Error: Please provide a valid Hugging Face API token."
-        repo_id = f"{owner}/{space_name}" if owner else space_name
-        files = list_repo_files(repo_id=repo_id, token=api_token, repo_type="space")
-        return "\n".join(files) or "No files found in the Space."
-    except Exception as e:
-        return f"Error: {str(e)}"
-
-# Function to update a Space file
-def update_space_file(api_token, space_name, owner, file_path, file_content, commit_message):
-    """Update a file in a Hugging Face Space with a commit."""
-    try:
-        if not api_token:
-            return "Error: Please provide a valid Hugging Face API token."
-        repo_id = f"{owner}/{space_name}" if owner else space_name
-
-        # Create temporary directory for cloning
-        with tempfile.TemporaryDirectory() as temp_dir:
-            repo_path = os.path.join(temp_dir, space_name)
-            repo = Repository(
-                local_dir=repo_path,
-                clone_from=f"https://huggingface.co/spaces/{repo_id}",
-                repo_type="space",
-                use_auth_token=api_token,
-            )
-
-            # Write updated file
-            full_path = os.path.join(repo_path, file_path)
-            os.makedirs(os.path.dirname(full_path), exist_ok=True)
-            with open(full_path, "w", encoding="utf-8") as f:
-                f.write(file_content)
-
-            # Commit and push changes
-            repo.git_add(file_path)
-            repo.git_commit(commit_message or f"Update {file_path}")
-            repo.git_push()
-
-            return f"Successfully updated {file_path} in Space: https://huggingface.co/spaces/{repo_id}"
-
-    except Exception as e:
-        return f"Error: {str(e)}"
 
 # Gradio interface
-def main():
-    with gr.Blocks(theme=gr.themes.Soft()) as demo:
-        gr.Markdown("# Hugging Face Space Builder")
-        gr.Markdown("Create, view, and update Hugging Face Spaces with a custom file structure.")
-
-        # Authentication
-        with gr.Group():
-            gr.Markdown("## Authentication")
-            api_token = gr.Textbox(
-                label="Hugging Face API Token",
-                type="password",
-                placeholder="Enter your Hugging Face API token",
-            )
-
-        # Create Space
-        with gr.Group():
-            gr.Markdown("## Create a New Space")
-            space_name = gr.Textbox(label="Space Name", placeholder="my-space")
-            owner = gr.Textbox(
-                label="Owner (optional)", placeholder="Leave blank for current user"
-            )
-            sdk = gr.Dropdown(
-                label="SDK", choices=["gradio", "streamlit", "docker", "static"], value="gradio"
-            )
-            markdown_input = gr.Textbox(
-                label="Markdown File Structure",
-                placeholder="Paste markdown with file structure and contents",
-                lines=10,
-            )
-            create_btn = gr.Button("Create Space")
-            create_output = gr.Textbox(label="Result")
-
-        # View Space Files
-        with gr.Group():
-            gr.Markdown("## View Space Files")
-            view_space_name = gr.Textbox(label="Space Name", placeholder="my-space")
-            view_owner = gr.Textbox(
-                label="Owner (optional)", placeholder="Leave blank for current user"
-            )
-            view_btn = gr.Button("List Files")
-            view_output = gr.Textbox(label="Files in Space")
+def main_ui():
+    with gr.Blocks(theme=gr.themes.Soft(primary_hue=gr.themes.colors.blue, secondary_hue=gr.themes.colors.sky), title="Hugging Face Space Builder") as demo:
+        gr.Markdown(
+            """
+            # 🛠️ Hugging Face Space Builder
+            Create, view, and update Hugging Face Spaces with a custom file structure defined via markdown input.
+            """
+        )
 
-        # Update Space File
-        with gr.Group():
-            gr.Markdown("## Update Space File")
-            update_space_name = gr.Textbox(label="Space Name", placeholder="my-space")
-            update_owner = gr.Textbox(
-                label="Owner (optional)", placeholder="Leave blank for current user"
-            )
-            file_path = gr.Textbox(label="File Path", placeholder="path/to/file.py")
-            file_content = gr.Textbox(
-                label="File Content", placeholder="Enter new file content", lines=10
-            )
-            commit_message = gr.Textbox(
-                label="Commit Message", placeholder="Update file content"
-            )
-            update_btn = gr.Button("Update File")
-            update_output = gr.Textbox(label="Result")
+        with gr.Row():
+            with gr.Column(scale=1):
+                api_token_input = gr.Textbox(
+                    label="Hugging Face API Token",
+                    type="password",
+                    placeholder="Enter your HF token (hf_xxx)",
+                    info="Get your token from hf.co/settings/tokens. Needs 'write' access for creating/updating spaces."
+                )
+            with gr.Column(scale=2):
+                 gr.Markdown(" ") # Spacer or additional info can go here
+
+        with gr.Tabs():
+            with gr.TabItem("🚀 Create New Space"):
+                with gr.Row():
+                    space_name_create_input = gr.Textbox(label="Space Name", placeholder="my-awesome-app (no slashes)", scale=2)
+                    owner_create_input = gr.Textbox(label="Owner Username/Org", placeholder="Leave blank for your HF username", scale=1)
+                
+                sdk_create_input = gr.Dropdown(
+                    label="Space SDK", 
+                    choices=["gradio", "streamlit", "docker", "static"], 
+                    value="gradio",
+                    info="Select the type of Space (Gradio, Streamlit, etc.)."
+                )
+                markdown_input_create = gr.Textbox(
+                    label="Markdown File Structure & Content",
+                    placeholder="""Define files using '### File: path/to/file.ext'
+followed by content, optionally in code blocks.
+Example:
+### File: app.py
+```python
+print("Hello World!")
+```
+
+### File: README.md
+```markdown
+# My Awesome App
+This is a test.
+```
+""", # Triple backticks within the example are commented out for display
+                    lines=15,
+                    interactive=True,
+                    info="Define files using '### File: path/to/your/file.ext' followed by content, optionally in ```code blocks```."
+                )
+                create_btn = gr.Button("Create Space", variant="primary")
+                create_output_md = gr.Markdown(label="Result")
+
+            with gr.TabItem("📄 View Space Files"):
+                with gr.Row():
+                    space_name_view_input = gr.Textbox(label="Space Name", placeholder="my-existing-app (no slashes)", scale=2)
+                    owner_view_input = gr.Textbox(label="Owner Username/Org", placeholder="Leave blank if it's your space", scale=1)
+                view_btn = gr.Button("List Files", variant="primary")
+                view_output_md = gr.Markdown(label="Files in Space")
+
+            with gr.TabItem("✏️ Update Space File"):
+                with gr.Row():
+                    space_name_update_input = gr.Textbox(label="Space Name", placeholder="my-target-app (no slashes)", scale=2)
+                    owner_update_input = gr.Textbox(label="Owner Username/Org", placeholder="Leave blank if it's your space", scale=1)
+                
+                file_path_update_input = gr.Textbox(
+                    label="File Path in Repository", 
+                    placeholder="e.g., app.py or src/utils.py",
+                    info="The full path to the file within the space you want to update/create."
+                    )
+                file_content_update_input = gr.Textbox(
+                    label="New File Content", 
+                    placeholder="Enter the complete new content for the file.", 
+                    lines=10, 
+                    interactive=True
+                )
+                commit_message_update_input = gr.Textbox(
+                    label="Commit Message", 
+                    placeholder="e.g., Update app.py with new feature",
+                    info="Describe the changes you're making."
+                )
+                update_btn = gr.Button("Update File", variant="primary")
+                update_output_md = gr.Markdown(label="Result")
 
         # Event handlers
         create_btn.click(
             fn=create_space,
-            inputs=[api_token, space_name, owner, sdk, markdown_input],
-            outputs=create_output,
+            inputs=[api_token_input, space_name_create_input, owner_create_input, sdk_create_input, markdown_input_create],
+            outputs=create_output_md,
         )
         view_btn.click(
             fn=view_space_files,
-            inputs=[api_token, view_space_name, view_owner],
-            outputs=view_output,
+            inputs=[api_token_input, space_name_view_input, owner_view_input],
+            outputs=view_output_md,
         )
         update_btn.click(
             fn=update_space_file,
             inputs=[
-                api_token,
-                update_space_name,
-                update_owner,
-                file_path,
-                file_content,
-                commit_message,
+                api_token_input,
+                space_name_update_input,
+                owner_update_input,
+                file_path_update_input,
+                file_content_update_input,
+                commit_message_update_input,
             ],
-            outputs=update_output,
+            outputs=update_output_md,
         )
 
     return demo
 
 if __name__ == "__main__":
-    demo = main()
+    demo = main_ui()
     demo.launch()