Update app.py

app.py

@@ -33,7 +33,6 @@ state_store: Dict[str, Dict[str, Any]] = {}
 
 def start_app(gemini_key: str, hf_token: str, hf_username: str, repo_name: str) -> Dict[str, Any]:
     """Initialise chat with Gemini + create a local workspace for the new app."""
-
     os.makedirs(WORKSPACE_DIR, exist_ok=True)
 
     # Gemini client & chat session
@@ -61,7 +60,6 @@ def start_app(gemini_key: str, hf_token: str, hf_username: str, repo_name: str)
 
 def handle_message(user_msg: str, state: Dict[str, Any]) -> Tuple[str, Dict[str, Any]]:
     """Send *user_msg* to Gemini, act on the JSON response, and return assistant reply."""
-
     chat = state["chat"]
     logs = state.setdefault("logs", [])
 
@@ -132,12 +130,15 @@ with gr.Blocks(title="Gemini → HF Space scaffolder") as demo:
     # OAuth UI row (always visible)
     # -------------------------------------------------------------------------
     with gr.Row():
-        login_btn  = gr.LoginButton()
-        logout_btn = gr.LogoutButton()
-        status_md  = gr.Markdown("Not logged in")
+        login_btn = gr.LoginButton(
+            logout_value="Logout ({username})",  # label once signed in
+            variant="huggingface",               # optional styling
+            size="lg"
+        )
+        status_md = gr.Markdown("Not logged in")
 
     # Callback to refresh the status text whenever the page loads OR OAuth changes
-    def show_profile(profile: gr.OAuthProfile | None):  # automatically injected
+    def show_profile(profile: gr.OAuthProfile | None):
         if profile is None:
             return gr.Markdown.update(value="*Not logged in.*")
         return gr.Markdown.update(value=f"Logged in as **{profile.username}**")
@@ -150,14 +151,14 @@ with gr.Blocks(title="Gemini → HF Space scaffolder") as demo:
     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")
+            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")
+            start_btn = gr.Button("Start a new app")
 
         with gr.Column(scale=3):
-            chatbot        = gr.Chatbot(type="messages")
-            logs_display   = gr.Textbox(label="Operation Logs", interactive=False, lines=8)
+            chatbot = gr.Chatbot(type="messages")
+            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")
@@ -193,7 +194,7 @@ with gr.Blocks(title="Gemini → HF Space scaffolder") as demo:
         state_store[sess_id] = new_state
 
         chat_history.append((msg, reply))
-        logs  = "\n".join(new_state["logs"])
+        logs = "\n".join(new_state["logs"])
         embed = (
             f'<iframe src="{new_state["embed_url"]}" width="100%" height="500px"></iframe>'
             if new_state.get("embed_url") else ""