Update app.py

app.py

@@ -1,6 +1,5 @@
 import os, json
 import gradio as gr
-import requests
 from huggingface_hub import (
     create_repo,
     list_models,
@@ -196,16 +195,13 @@ def process_message(
         session["repo_id"] = None
 
     chat = session["chat"]
-    # user turn
     chat_history = chat_history + [(user_msg, None)]
     resp = chat.send_message(user_msg)
     part = resp.candidates[0].content.parts[0]
 
-    # function call?
     if part.function_call:
         name = part.function_call.name
         args = json.loads(part.function_call.args)
-        # dispatch
         if name=="create_space":
             rid, log, iframe = create_space(args["repo_name"], args["sdk"], profile, oauth_token)
             session["repo_id"] = rid
@@ -220,7 +216,6 @@ def process_message(
         else:
             result = {"log":f"⚠️ Unknown function {name}"}
 
-        # send back
         chat.send_message(
             types.Content(
                 role="function",
@@ -231,7 +226,7 @@ def process_message(
     else:
         final = part.text
 
-    # pick up panels
+    # panels
     iframe = session.get("iframe", "")
     log    = session.get("log", "")
     files  = session.get("files", "")
@@ -248,7 +243,7 @@ def process_message(
 
 with gr.Blocks(title="HF Spaces + Gemini Chat") as demo:
     with gr.Row():
-        # ◀ Sidebar: login, models, chat config
+        # ◀ Sidebar: login + chat config
         with gr.Column(scale=1):
             gr.Markdown("## 🔑 Sign in & Chat Config")
             login_btn   = gr.LoginButton(variant="huggingface", size="sm")
@@ -262,10 +257,10 @@ with gr.Blocks(title="HF Spaces + Gemini Chat") as demo:
             gemini_key  = gr.Textbox(label="Gemini API Key", type="password")
             repo_name   = gr.Textbox(label="Space name", placeholder="e.g. my-space")
             sdk_choice  = gr.Radio(["gradio","streamlit"], value="gradio", label="SDK")
-        # ▶ Main: chat + panels + manual UI
+        # ▶ Main: chat + panels + manual controls
         with gr.Column(scale=3):
             gr.Markdown("## 💬 Chat to manage your Space")
-            chatbox   = gr.Chatbot()
+            chatbox   = gr.Chatbot(type="messages")
             user_in   = gr.Textbox(show_label=False, placeholder="e.g. Create Space")
             send_btn  = gr.Button("Send")
 
@@ -273,52 +268,54 @@ with gr.Blocks(title="HF Spaces + Gemini Chat") as demo:
             log_out    = gr.Textbox(label="📋 Latest Log", lines=8)
             files_out  = gr.Textbox(label="📚 Files in Space", lines=4)
 
-            gr.Markdown("---\n## 🛠️ Manual Controls (unchanged)")
-            # — Manual CREATE SPACE —
-            repo_name_m   = gr.Textbox(label="New Space name", placeholder="my-space")
-            sdk_selector  = gr.Radio(["gradio","streamlit"], value="gradio", label="SDK for manual")
-            create_btn    = gr.Button("Create Space", interactive=False)
-            session_id    = gr.Textbox(visible=False)
-            create_logs   = gr.Textbox(label="Create Logs", interactive=False, lines=3)
-            preview_m     = gr.HTML("<p>No Space created yet.</p>")
-
-            demo.load(enable_create, inputs=None, outputs=[create_btn])
-            login_btn.click(enable_create, inputs=None, outputs=[create_btn])
+            gr.Markdown("---\n## 🛠️ Manual Controls")
+            # CREATE SPACE
+            repo_name_m  = gr.Textbox(label="New Space name", placeholder="my-space")
+            sdk_selector = gr.Radio(["gradio","streamlit"], value="gradio", label="SDK")
+            create_btn   = gr.Button("Create Space", interactive=False)
+            session_id   = gr.Textbox(visible=False)
+            create_logs  = gr.Textbox(label="Create Logs", interactive=False, lines=3)
+            preview_m    = gr.HTML("<p>No Space created yet.</p>")
+
+            demo.load(enable_create,        inputs=None,        outputs=[create_btn])
+            login_btn.click(enable_create,  inputs=None,        outputs=[create_btn])
             create_btn.click(
                 fn=create_space,
-                inputs=[repo_name_m, sdk_selector, status_md, login_btn.token],
+                inputs=[repo_name_m, sdk_selector],
                 outputs=[session_id, create_logs, preview_m]
             )
 
-            # — Manual UPLOAD FILES —
-            path_in_repo   = gr.Textbox(label="Path in Space", value="app.py")
-            file_uploader  = gr.File(label="Select file")
-            upload_btn     = gr.Button("Upload File", interactive=False)
-            upload_logs    = gr.Textbox(label="Upload Logs", interactive=False, lines=2)
+            # UPLOAD FILE
+            path_in_repo = gr.Textbox(label="Path in Space", value="app.py")
+            file_uploader = gr.File(label="Select file")
+            upload_btn   = gr.Button("Upload File", interactive=False)
+            upload_logs  = gr.Textbox(label="Upload Logs", interactive=False, lines=2)
 
-            demo.load(enable_repo_actions, inputs=[session_id, status_md, login_btn.token], outputs=[upload_btn])
+            demo.load(enable_repo_actions, inputs=[session_id], outputs=[upload_btn])
+            login_btn.click(enable_repo_actions, inputs=[session_id], outputs=[upload_btn])
             upload_btn.click(
                 fn=upload_file_to_space,
-                inputs=[file_uploader, path_in_repo, session_id, status_md, login_btn.token],
+                inputs=[file_uploader, path_in_repo, session_id],
                 outputs=[upload_logs]
             )
 
-            # — Manual FETCH LOGS —
-            build_btn      = gr.Button("Get Build Logs", interactive=False)
-            run_btn        = gr.Button("Get Container Logs", interactive=False)
-            build_logs_md  = gr.Textbox(label="Build Logs", interactive=False, lines=10)
-            run_logs_md    = gr.Textbox(label="Container Logs", interactive=False, lines=10)
+            # FETCH LOGS
+            build_btn     = gr.Button("Get Build Logs", interactive=False)
+            run_btn       = gr.Button("Get Container Logs", interactive=False)
+            build_logs_md = gr.Textbox(label="Build Logs", interactive=False, lines=10)
+            run_logs_md   = gr.Textbox(label="Container Logs", interactive=False, lines=10)
 
             for b in (build_btn, run_btn):
-                demo.load(enable_repo_actions, inputs=[session_id, status_md, login_btn.token], outputs=[b])
+                demo.load(enable_repo_actions, inputs=[session_id], outputs=[b])
+                login_btn.click(enable_repo_actions, inputs=[session_id], outputs=[b])
 
-            build_btn.click(fn=get_build_logs, inputs=[session_id, status_md, login_btn.token], outputs=[build_logs_md])
-            run_btn.click(  fn=get_container_logs, inputs=[session_id, status_md, login_btn.token], outputs=[run_logs_md])
+            build_btn.click(fn=get_build_logs,      inputs=[session_id], outputs=[build_logs_md])
+            run_btn.click(  fn=get_container_logs,  inputs=[session_id], outputs=[run_logs_md])
 
     state = gr.State({})
     send_btn.click(
         fn=process_message,
-        inputs=[user_in, status_md, login_btn.token, gemini_key, repo_name, sdk_choice, chatbox, state],
+        inputs=[user_in, status_md, login_btn, gemini_key, repo_name, sdk_choice, chatbox, state],
         outputs=[chatbox, iframe_out, log_out, files_out, state]
     )