Update app.py

app.py

@@ -1,4 +1,4 @@
-import os, json
+import json, os
 import gradio as gr
 from huggingface_hub import (
     create_repo, list_models, upload_file, list_repo_files, constants
@@ -21,7 +21,13 @@ def list_private_models(profile, oauth_token):
     ]
     return "No models found." if not models else "Models:\n\n" + "\n - ".join(models)
 
-# — HF SPACE HELPERS — #
+# — MANUAL HELPERS — #
+
+def enable_create(profile, oauth_token):
+    return gr.update(interactive=bool(profile and oauth_token))
+
+def enable_repo_actions(repo_id, profile, oauth_token):
+    return gr.update(interactive=bool(repo_id and profile and oauth_token))
 
 def create_space(repo_name, sdk, profile, oauth_token):
     if not (profile and oauth_token):
@@ -45,9 +51,9 @@ def fetch_logs(repo_id, profile, oauth_token, level):
     if not (profile and oauth_token and repo_id):
         return "⚠️ Log in & create a Space first."
     # get JWT
-    r = get_session().get(f"{constants.ENDPOINT}/api/spaces/{repo_id}/jwt",
-                          headers=build_hf_headers())
-    hf_raise_for_status(r); jwt = r.json()["token"]
+    r = get_session().get(f"{constants.ENDPOINT}/api/spaces/{repo_id}/jwt", headers=build_hf_headers())
+    hf_raise_for_status(r)
+    jwt = r.json()["token"]
     url = f"https://api.hf.space/v1/{repo_id}/logs/{level}"
     lines = []
     with get_session().get(url, headers=build_hf_headers(token=jwt), stream=True) as resp:
@@ -63,7 +69,7 @@ def list_files(repo_id, profile, oauth_token):
         return "⚠️ Log in & create a Space first."
     return "\n".join(list_repo_files(repo_id, token=oauth_token.token, repo_type="space"))
 
