Update app.py

app.py

@@ -148,7 +148,7 @@ def handle_user_message(
             backoff = min(backoff * 2, 30)
             continue
 
-        # write code, README, requirements
+        # write files
         sdk_version = get_sdk_version(sdk_choice)
         files = {
             code_fn: code,
@@ -181,13 +181,13 @@ See https://huggingface.co/docs/hub/spaces-config-reference
         run_logs   = fetch_logs(repo_id, "run")
         errors = classify_errors(build_logs + "\n" + run_logs)
 
-        # success criteria
+        # success?
         if "ERROR" not in build_logs.upper() and \
            "ERROR" not in run_logs.upper() and \
            check_iframe(iframe_url):
             break
 
-        # ask for a fix
+        # ask for fixes
         chat.append({
             "role":"user",
             "content":(
@@ -200,13 +200,12 @@ See https://huggingface.co/docs/hub/spaces-config-reference
         time.sleep(backoff)
         backoff = min(backoff * 2, 30)
 
-    messages = [{"role":m["role"], "content":m["content"]} for m in chat if m["role"]!="system"]
+    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
            "<p style='color:red;'>⚠️ iframe not responding.</p>")
     )
-
     return messages, build_logs, run_logs, iframe_html
 
 # — BUILD THE UI (Improved) —
@@ -218,30 +217,21 @@ custom_css = """
 
 with gr.Blocks(title="HF Space Auto‑Builder", css=custom_css) as demo:
     with gr.Row():
-        # Sidebar
         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")
+            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")
 
-        # Main area
         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
-            )
+            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():
@@ -249,35 +239,31 @@ with gr.Blocks(title="HF Space Auto‑Builder", css=custom_css) as demo:
                     pass
                 with gr.TabItem("Logs"):
                     with gr.Accordion("🔨 Build Logs", open=True):
-                        build_box = gr.Textbox(
-                            lines=8, interactive=False, elem_classes="log-box"
-                        )
+                        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"
-                        )
+                        run_box = gr.Textbox(lines=8, interactive=False, elem_classes="log-box")
                 with gr.TabItem("Preview"):
                     preview = gr.HTML("<p>No Space yet.</p>")
 
-    # Wiring callbacks
+    # load auth status
     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(list_private_models, inputs=None, outputs=models_md)
 
-    # Wrap both click and submit in a spinner
-    with gr.Spinner():
-        send_btn.click(
-            fn=handle_user_message,
-            inputs=[chatbot, sdk_choice, api_key, grounding, temp, max_tokens, status_md, models_md],
-            outputs=[chatbot, build_box, run_box, preview]
-        )
-        user_in.submit(
-            fn=handle_user_message,
-            inputs=[chatbot, sdk_choice, api_key, grounding, temp, max_tokens, status_md, models_md],
-            outputs=[chatbot, build_box, run_box, preview]
-        )
+    # send / submit callbacks (Gradio auto-injects profile & oauth_token)
+    send_btn.click(
+        fn=handle_user_message,
+        inputs=[chatbot, sdk_choice, api_key, grounding, temp, max_tokens],
+        outputs=[chatbot, build_box, run_box, preview]
+    )
+    user_in.submit(
+        fn=handle_user_message,
+        inputs=[chatbot, sdk_choice, api_key, grounding, temp, max_tokens],
+        outputs=[chatbot, build_box, run_box, preview]
+    )
 
+    # clear chat
     def _clear_chat():
         return [], "", "", "<p>No Space yet.</p>"
     clear_btn.click(
@@ -286,13 +272,13 @@ with gr.Blocks(title="HF Space Auto‑Builder", css=custom_css) as demo:
         outputs=[chatbot, build_box, run_box, preview]
     )
 
-    # Refresh Logs
+    # refresh logs (auto-injected profile & oauth_token)
     refresh_btn = gr.Button("Refresh Logs", variant="secondary", size="sm")
-    def _refresh(profile, oauth_token):
-        if not profile or not oauth_token:
-            return "", ""
-        repo = f"{profile.username}/{profile.username}-auto-space"
-        return fetch_logs(repo, "build"), fetch_logs(repo, "run")
-    refresh_btn.click(_refresh, inputs=None, outputs=[build_box, run_box])
+    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]
+    )
 
     demo.launch(server_name="0.0.0.0", server_port=7860)