Update app.py

app.py

@@ -1,4 +1,4 @@
-import json, os
+import re, json, os
 import gradio as gr
 from huggingface_hub import (
     create_repo, list_models, upload_file, list_repo_files, constants
@@ -35,14 +35,11 @@ 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} (SDK: {sdk})",
-        f'<iframe src="{url}" width="100%" height="400px"></iframe>'
-    )
+    log = f"✅ Space ready: {url} (SDK: {sdk})"
+    iframe = f'<iframe src="{url}" width="100%" height="400px"></iframe>'
+    return rid, log, iframe
 
 def upload_file_to_space(file, path, repo_id, profile, token):
     if not (profile and token):
@@ -55,10 +52,8 @@ 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}"
@@ -85,9 +80,8 @@ 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"},
@@ -97,27 +91,24 @@ 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"]
@@ -125,12 +116,12 @@ func_decls = [
     },
 ]
 
-# — CHAT HANDLER —
+# — CHAT HANDLER — 
 
 def process_message(profile, token, user_msg,
                     gemini_key, sidebar_repo, sidebar_sdk,
                     chat_history, session):
-    # Initialize on first call
+    # 1) Initialize chat on first call
     if session.get("chat") is None:
         client = genai.Client(api_key=gemini_key)
         cfg = types.GenerateContentConfig(
@@ -141,60 +132,79 @@ def process_message(profile, token, user_msg,
             temperature=0,
             tools=[ types.Tool(function_declarations=func_decls) ]
         )
-        session["chat"]    = client.chats.create(model="gemini-2.0-flash", config=cfg)
-        session["repo_id"] = None
+        session["chat"]     = client.chats.create(model="gemini-2.0-flash", config=cfg)
+        session["repo_id"]  = None
         session["messages"] = []
-
-    # Record user message
+    
+    # 1.5) Auto‐create from “call it X” in free chat if no space yet:
+    if session["repo_id"] is None:
+        m = re.search(r"call it ([A-Za-z0-9_-]+)", user_msg, re.IGNORECASE)
+        if m:
+            name = m.group(1)
+            rid, log, iframe = create_space(name, sidebar_sdk, profile, token)
+            session["repo_id"] = rid
+            # update panels & messages
+            session["iframe"] = iframe
+            session["log"]    = log
+            session["files"]  = ""
+            session["messages"].append({"role":"assistant",
+                "content":f"Created Space `{name}` with SDK `{sidebar_sdk}`. {log}"})
+            return (session["messages"], iframe, log, "", session)
+
+    # Record user turn
     session["messages"].append({"role":"user","content":user_msg})
 
+    # 2) Send to Gemini
     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
+        # parse args
         args = part.function_call.args
         if isinstance(args, str):
             args = json.loads(args)
-
         name = part.function_call.name
+
         if name == "create_space":
             rid, log, iframe = create_space(
                 args["repo_name"], args["sdk"], profile, token
             )
             session["repo_id"] = rid
-            result = {"log": log, "iframe": iframe}
+            result = {"log":log,"iframe":iframe}
+
         elif name == "list_files":
             fl = list_repo_files(session["repo_id"], token=token.token, repo_type="space")
-            result = {"files": "\n".join(fl)}
+            result = {"files":"\n".join(fl)}
+
         elif name == "get_build_logs":
-            result = {"log": get_build_logs(session["repo_id"], profile, token)}
+            result = {"log":get_build_logs(session["repo_id"], profile, token)}
+
         elif name == "get_run_logs":
-            result = {"log": get_container_logs(session["repo_id"], profile, token)}
+            result = {"log":get_container_logs(session["repo_id"], profile, token)}
+
         else:
-            result = {"log": f"⚠️ Unknown function {name}"}
+            result = {"log":f"⚠️ Unknown function {name}"}
 
-        # Send the function result back into the chat
+        # feed the function result back as a dict
         session["chat"].send_message(
             types.Content(
                 role="function",
                 parts=[ types.Part(
                     function_call=part.function_call,
-                    function_response=json.dumps(result)
+                    function_response=result    # <-- dict, not json.dumps
                 )]
             )
         )
-        # The assistant’s final message
-        assistant_text = session["chat"].get_history()[-1].parts[0].text
+        # assistant’s final turn
+        assistant_content = session["chat"].get_history()[-1].parts[0].text
     else:
-        assistant_text = part.text
+        assistant_content = part.text
 
-    # Record assistant message
-    session["messages"].append({"role":"assistant","content":assistant_text})
+    # Record assistant turn
+    session["messages"].append({"role":"assistant","content":assistant_content})
 
-    # Update the panels
+    # Update 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"]
@@ -207,14 +217,14 @@ def process_message(profile, token, user_msg,
         session
     )
 
+# — SYNC MANUAL CHANGES — 
+
 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."
@@ -225,7 +235,7 @@ def sync_manual(profile, token, session):
         session
     )
 
-# — BUILD UI —
+# — BUILD THE UI — 
 
 with gr.Blocks(css="""
 #sidebar { background:#f2f2f2; padding:1rem; border-right:1px solid #ccc; }
@@ -240,7 +250,7 @@ with gr.Blocks(css="""
             profile_state = gr.State(None)
             token_state   = gr.State(None)
 
-            # Capture both profile and token from the login button
+            # capture both
             login_btn.click(None, [], [profile_state, token_state])
 
             status_md = gr.Markdown("*Not logged in.*")
@@ -276,8 +286,7 @@ with gr.Blocks(css="""
                                            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,
@@ -289,8 +298,7 @@ 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,
@@ -300,10 +308,8 @@ 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],
@@ -324,8 +330,7 @@ with gr.Blocks(css="""
                            user_input, gemini_key,
                            sidebar_repo, sidebar_sdk,
                            chatbox, state],
-                   outputs=[chatbox, iframe_out,
-                            log_out, files_out, state])
+                   outputs=[chatbox, iframe_out, log_out, files_out, state])
 
     confirm_btn.click(sync_manual,
                       inputs=[profile_state, token_state, state],