Update app.py

app.py

@@ -78,8 +78,7 @@ def fetch_logs(repo_id: str, level: str) -> str:
 
 def check_iframe(url: str, timeout: int = 5) -> bool:
     try:
-        r = requests.get(url, timeout=timeout)
-        return r.status_code == 200
+        return requests.get(url, timeout=timeout).status_code == 200
     except:
         return False
 
@@ -102,9 +101,8 @@ def handle_user_message(
     system_msg = {
         "role":"system",
         "content":(
-            f"You are an AI assistant that scaffolds a complete HuggingFace Space using the "
-            f"{sdk_choice} SDK. Generate working code, handle errors, and ensure the deployed "
-            "iframe loads successfully."
+            f"You are an AI assistant scaffolding a complete HuggingFace Space using "
+            f"{sdk_choice}. Generate working code, handle errors, and verify the iframe loads."
         )
     }
     chat = [system_msg] + history
@@ -127,7 +125,6 @@ def handle_user_message(
             max_output_tokens=max_output_tokens,
         )
 
-        # generate code
         resp = client.models.generate_content(
             model="gemini-2.5-flash-preview-04-17",
             contents=[m["content"] for m in chat],
@@ -136,19 +133,15 @@ def handle_user_message(
         code = extract_code(resp.text)
         chat.append({"role":"assistant","content":code})
 
-        # local syntax check
+        # Syntax check
         try:
             compile(code, code_fn, "exec")
         except SyntaxError as e:
-            chat.append({
-                "role":"user",
-                "content": f"SyntaxError caught locally: {e}. Please fix the code."
-            })
-            time.sleep(backoff)
-            backoff = min(backoff * 2, 30)
+            chat.append({"role":"user","content":f"SyntaxError: {e}. Please fix."})
+            time.sleep(backoff); backoff = min(backoff*2, 30)
             continue
 
-        # write files
+        # Write files
         sdk_version = get_sdk_version(sdk_choice)
         files = {
             code_fn: code,
@@ -169,89 +162,75 @@ See https://huggingface.co/docs/hub/spaces-config-reference
             with open(fn, "w") as f:
                 f.write(content)
 
-        # push to HuggingFace
+        # Push to HF
         create_repo(repo_id=repo_id, token=oauth_token.token,
                     exist_ok=True, repo_type="space", space_sdk=sdk_choice)
         for fn in files:
-            upload_file(fn, fn, repo_id=repo_id,
-                        token=oauth_token.token, repo_type="space")
+            upload_file(
+                path_or_fileobj=fn,
+                path_in_repo=fn,
+                repo_id=repo_id,
+                token=oauth_token.token,
+                repo_type="space"
+            )
 
-        # fetch logs
         build_logs = fetch_logs(repo_id, "build")
         run_logs   = fetch_logs(repo_id, "run")
         errors = classify_errors(build_logs + "\n" + run_logs)
 
-        # success?
         if "ERROR" not in build_logs.upper() and \
            "ERROR" not in run_logs.upper() and \
            check_iframe(iframe_url):
             break
 
-        # ask for fixes
         chat.append({
             "role":"user",
             "content":(
-                f"Attempt {attempt} encountered {errors} errors.\n"
+                f"Attempt {attempt} had {errors} errors.\n"
                 f"Build logs:\n{build_logs}\n\n"
                 f"Run logs:\n{run_logs}\n\n"
-                f"Please fix the code and ensure the iframe at {iframe_url} returns HTTP 200."
+                f"Please fix and ensure iframe at {iframe_url} returns HTTP 200."
             )
         })
-        time.sleep(backoff)
-        backoff = min(backoff * 2, 30)
+        time.sleep(backoff); backoff = min(backoff*2, 30)
 
     messages = [{"role":m["role"],"content":m["content"]} for m in chat if m["role"]!="system"]
     iframe_html = (
         f'<iframe src="{iframe_url}" width="100%" height="500px"></iframe>'
-        + ("" if check_iframe(iframe_url) else
+        + ("" if check_iframe(iframe_url) else 
            "<p style='color:red;'>⚠️ iframe not responding.</p>")
     )
     return messages, build_logs, run_logs, iframe_html
 
-# — BUILD THE UI (Improved) —
-
-custom_css = """
-.gradio-container { max-width: 1200px; margin: auto; }
-.log-box { font-family: monospace; white-space: pre; }
-"""
-
-with gr.Blocks(title="HF Space Auto‑Builder", css=custom_css) as demo:
-    with gr.Row():
-        with gr.Column(scale=1, min_width=250):
-            login_btn = gr.LoginButton(variant="huggingface", size="sm")
-            status_md = gr.Markdown("*Not logged in.*")
-            models_md = gr.Markdown()
-            sdk_choice = gr.Radio(["gradio","streamlit"], value="gradio", label="SDK template")
-            api_key    = gr.Textbox(label="Gemini API Key", type="password")
-            grounding  = gr.Checkbox(label="Enable grounding", value=False)
-            temp       = gr.Slider(0,1,value=0.2,label="LLM temperature")
-            max_tokens = gr.Slider(256,4096,value=1024,step=256,label="Max tokens")
-            clear_btn  = gr.Button("Clear Chat", variant="secondary")
-
-        with gr.Column(scale=3):
-            gr.Markdown("## 🐢 Wuhp Auto‑Builder\nSign in, enter your prompt, and watch your Space live.")
-            chatbot = gr.Chatbot(elem_id="chatbot", type="messages")
-            user_in = gr.Textbox(placeholder="Type your prompt and hit Enter…", label="Prompt", lines=1)
-            send_btn = gr.Button("Send", variant="primary")
-
-            with gr.Tabs():
-                with gr.TabItem("Chat"):
-                    pass
-                with gr.TabItem("Logs"):
-                    with gr.Accordion("🔨 Build Logs", open=True):
-                        build_box = gr.Textbox(lines=8, interactive=False, elem_classes="log-box")
-                    with gr.Accordion("🚀 Run Logs", open=False):
-                        run_box = gr.Textbox(lines=8, interactive=False, elem_classes="log-box")
-                with gr.TabItem("Preview"):
-                    preview = gr.HTML("<p>No Space yet.</p>")
-
-    # load auth status
-    demo.load(show_profile, inputs=None, outputs=status_md)
+# — SIMPLE UI —
+
+with gr.Blocks(title="HF Space Auto‑Builder") as demo:
+    gr.Markdown("## 🐢 HF Space Auto‑Builder\n1) Sign in  2) Enter prompt  3) Watch code, logs & preview")
+
+    # Auth
+    login_btn = gr.LoginButton(variant="huggingface", size="lg")
+    status_md = gr.Markdown("*Not logged in.*")
+    models_md = gr.Markdown()
+    demo.load(show_profile,     inputs=None, outputs=status_md)
     demo.load(list_private_models, inputs=None, outputs=models_md)
-    login_btn.click(show_profile, inputs=None, outputs=status_md)
+    login_btn.click(show_profile,     inputs=None, outputs=status_md)
     login_btn.click(list_private_models, inputs=None, outputs=models_md)
 
-    # send / submit callbacks (Gradio auto-injects profile & oauth_token)
+    # Controls
+    sdk_choice = gr.Radio(["gradio","streamlit"], value="gradio", label="SDK")
+    api_key    = gr.Textbox(label="Gemini API Key", type="password")
+    grounding  = gr.Checkbox(label="Enable grounding", value=False)
+    temp       = gr.Slider(0,1,value=0.2, label="Temperature")
+    max_tokens = gr.Slider(256,4096,value=1024,step=256, label="Max tokens")
+
+    # Chat + outputs
+    chatbot   = gr.Chatbot(type="messages")
+    user_in   = gr.Textbox(placeholder="Your prompt…", label="Prompt", lines=1)
+    send_btn  = gr.Button("Send")
+    build_box = gr.Textbox(label="Build logs", lines=5, interactive=False)
+    run_box   = gr.Textbox(label="Run logs",   lines=5, interactive=False)
+    preview   = gr.HTML("<p>No Space yet.</p>")
+
     send_btn.click(
         fn=handle_user_message,
         inputs=[chatbot, sdk_choice, api_key, grounding, temp, max_tokens],
@@ -263,22 +242,11 @@ with gr.Blocks(title="HF Space Auto‑Builder", css=custom_css) as demo:
         outputs=[chatbot, build_box, run_box, preview]
     )
 
-    # clear chat
-    def _clear_chat():
-        return [], "", "", "<p>No Space yet.</p>"
-    clear_btn.click(
-        fn=_clear_chat,
-        inputs=None,
-        outputs=[chatbot, build_box, run_box, preview]
-    )
-
-    # refresh logs (auto-injected profile & oauth_token)
-    refresh_btn = gr.Button("Refresh Logs", variant="secondary", size="sm")
-    refresh_btn.click(
-        fn=lambda profile,token: (fetch_logs(f"{profile.username}/{profile.username}-auto-space","build"),
-                                  fetch_logs(f"{profile.username}/{profile.username}-auto-space","run")),
-        inputs=None,
-        outputs=[build_box, run_box]
-    )
+    # Refresh logs (auto-injects profile & oauth_token)
+    refresh_btn = gr.Button("Refresh Logs")
+    def _refresh(profile, oauth_token):
+        repo = f"{profile.username}/{profile.username}-auto-space"
+        return fetch_logs(repo, "build"), fetch_logs(repo, "run")
+    refresh_btn.click(fn=_refresh, outputs=[build_box, run_box])
 
     demo.launch(server_name="0.0.0.0", server_port=7860)