Update app.py

app.py

@@ -46,18 +46,12 @@ def create_space(repo_name, sdk, profile, token):
     return rid, log, iframe
 
 def write_file(path, content, repo_id, profile, token):
-    """
-    Write or overwrite any file in the Space. 
-    The LLM can call this for app.py, requirements.txt, README.md, etc.
-    """
     if not (profile and token):
         return "⚠️ Please log in first."
     if not repo_id:
         return "⚠️ Please create a Space first."
-    # Save locally
     with open(path, "w") as f:
         f.write(content)
-    # Upload to HF Space
     upload_file(path, path, repo_id,
                 token=token.token, repo_type="space")
     return f"✅ Wrote and uploaded `{path}`"
@@ -107,49 +101,23 @@ def get_container_logs(repo_id, profile, token):
 func_decls = [
     {
       "name":"create_space","description":"Create/get a HF Space",
-      "parameters":{
-        "type":"object",
-        "properties":{
-          "repo_name":{"type":"string"},
-          "sdk":{"type":"string","enum":["gradio","streamlit"]}
-        },
-        "required":["repo_name","sdk"]
-      }
+      "parameters":{...}
     },
     {
       "name":"write_file","description":"Write/overwrite a file in the Space",
-      "parameters":{
-        "type":"object",
-        "properties":{
-          "path":{"type":"string","description":"File path, e.g. app.py or requirements.txt"},
-          "content":{"type":"string","description":"The full text content to write to the file."}
-        },
-        "required":["path","content"]
-      }
+      "parameters":{...}
     },
     {
       "name":"list_files","description":"List files in the Space",
-      "parameters":{
-        "type":"object",
-        "properties":{"repo_id":{"type":"string"}},
-        "required":["repo_id"]
-      }
+      "parameters":{...}
     },
     {
       "name":"get_build_logs","description":"Fetch build logs",
-      "parameters":{
-        "type":"object",
-        "properties":{"repo_id":{"type":"string"}},
-        "required":["repo_id"]
-      }
+      "parameters":{...}
     },
     {
       "name":"get_run_logs","description":"Fetch run logs",
-      "parameters":{
-        "type":"object",
-        "properties":{"repo_id":{"type":"string"}},
-        "required":["repo_id"]
-      }
+      "parameters":{...}
     }
 ]
 
@@ -176,7 +144,7 @@ def process_message(profile, token, user_msg,
         session["repo_id"]  = None
         session["messages"]= []
 
-    # Allow “call it NAME” in free chat
+    # Quick create via “call it NAME”
     if session["repo_id"] is None:
         m = re.search(r"call (?:it )?([A-Za-z0-9_-]+)", user_msg, re.IGNORECASE)
         if m:
@@ -192,8 +160,10 @@ def process_message(profile, token, user_msg,
                 "role":"assistant",
                 "content":f"✅ Created Space `{name}`. {log}"
             })
-            return (session["messages"],
-                    iframe, log, "", session)
+            return (
+                session["messages"],
+                iframe, log, "", session
+            )
 
     # Record user message
     session["messages"].append({"role":"user","content":user_msg})
@@ -214,42 +184,42 @@ def process_message(profile, token, user_msg,
                 args["repo_name"], args["sdk"], profile, token
             )
             session["repo_id"] = rid
-            result = {"log":log,"iframe":iframe}
+            result = {"log": log, "iframe": iframe}
 
         elif fn == "write_file":
             status = write_file(
                 args["path"], args["content"],
                 session["repo_id"], profile, token
             )
-            result = {"status":status}
+            result = {"status": status}
 
         elif fn == "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 fn == "get_build_logs":
-            result = {"log":get_build_logs(
+            result = {"log": get_build_logs(
                 session["repo_id"], profile, token
             )}
 
         elif fn == "get_run_logs":
-            result = {"log":get_container_logs(
+            result = {"log": get_container_logs(
                 session["repo_id"], profile, token
             )}
 
         else:
-            result = {"log":f"⚠️ Unknown function {fn}"}
+            result = {"log": f"⚠️ Unknown function {fn}"}
 
-        # **Always** pass a JSON string here
+        # Pass the raw dict so Pydantic can validate it
         session["chat"].send_message(
             types.Content(
                 role="function",
                 parts=[ types.Part(
                     function_call=part.function_call,
-                    function_response=json.dumps(result)
-                )]
+                    function_response=result
+                ) ]
             )
         )
         assistant_text = session["chat"].get_history()[-1].parts[0].text