Update app.py

app.py

@@ -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 JSON, WRITE & DEPLOY —
 
 def handle_user_message(
     history,               # list of {"role","content"} dicts
@@ -63,20 +63,18 @@ def handle_user_message(
     profile: gr.OAuthProfile | None,
     oauth_token: gr.OAuthToken | None
 ):
-    # 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_msg = {
         "role":"system",
         "content":(
             "Return exactly one JSON object matching this schema:\n"
             "  • repo_name (string)\n"
             "  • files (object mapping filename→file-content)\n\n"
-            "Files must include at least:\n"
+            "Files must include:\n"
             "  - A code file (default name: app.py unless you choose otherwise)\n"
             "  - requirements.txt with dependencies\n"
             "  - README.md with frontmatter (title, emoji, sdk, sdk_version, app_file)\n\n"
@@ -84,14 +82,12 @@ def handle_user_message(
         )
     }
 
-    # 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
         if sdk_choice == "gradio":
             import gradio as _gr; sdk_version = _gr.__version__
         else:
@@ -102,20 +98,19 @@ def handle_user_message(
             tools=tools,
             response_modalities=["TEXT"],
             response_mime_type="application/json",
-            response_schema=RepoSpec
+            response_schema=RepoSpec    # <<< use the Pydantic model here
         )
 
-        # Call the LLM
         resp = client.models.generate_content(
             model="gemini-2.5-flash-preview-04-17",
             contents=[m["content"] for m in chat],
             config=cfg
         )
+        # validate & parse
         spec = RepoSpec.model_validate_json(resp.text)
         repo_name = spec.repo_name
         files     = spec.files
 
-        # Prepare repo
         repo_id = f"{profile.username}/{repo_name}"
         create_repo(
             repo_id=repo_id,
@@ -125,9 +120,8 @@ def handle_user_message(
             space_sdk=sdk_choice
         )
 
-        # Write & upload all files
+        # write & upload files
         for fn, content in files.items():
-            # Replace placeholder in README.md
             if fn.lower() == "readme.md":
                 content = content.replace("<SDK_VERSION>", sdk_version)
             with open(fn, "w") as f:
@@ -140,15 +134,11 @@ def handle_user_message(
                 repo_type="space"
             )
 
-        # 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
         chat.append({
             "role":"user",
             "content":(
@@ -159,7 +149,6 @@ 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
@@ -169,9 +158,8 @@ 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"
-                "_The LLM controls filenames, code, README, requirements, and loops until successful._\n\n---")
+                "_LLM controls filenames, code, README, requirements, and loops until successful._\n\n---")
 
-    # Login
     login_btn = gr.LoginButton(variant="huggingface", size="lg")
     status_md = gr.Markdown("*Not logged in.*")
     models_md = gr.Markdown()
@@ -180,12 +168,10 @@ 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")