Update app.py

app.py

@@ -6,7 +6,7 @@ from huggingface_hub.utils import build_hf_headers, get_session, hf_raise_for_st
 from google import genai
 from google.genai.types import Tool, GenerateContentConfig, GoogleSearch
 
-# — JSON spec model —
+# — JSON SPEC MODEL —
 
 class RepoSpec(BaseModel):
     repo_name: str
@@ -52,7 +52,7 @@ def fetch_logs(repo_id: str, level: str):
                     continue
     return "\n".join(lines)
 
-# — CORE LOOP: ask LLM for structured JSON, write & deploy —
+# — CORE LOOP: ASK LLM FOR STRUCTURED JSON, WRITE & DEPLOY —
 
 def handle_user_message(
     history,               # list of {"role","content"} dicts
@@ -63,13 +63,13 @@ def handle_user_message(
     profile: gr.OAuthProfile | None,
     oauth_token: gr.OAuthToken | None
 ):
-    # require login
+    # Require login
     if not (profile and oauth_token):
         return history + [{"role":"assistant","content":"⚠️ Please log in first."}], "", "", "<p>No Space yet.</p>"
 
     client = genai.Client(api_key=gemini_api_key)
 
-    # system instructions
+    # System instructions
     system_msg = {
         "role":"system",
         "content":(
@@ -84,14 +84,14 @@ def handle_user_message(
         )
     }
 
-    # build the chat context
+    # Build chat context
     chat = [system_msg] + history + [{"role":"user", "content":user_prompt}]
 
     repo_id = None
     build_logs = run_logs = ""
 
     for _ in range(5):
-        # detect sdk_version at runtime
+        # Detect SDK version at runtime
         if sdk_choice == "gradio":
             import gradio as _gr; sdk_version = _gr.__version__
         else:
@@ -105,7 +105,7 @@ def handle_user_message(
             response_schema=RepoSpec
         )
 
-        # call the LLM
+        # Call the LLM
         resp = client.models.generate_content(
             model="gemini-2.5-flash-preview-04-17",
             contents=[m["content"] for m in chat],
@@ -115,7 +115,7 @@ def handle_user_message(
         repo_name = spec.repo_name
         files     = spec.files
 
-        # prepare repo
+        # Prepare repo
         repo_id = f"{profile.username}/{repo_name}"
         create_repo(
             repo_id=repo_id,
@@ -125,9 +125,9 @@ def handle_user_message(
             space_sdk=sdk_choice
         )
 
-        # write & upload all files
+        # Write & upload all files
         for fn, content in files.items():
-            # replace placeholder in README
+            # Replace placeholder in README.md
             if fn.lower() == "readme.md":
                 content = content.replace("<SDK_VERSION>", sdk_version)
             with open(fn, "w") as f:
@@ -140,13 +140,15 @@ def handle_user_message(
                 repo_type="space"
             )
 
-        # fetch logs
+        # Fetch logs
         build_logs = fetch_logs(repo_id, "build")
         run_logs   = fetch_logs(repo_id, "run")
+
+        # Stop if clean
         if "ERROR" not in build_logs.upper() and "ERROR" not in run_logs.upper():
             break
 
-        # feed errors back
+        # Feed errors back
         chat.append({
             "role":"user",
             "content":(
@@ -157,6 +159,7 @@ def handle_user_message(
         })
         time.sleep(2)
 
+    # Prepare UI outputs
     messages = [{"role":m["role"], "content":m["content"]} for m in chat if m["role"]!="system"]
     iframe   = f'<iframe src="https://huggingface.co/spaces/{repo_id}" width="100%" height="500px"></iframe>'
     return messages, build_logs, run_logs, iframe
@@ -165,10 +168,11 @@ def handle_user_message(
 
 with gr.Blocks(title="HF Space Auto‑Builder") as demo:
     gr.Markdown("## Sign in + Auto‑Build Spaces\n\n"
-                "1. Sign in  2. Prompt  3. Deploy & Debug\n\n"
-                "_LLM controls filenames, code, README, requirements, and iterates until successful._\n\n---")
+                "1. Sign in   2. Prompt   3. Deploy & Debug\n\n"
+                "_The LLM controls filenames, code, README, requirements, and loops until successful._\n\n---")
 
-    login_btn = gr.LoginButton("huggingface", size="lg")
+    # Login
+    login_btn = gr.LoginButton(variant="huggingface", size="lg")
     status_md = gr.Markdown("*Not logged in.*")
     models_md = gr.Markdown()
     demo.load(show_profile,        None, status_md)
@@ -176,10 +180,12 @@ with gr.Blocks(title="HF Space Auto‑Builder") as demo:
     login_btn.click(show_profile,        None, status_md)
     login_btn.click(list_private_models, None, models_md)
 
+    # Controls
     sdk_choice = gr.Radio(["gradio","streamlit"], "gradio", label="SDK")
     api_key    = gr.Textbox(label="Gemini API Key", type="password")
     grounding  = gr.Checkbox(label="Enable grounding")
 
+    # Chat + Outputs
     chatbot   = gr.Chatbot(type="messages")
     user_in   = gr.Textbox(label="Prompt", placeholder="e.g. Build a CSV inspector…")
     send_btn  = gr.Button("Send")