-# — GEMINI FUNCTION DECLS — #
+# — GEMINI FUNCTION DECLARATIONS — #
 
 func_decls = [
     {
@@ -106,7 +112,7 @@ func_decls = [
 # — CHAT HANDLER — #
 
 def process_message(profile, oauth_token, user_msg, gemini_key, repo_name, sdk, chat_history, session):
-    # init on first call
+    # initialize on first call
     if session.get("chat") is None:
         client = genai.Client(api_key=gemini_key)
         cfg = types.GenerateContentConfig(
@@ -142,7 +148,7 @@ def process_message(profile, oauth_token, user_msg, gemini_key, repo_name, sdk,
         else:
             result = {"log": f"⚠️ Unknown function {name}"}
 
-        # feed back into chat
+        # feed back
         chat.send_message(
             types.Content(
                 role="function",
@@ -162,8 +168,9 @@ def process_message(profile, oauth_token, user_msg, gemini_key, repo_name, sdk,
     chat_history[-1] = (user_msg, final)
     return chat_history, session.get("iframe",""), session.get("log",""), session.get("files",""), session
 
+# — SYNC MANUAL CHANGES — #
+
 def sync_manual(profile, oauth_token, session):
-    # pull in any manual file changes
     if not (profile and oauth_token and session.get("repo_id")):
         return session.get("iframe",""), "⚠️ Cannot sync.", session.get("files",""), session
     files = list_files(session["repo_id"], profile, oauth_token)
@@ -171,112 +178,83 @@ def sync_manual(profile, oauth_token, session):
     session["log"]   = "🔄 Manual changes synced."
     return session.get("iframe",""), session["log"], session["files"], session
 
-# — LAYOUT — #
+# — BUILD THE UI — #
 
 with gr.Blocks(css="""
-#sidebar {background:#f7f7f7;padding:1em;border-right:1px solid #ddd;}
-#main    {padding:1em;}
+#sidebar {background:#f7f7f7;padding:1rem;border-right:1px solid #ddd;}
+#main    {padding:1rem;}
 """) as demo:
-
     with gr.Row():
         # Sidebar
         with gr.Column(elem_id="sidebar", scale=1):
             gr.Markdown("### 🔑 Login & Config")
-            login_btn     = gr.LoginButton(variant="huggingface", size="sm")
-            profile_state = gr.State(None)
-            token_state   = gr.State(None)
-            # capture login outputs
-            login_btn.click(None, outputs=[profile_state, token_state])
-
-            status_md = gr.Markdown("*Not logged in.*")
-            profile_state.change(show_profile, inputs=[profile_state], outputs=[status_md])
-            login_btn.click(show_profile, inputs=[profile_state], outputs=[status_md])
-
-            models_md = gr.Markdown()
-            profile_state.change(list_private_models,
-                                 inputs=[profile_state, token_state],
-                                 outputs=[models_md])
-            login_btn.click(list_private_models,
-                            inputs=[profile_state, token_state],
-                            outputs=[models_md])
+            hf_login = gr.LoginButton(variant="huggingface", size="sm")
+            status   = gr.Markdown("*Not logged in.*")
+            models   = gr.Markdown()
+
+            hf_login.click(show_profile, inputs=[hf_login], outputs=[status])
+            hf_login.click(list_private_models, inputs=[hf_login], outputs=[models])
 
             gemini_key = gr.Textbox(label="Gemini API Key", type="password")
-            repo_name  = gr.Textbox(label="Space name", placeholder="e.g. my-space")
+            repo_name  = gr.Textbox(label="Space name", placeholder="my-space")
             sdk_choice = gr.Radio(["gradio","streamlit"], value="gradio", label="SDK")
 
             gr.Markdown("---")
             sync_btn = gr.Button("🔄 Confirm Manual Changes")
-            sync_out = gr.Markdown()
-            sync_btn.click(sync_manual,
-                           inputs=[profile_state, token_state, gr.State({})],
-                           outputs=[gr.HTML(), sync_out, gr.Textbox(), gr.State({})])
 
         # Main area
         with gr.Column(elem_id="main", scale=3):
             tabs = gr.Tabs()
+
             with tabs:
                 with gr.TabItem("💬 Chat"):
-                    chatbox = gr.Chatbot(type="messages")
+                    chatbox = gr.Chatbot(type="messages", label="Chat")
                     user_in = gr.Textbox(show_label=False, placeholder="Ask the LLM…")
                     send    = gr.Button("Send")
                 with gr.TabItem("🛠️ Manual"):
                     gr.Markdown("#### Create a Space")
-                    repo_m  = gr.Textbox(label="Name")
-                    sdk_m   = gr.Radio(["gradio","streamlit"], label="SDK")
-                    create  = gr.Button("Create", interactive=False)
+                    repo_m = gr.Textbox(label="Name")
+                    sdk_m  = gr.Radio(["gradio","streamlit"], value="gradio", label="SDK")
+                    create_btn = gr.Button("Create Space", interactive=False)
                     sess    = gr.Textbox(visible=False)
                     log_c   = gr.Textbox(label="Log", interactive=False, lines=2)
                     preview = gr.HTML("<p>No Space yet</p>")
 
-                    profile_state.change(enable_create,
-                                         inputs=[profile_state, token_state],
-                                         outputs=[create])
-                    login_btn.click(enable_create,
-                                    inputs=[profile_state, token_state],
-                                    outputs=[create])
-
-                    create.click(create_space,
-                                 inputs=[repo_m, sdk_m, profile_state, token_state],
-                                 outputs=[sess, log_c, preview])
+                    hf_login.change(enable_create, inputs=[hf_login], outputs=[create_btn])
+                    create_btn.click(create_space,
+                                     inputs=[repo_m, sdk_m, hf_login, hf_login],
+                                     outputs=[sess, log_c, preview])
 
                     gr.Markdown("#### Upload File")
-                    path    = gr.Textbox(label="Path in Repo", value="app.py")
-                    file_u  = gr.File()
-                    up_btn  = gr.Button("Upload", interactive=False)
-                    log_u   = gr.Textbox(label="Log", interactive=False, lines=2)
-
-                    profile_state.change(enable_repo_actions,
-                                         inputs=[sess, profile_state, token_state],
-                                         outputs=[up_btn])
-                    login_btn.click(enable_repo_actions,
-                                    inputs=[sess, profile_state, token_state],
-                                    outputs=[up_btn])
+                    path   = gr.Textbox(label="Path in Repo", value="app.py")
+                    file_u = gr.File()
+                    up_btn = gr.Button("Upload File", interactive=False)
+                    log_u  = gr.Textbox(label="Log", interactive=False, lines=2)
 
+                    hf_login.change(enable_repo_actions,
+                                    inputs=[sess, hf_login, hf_login],
+                                    outputs=[up_btn])
                     up_btn.click(upload_file_to_space,
-                                 inputs=[file_u, path, sess, profile_state, token_state],
+                                 inputs=[file_u, path, sess, hf_login, hf_login],
                                  outputs=[log_u])
 
                     gr.Markdown("#### Fetch Logs")
-                    b_btn = gr.Button("Build Logs", interactive=False)
-                    r_btn = gr.Button("Run Logs",   interactive=False)
-                    log_b = gr.Textbox(label="Build", interactive=False, lines=5)
-                    log_r = gr.Textbox(label="Run",   interactive=False, lines=5)
-
-                    profile_state.change(enable_repo_actions,
-                                         inputs=[sess, profile_state, token_state],
-                                         outputs=[b_btn, r_btn])
-                    login_btn.click(enable_repo_actions,
-                                    inputs=[sess, profile_state, token_state],
-                                    outputs=[b_btn, r_btn])
+                    b_btn  = gr.Button("Build Logs", interactive=False)
+                    r_btn  = gr.Button("Run Logs", interactive=False)
+                    log_b  = gr.Textbox(label="Build", interactive=False, lines=5)
+                    log_r  = gr.Textbox(label="Run", interactive=False, lines=5)
 
+                    hf_login.change(enable_repo_actions,
+                                    inputs=[sess, hf_login, hf_login],
+                                    outputs=[b_btn, r_btn])
                     b_btn.click(fetch_logs,
-                                inputs=[sess, profile_state, token_state, gr.State("build")],
+                                inputs=[sess, hf_login, hf_login, gr.State("build")],
                                 outputs=[log_b])
                     r_btn.click(fetch_logs,
-                                inputs=[sess, profile_state, token_state, gr.State("run")],
+                                inputs=[sess, hf_login, hf_login, gr.State("run")],
                                 outputs=[log_r])
 
-            # always‑visible panels
+            # Persistent panels
             gr.Markdown("---")
             iframe_out = gr.HTML(label="🖼️ Preview")
             log_out    = gr.Textbox(label="📋 Latest Log", lines=4)
@@ -284,8 +262,12 @@ with gr.Blocks(css="""
 
     state = gr.State({})
     send.click(process_message,
-               inputs=[profile_state, token_state, user_in, gemini_key, repo_name, sdk_choice, chatbox, state],
+               inputs=[hf_login, hf_login, user_in, gemini_key, repo_name, sdk_choice, chatbox, state],
                outputs=[chatbox, iframe_out, log_out, files_out, state])
 
+    sync_btn.click(sync_manual,
+                   inputs=[hf_login, hf_login, state],
+                   outputs=[iframe_out, log_out, files_out, state])
+
 if __name__ == "__main__":
     demo.launch()