Update app.py

app.py

@@ -3,6 +3,7 @@ import json
 import os
 
 import gradio as gr
+from PIL import Image, ImageFilter
 from huggingface_hub import (
     create_repo, list_models, upload_file, list_repo_files, constants
 )
@@ -38,7 +39,7 @@ def enable_repo_actions(repo_id, profile, token):
 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}"
+    rid = f"{profile.username}/{repo_name.replace(' ', '-') }"
     create_repo(
         repo_id=rid,
         token=token.token,
@@ -110,9 +111,7 @@ def get_container_logs(repo_id, profile, token):
         return "⚠️ Please log in and create a Space first."
     return _fetch_space_logs_level(repo_id, "run")
 
-
-# — FUNCTION DECLARATIONS & TOOL — 
-# Build a single Tool containing all function declarations.
+# — FUNCTION DECLARATIONS & TOOL —
 
 func_decls = [
     types.FunctionDeclaration(
@@ -169,16 +168,16 @@ func_decls = [
 ]
 tool = types.Tool(function_declarations=func_decls)
 
-
-# — CHAT HANDLER — 
+# — CHAT HANDLER —
 
 def execute_function_by_name(name, args, profile, token, session):
-    """Dispatch to the right function and return a raw dict result."""
     if name == "create_space":
         rid, log, iframe = create_space(
             args["repo_name"], session["sdk"], profile, token
         )
         session["repo_id"] = rid
+        session["iframe"] = iframe
+        session["log"] = log
         return {"log": log, "iframe": iframe}
     if name == "write_file":
         status = write_file(
@@ -201,18 +200,15 @@ def execute_function_by_name(name, args, profile, token, session):
         )}
     return {"error": f"Unknown function {name}"}
 
-
 def process_message(
     profile, token, user_msg,
     gemini_key, sidebar_repo, sidebar_sdk,
     chat_history, session
 ):
-    # Initialize chat once
     if session.get("chat") is None:
         client = genai.Client(api_key=gemini_key)
-        # Pass the Tool object, not a list of dicts
         model = client.chats.create(
-            model="gemini-2.0-flash",
+            model="gemini-2.5-flash-preview-04-17",
             config=types.GenerateContentConfig(
                 system_instruction=(
                     "You are a HF Spaces admin. You can create spaces, "
@@ -228,150 +224,97 @@ def process_message(
             "sdk": sidebar_sdk,
             "messages": []
         })
-
-    # Remember current SDK choice
     session["sdk"] = sidebar_sdk
-
-    # Pre-chat “call it X” handler
     if session["repo_id"] is None:
-        m = re.search(r"call (?:it )?([A-Za-z0-9_-]+)", user_msg)
+        m = re.search(r"(?:call it|called)\s+([\w- ]+)", user_msg, re.IGNORECASE)
         if m:
             args = {"repo_name": m.group(1), "sdk": sidebar_sdk}
             result = execute_function_by_name("create_space", args, profile, token, session)
             session["messages"].append({"role": "assistant", "content": result["log"]})
-            return session["messages"], result.get("iframe", ""), result["log"], "", session
-
-    # Append user message
+            return session["messages"], session["iframe"], session["log"], "", session
     session["messages"].append({"role": "user", "content": user_msg})
     response = session["chat"].send_message(user_msg)
-
-    # Handle function-calling loop
     while True:
         part = response.candidates[0].content.parts[0]
         if not part.function_call:
-            break  # no more tools needed
+            break
         fname = part.function_call.name
         fargs = part.function_call.args or {}
-        # Run the tool
         result = execute_function_by_name(fname, fargs, profile, token, session)
-        # Send function result back to model
         resp_part = types.Part.from_function_response(name=fname, response=result)
-        response = session["chat"].send_message(types.Content(role="user", parts=[resp_part]))
-
-    # Final assistant text
+        response = session["chat"].send_message(resp_part)
     assistant_text = response.text
     session["messages"].append({"role": "assistant", "content": assistant_text})
-
-    # Update UI panels
-    iframe = session.get("iframe", "")
-    log    = session.get("log", "")
-    files  = session.get("files", "")
-    return session["messages"], iframe, log, files, session
-
-
-# — MANUAL SYNC — 
+    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 "", "⚠️ Cannot sync manual changes.", "", session
     files = list_repo_files(session["repo_id"], token=token.token, repo_type="space")
     log   = "🔄 Manual changes synced."
-    return session.get("iframe",""), log, "\n".join(files), session
-
-
-# — GRADIO UI — 
+    return session.get("iframe", ""), log, "\n".join(files), session
 
 with gr.Blocks(css="""
 #sidebar { background:#f2f2f2; padding:1rem; border-right:1px solid #ccc; }
 #main    { padding:1rem; }
 """) as demo:
-
-    # Sidebar
     with gr.Column(elem_id="sidebar", scale=1):
         gr.Markdown("### 🔑 HF Login & Config")
         login_btn     = gr.LoginButton(variant="huggingface", size="sm")
         profile_state = gr.State(None)
         token_state   = gr.State(None)
         login_btn.click(None, [], [profile_state, token_state])
-
         status_md = gr.Markdown("*Not logged in.*")
         profile_state.change(show_profile, [profile_state], [status_md])
-
         models_md = gr.Markdown()
-        profile_state.change(list_private_models,
-                             [profile_state, token_state], [models_md])
-
+        profile_state.change(list_private_models,[profile_state, token_state],[models_md])
         gemini_key   = gr.Textbox(label="Gemini API Key", type="password")
         sidebar_repo = gr.Textbox(label="Space name", placeholder="blurrtest")
-        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("🔄 Sync Manual Changes")
-
-    # Main area
     with gr.Column(elem_id="main", scale=3):
         tabs = gr.Tabs()
         with tabs:
             with gr.TabItem("💬 Chat"):
-                chatbox    = gr.Chatbot(type="messages")
-                user_input = gr.Textbox(show_label=False,
-                                        placeholder="Generate code…")
-                send_btn   = gr.Button("Send")
+                chatbox = gr.Chatbot(type="messages")
+                user_input = gr.Textbox(show_label=False, placeholder="Generate code…")
+                send_btn = gr.Button("Send")
             with gr.TabItem("🛠️ Manual"):
                 gr.Markdown("#### Create / Rename Space")
-                repo_m     = gr.Textbox(label="Name", value="")
-                sdk_m      = gr.Radio(["gradio","streamlit"],
-                                       value="gradio", label="SDK")
+                repo_m = gr.Textbox(label="Name", value="")
+                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)
-                preview    = gr.HTML("<p>No Space yet.</p>")
-
-                create_btn.click(create_space,
-                                 [repo_m, sdk_m, profile_state, token_state],
-                                 [sess_id, log_c, preview])
-
+                sess_id = gr.Textbox(visible=False)
+                log_c = gr.Textbox(label="Log",interactive=False,lines=2)
+                preview = gr.HTML("<p>No Space yet.</p>")
+                create_btn.click(create_space,[repo_m,sdk_m,profile_state,token_state],[sess_id,log_c,preview])
                 gr.Markdown("#### Upload File")
-                path        = gr.Textbox(label="Path", value="app.py")
-                file_u      = gr.File()
-                upload_btn  = gr.Button("Upload File")
-                log_u       = gr.Textbox(label="Upload Log",
-                                          interactive=False, lines=2)
-
-                upload_btn.click(upload_file_to_space,
-                                 [file_u, path, sess_id, profile_state, token_state],
-                                 [log_u])
-
+                path = gr.Textbox(label="Path",value="app.py")
+                file_u = gr.File()
+                upload_btn = gr.Button("Upload File")
+                log_u = gr.Textbox(label="Upload Log",interactive=False,lines=2)
+                upload_btn.click(upload_file_to_space,[file_u,path,sess_id,profile_state,token_state],[log_u])
                 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)
-
-                b_btn.click(get_build_logs,
-                            [sess_id, profile_state, token_state],
-                            [log_b])
-                r_btn.click(get_container_logs,
-                            [sess_id, profile_state, token_state],
-                            [log_r])
-
+                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,[sess_id,profile_state,token_state],[log_b])
+                r_btn.click(get_container_logs,[sess_id,profile_state,token_state],[log_r])
         gr.Markdown("---")
         iframe_out = gr.HTML(label="🖼️ Preview")
-        log_out    = gr.Textbox(label="📋 Latest Log", lines=4)
-        files_out  = gr.Textbox(label="📚 Files", lines=4)
-
+        log_out = gr.Textbox(label="📋 Latest Log",lines=4)
+        files_out = gr.Textbox(label="📚 Files",lines=4)
     state = gr.State({})
-    send_btn.click(process_message,
-                   [profile_state, token_state,
-                    user_input, gemini_key,
-                    sidebar_repo, sidebar_sdk,
-                    chatbox, state],
-                   [chatbox, iframe_out, log_out, files_out, state])
-
-    confirm_btn.click(sync_manual,
-                      [profile_state, token_state, state],
-                      [iframe_out, log_out, files_out, state])
+    send_btn.click(process_message,[profile_state,token_state,user_input,gemini_key,sidebar_repo,sidebar_sdk,chatbox,state],[chatbox,iframe_out,log_out,files_out,state])
+    confirm_btn.click(sync_manual,[profile_state,token_state,state],[iframe_out,log_out,files_out,state])
 
 if __name__ == "__main__":
     demo.launch()