Update app.py

app.py

@@ -67,7 +67,7 @@ def handle_message(user_msg: str, state: Dict[str, Any]) -> Tuple[str, Dict[str,
         logs.append("⚠️ Failed to parse assistant JSON.\n" + resp.text)
         return "⚠️ Parsing error. Check logs.", state
 
-    # on first round, create the Space
+    # On first round, create the Space repo
     if not state["created"]:
         full_repo = f"{state['hf_username']}/{state['repo_name']}"
         create_repo(
@@ -83,7 +83,7 @@ def handle_message(user_msg: str, state: Dict[str, Any]) -> Tuple[str, Dict[str,
             "embed_url": f"https://huggingface.co/spaces/{full_repo}",
         })
 
-    # write any returned files
+    # Write any returned files
     if files:
         logs.append(f"Writing {len(files)} file(s): {list(files)}")
         for relpath, content in files.items():
@@ -92,7 +92,7 @@ def handle_message(user_msg: str, state: Dict[str, Any]) -> Tuple[str, Dict[str,
             with open(dest, "w", encoding="utf‑8") as fp:
                 fp.write(content)
 
-    # push a snapshot
+    # Push a snapshot
     HfApi(token=state["hf_token"]).upload_folder(
         folder_path=state["local_path"],
         repo_id=state["repo_id"],
@@ -103,34 +103,35 @@ def handle_message(user_msg: str, state: Dict[str, Any]) -> Tuple[str, Dict[str,
 
 with gr.Blocks(title="Gemini → HF Space scaffolder") as demo:
     with gr.Row():
-        login_btn   = gr.LoginButton(
+        login_btn = gr.LoginButton(
             logout_value="Logout ({username})",
             variant="huggingface",
             size="lg"
         )
         login_btn.activate()
-        status_md   = gr.Markdown("Not logged in")
+        status_md = gr.Markdown("Not logged in")
 
-    def show_profile(profile: gr.OAuthProfile | None):
+    # Accept the event-plus-profile so that Gradio.load injects correctly
+    def show_profile(evt, profile: gr.OAuthProfile | None):
         return "*Not logged in.*" if profile is None else f"Logged in as **{profile.username}**"
 
     demo.load(fn=show_profile, inputs=[login_btn], outputs=status_md)
 
     with gr.Row():
         with gr.Column(scale=1):
-            gemini_key     = gr.Textbox(label="Gemini API Key", type="password")
-            hf_user        = gr.Textbox(label="HF Username")
-            repo_name      = gr.Textbox(label="New App (repo) name")
-            session_id     = gr.Textbox(value="", visible=False)
-            start_btn      = gr.Button("Start a new app")
+            gemini_key  = gr.Textbox(label="Gemini API Key", type="password")
+            hf_user     = gr.Textbox(label="HF Username")
+            repo_name   = gr.Textbox(label="New App (repo) name")
+            session_id  = gr.Textbox(value="", visible=False)
+            start_btn   = gr.Button("Start a new app")
         with gr.Column(scale=3):
-            chatbot        = gr.Chatbot(type="messages", value=[])
-            logs_display   = gr.Textbox(label="Operation Logs", interactive=False, lines=8)
-            preview_iframe = gr.HTML("<p>No deployed app yet.</p>")
-            user_msg       = gr.Textbox(label="Your message")
-            send_btn       = gr.Button("Send", interactive=False)
+            chatbot       = gr.Chatbot(type="messages", value=[])
+            logs_display  = gr.Textbox(label="Operation Logs", interactive=False, lines=8)
+            preview_iframe= gr.HTML("<p>No deployed app yet.</p>")
+            user_msg      = gr.Textbox(label="Your message")
+            send_btn      = gr.Button("Send", interactive=False)
 
-    # on_start needs the extra evt param for .click()
+    # on_start needs the event param
     def on_start(
         evt,
         gemini_key: str,
@@ -151,25 +152,8 @@ with gr.Blocks(title="Gemini → HF Space scaffolder") as demo:
         outputs=[session_id, logs_display, preview_iframe, send_btn],
     )
 
-    # on_send for the button (includes evt)
-    def on_send_click(
-        evt,
-        msg: str,
-        chat_history: list[dict],
-        sess_id: str
-    ):
-        return _handle_send(msg, chat_history, sess_id)
-
-    # on_send for the textbox (no evt)
-    def on_send_submit(
-        msg: str,
-        chat_history: list[dict],
-        sess_id: str
-    ):
-        return _handle_send(msg, chat_history, sess_id)
-
-    # shared send logic
-    def _handle_send(
+    # Single handler for both button click and textbox submit (3 args)
+    def on_send(
         msg: str,
         chat_history: list[dict],
         sess_id: str
@@ -180,8 +164,8 @@ with gr.Blocks(title="Gemini → HF Space scaffolder") as demo:
         state_store[sess_id] = new_state
 
         history = chat_history or []
-        history.append({"role": "user",      "content": msg})
-        history.append({"role": "assistant", "content": reply})
+        history.append({"role": "user",    "content": msg})
+        history.append({"role": "assistant","content": reply})
 
         logs = "\n".join(new_state["logs"])
         embed = ""
@@ -190,12 +174,12 @@ with gr.Blocks(title="Gemini → HF Space scaffolder") as demo:
         return history, sess_id, logs, embed
 
     send_btn.click(
-        on_send_click,
+        on_send,
         inputs=[user_msg, chatbot, session_id],
         outputs=[chatbot, session_id, logs_display, preview_iframe],
     )
     user_msg.submit(
-        on_send_submit,
+        on_send,
         inputs=[user_msg, chatbot, session_id],
         outputs=[chatbot, session_id, logs_display, preview_iframe],
     )