Update app.py

app.py

@@ -237,12 +237,12 @@ with gr.Blocks(title="HF Space Auto‑Builder", css=custom_css) as demo:
         # 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")
+            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", show_progress=True)
+            send_btn = gr.Button("Send", variant="primary")
 
             with gr.Tabs():
                 with gr.TabItem("Chat"):
@@ -265,16 +265,18 @@ with gr.Blocks(title="HF Space Auto‑Builder", css=custom_css) as demo:
     login_btn.click(show_profile, inputs=None, outputs=status_md)
     login_btn.click(list_private_models, inputs=None, outputs=models_md)
 
-    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]
-    )
+    # 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]
+        )
 
     def _clear_chat():
         return [], "", "", "<p>No Space yet.</p>"