Update app.py

app.py

@@ -35,9 +35,14 @@ def create_space(repo_name, sdk, profile, token):
     if not (profile and token):
         return "", "⚠️ Please log in first.", "<p>No Space created yet.</p>"
     rid = f"{profile.username}/{repo_name}"
-    create_repo(rid, token=token.token, exist_ok=True, repo_type="space", space_sdk=sdk)
+    create_repo(rid, token=token.token, exist_ok=True,
+                repo_type="space", space_sdk=sdk)
     url = f"https://huggingface.co/spaces/{rid}"
-    return rid, f"✅ Space ready: {url}", f'<iframe src="{url}" width="100%" height="400px"></iframe>'
+    return (
+        rid,
+        f"✅ Space ready: {url} (SDK: {sdk})",
+        f'<iframe src="{url}" width="100%" height="400px"></iframe>'
+    )
 
 def upload_file_to_space(file, path, repo_id, profile, token):
     if not (profile and token):
@@ -50,7 +55,10 @@ def upload_file_to_space(file, path, repo_id, profile, token):
     return f"✅ Uploaded `{path}` to `{repo_id}`"
 
 def _fetch_space_logs_level(repo_id, level):
-    r = get_session().get(f"{constants.ENDPOINT}/api/spaces/{repo_id}/jwt", headers=build_hf_headers())
+    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}"
@@ -77,8 +85,9 @@ def get_container_logs(repo_id, profile, token):
 
 func_decls = [
     {
-      "name":"create_space","description":"Create/get a HF Space",
-      "parameters":{
+      "name": "create_space",
+      "description": "Create/get a HF Space",
+      "parameters": {
         "type":"object",
         "properties":{
           "repo_name":{"type":"string"},
@@ -88,24 +97,27 @@ func_decls = [
       }
     },
     {
-      "name":"list_files","description":"List files in Space",
-      "parameters":{
+      "name": "list_files",
+      "description": "List files in Space",
+      "parameters": {
         "type":"object",
         "properties":{"repo_id":{"type":"string"}},
         "required":["repo_id"]
       }
     },
     {
-      "name":"get_build_logs","description":"Fetch build logs",
-      "parameters":{
+      "name": "get_build_logs",
+      "description": "Fetch build logs",
+      "parameters": {
         "type":"object",
         "properties":{"repo_id":{"type":"string"}},
         "required":["repo_id"]
       }
     },
     {
-      "name":"get_run_logs","description":"Fetch run logs",
-      "parameters":{
+      "name": "get_run_logs",
+      "description": "Fetch run logs",
+      "parameters": {
         "type":"object",
         "properties":{"repo_id":{"type":"string"}},
         "required":["repo_id"]
@@ -118,6 +130,7 @@ func_decls = [
 def process_message(profile, token, user_msg,
                     gemini_key, sidebar_repo, sidebar_sdk,
                     chat_history, session):
+    # Initialize on first call
     if session.get("chat") is None:
         client = genai.Client(api_key=gemini_key)
         cfg = types.GenerateContentConfig(
@@ -130,16 +143,23 @@ def process_message(profile, token, user_msg,
         )
         session["chat"]    = client.chats.create(model="gemini-2.0-flash", config=cfg)
         session["repo_id"] = None
+        session["messages"] = []
 
-    chat = session["chat"]
-    chat_history.append((user_msg, None))
-    resp = chat.send_message(user_msg)
+    # Record user message
+    session["messages"].append({"role":"user","content":user_msg})
+
+    resp = session["chat"].send_message(user_msg)
     part = resp.candidates[0].content.parts[0]
 
     result = {}
+    # If the model wants to call a function
     if part.function_call:
+        # args may already be dict
+        args = part.function_call.args
+        if isinstance(args, str):
+            args = json.loads(args)
+
         name = part.function_call.name
-        args = json.loads(part.function_call.args)
         if name == "create_space":
             rid, log, iframe = create_space(
                 args["repo_name"], args["sdk"], profile, token
@@ -156,7 +176,8 @@ def process_message(profile, token, user_msg,
         else:
             result = {"log": f"⚠️ Unknown function {name}"}
 
-        chat.send_message(
+        # Send the function result back into the chat
+        session["chat"].send_message(
             types.Content(
                 role="function",
                 parts=[ types.Part(
@@ -165,40 +186,50 @@ def process_message(profile, token, user_msg,
                 )]
             )
         )
-        final = chat.get_history()[-1].parts[0].text
+        # The assistant’s final message
+        assistant_text = session["chat"].get_history()[-1].parts[0].text
     else:
-        final = part.text
+        assistant_text = part.text
 
+    # Record assistant message
+    session["messages"].append({"role":"assistant","content":assistant_text})
+
+    # Update the panels
     if "iframe" in result: session["iframe"] = result["iframe"]
     if "log"    in result: session["log"]    = result["log"]
     if "files"  in result: session["files"]  = result["files"]
 
-    chat_history[-1] = (user_msg, final)
-    return (chat_history,
-            session.get("iframe",""),
-            session.get("log",""),
-            session.get("files",""),
-            session)
+    return (
+        session["messages"],
+        session.get("iframe",""),
+        session.get("log",""),
+        session.get("files",""),
+        session
+    )
 
 def sync_manual(profile, token, session):
     if not (profile and token and session.get("repo_id")):
-        return (session.get("iframe",""),
-                "⚠️ Cannot sync manual changes.",
-                session.get("files",""),
-                session)
+        return (
+            session.get("iframe",""),
+            "⚠️ Cannot sync manual changes.",
+            session.get("files",""),
+            session
+        )
     fl = list_repo_files(session["repo_id"], token=token.token, repo_type="space")
     session["files"] = "\n".join(fl)
     session["log"]   = "🔄 Manual changes synced."
-    return (session.get("iframe",""),
-            session["log"],
-            session["files"],
-            session)
+    return (
+        session.get("iframe",""),
+        session["log"],
+        session["files"],
+        session
+    )
 
 # — BUILD UI —
 
 with gr.Blocks(css="""
-#sidebar { background: #f2f2f2; padding: 1rem; border-right:1px solid #ccc; }
-#main    { padding: 1rem; }
+#sidebar { background:#f2f2f2; padding:1rem; border-right:1px solid #ccc; }
+#main    { padding:1rem; }
 """) as demo:
 
     with gr.Row():
@@ -209,11 +240,13 @@ with gr.Blocks(css="""
             profile_state = gr.State(None)
             token_state   = gr.State(None)
 
-            # capture both outputs
+            # Capture both profile and token from the login button
             login_btn.click(None, [], [profile_state, token_state])
 
             status_md = gr.Markdown("*Not logged in.*")
-            profile_state.change(show_profile, inputs=[profile_state], outputs=[status_md])
+            profile_state.change(show_profile,
+                                 inputs=[profile_state],
+                                 outputs=[status_md])
 
             models_md = gr.Markdown()
             profile_state.change(list_private_models,
@@ -222,7 +255,8 @@ with gr.Blocks(css="""
 
             gemini_key   = gr.Textbox(label="Gemini API Key", type="password")
             sidebar_repo = gr.Textbox(label="Space name", placeholder="my-space")
-            sidebar_sdk  = gr.Radio(["gradio","streamlit"], value="gradio", label="SDK")
+            sidebar_sdk  = gr.Radio(["gradio","streamlit"],
+                                    value="gradio", label="SDK")
 
             gr.Markdown("---")
             confirm_btn = gr.Button("🔄 Confirm Manual Changes")
@@ -232,16 +266,18 @@ with gr.Blocks(css="""
             tabs = gr.Tabs()
             with tabs:
                 with gr.TabItem("💬 Chat"):
-                    chatbox    = gr.Chatbot(type="tuples")
+                    chatbox    = gr.Chatbot(type="messages")
                     user_input = gr.Textbox(show_label=False, placeholder="Ask the LLM…")
                     send_btn   = gr.Button("Send")
                 with gr.TabItem("🛠️ Manual"):
                     gr.Markdown("#### Create a Space")
                     repo_m     = gr.Textbox(label="Name")
-                    sdk_m      = gr.Radio(["gradio","streamlit"], value="gradio", label="SDK")
+                    sdk_m      = gr.Radio(["gradio","streamlit"],
+                                           value="gradio", label="SDK")
                     create_btn = gr.Button("Create Space")
                     sess_id    = gr.Textbox(visible=False)
-                    log_c      = gr.Textbox(label="Log", interactive=False, lines=2)
+                    log_c      = gr.Textbox(label="Log",
+                                            interactive=False, lines=2)
                     preview    = gr.HTML("<p>No Space yet.</p>")
 
                     create_btn.click(create_space,
@@ -253,7 +289,8 @@ with gr.Blocks(css="""
                     path        = gr.Textbox(label="Path", value="app.py")
                     file_u      = gr.File()
                     upload_btn  = gr.Button("Upload File")
-                    log_u       = gr.Textbox(label="Log", interactive=False, lines=2)
+                    log_u       = gr.Textbox(label="Log",
+                                              interactive=False, lines=2)
 
                     upload_btn.click(upload_file_to_space,
                                      inputs=[file_u, path, sess_id,
@@ -263,8 +300,10 @@ with gr.Blocks(css="""
                     gr.Markdown("#### Fetch Logs")
                     b_btn = gr.Button("Build Logs")
                     r_btn = gr.Button("Run Logs")
-                    log_b = gr.Textbox(label="Build", interactive=False, lines=5)
-                    log_r = gr.Textbox(label="Run",   interactive=False, lines=5)
+                    log_b = gr.Textbox(label="Build",
+                                       interactive=False, lines=5)
+                    log_r = gr.Textbox(label="Run",
+                                       interactive=False, lines=5)
 
                     b_btn.click(get_build_logs,
                                 inputs=[sess_id, profile_state, token_state],
@@ -273,7 +312,7 @@ with gr.Blocks(css="""
                                 inputs=[sess_id, profile_state, token_state],
                                 outputs=[log_r])
 
-            # persistent panels
+            # Persistent panels
             gr.Markdown("---")
             iframe_out = gr.HTML(label="🖼️ Preview")
             log_out    = gr.Textbox(label="📋 Latest Log", lines=4)