Update app.py

app.py

@@ -1,228 +1,163 @@
-import gradio as gr
-import requests
+import os
 import json
-from huggingface_hub import (
-    create_repo,
-    list_models,
-    upload_file,
-    constants,
-)
+from huggingface_hub import create_repo, list_models, upload_file, constants
 from huggingface_hub.utils import build_hf_headers, get_session, hf_raise_for_status
+import gradio as gr
+from google import genai
+from google.genai import types
+
+# --- Initialize Gemini client ---
+client = genai.Client(api_key=os.getenv("GEMINI_API_KEY"))
+# System prompt for Gemini
+system_instruction = (
+    "You are a helpful assistant that writes, debugs, and pushes code to Hugging Face Spaces. "
+    "Treat Hugging Face as a sandbox: create spaces, upload code, and debug via function calls. "
+    "Respond in JSON with {success, data, message}."
+)
 
-# — USER INFO & MODEL LISTING —
-
-def show_profile(profile: gr.OAuthProfile | None) -> str:
-    if profile is None:
-        return "*Not logged in.*"
-    return f"✅ Logged in as **{profile.username}**"
-
-def list_private_models(
-    profile: gr.OAuthProfile | None,
-    oauth_token: gr.OAuthToken | None
-) -> str:
-    if profile is None or oauth_token is None:
-        return "Please log in to see your models."
-    models = [
-        f"{m.id} ({'private' if m.private else 'public'})"
-        for m in list_models(author=profile.username, token=oauth_token.token)
-    ]
-    return "No models found." if not models else "Models:\n\n" + "\n - ".join(models)
-
-# — BUTTON‑ENABLING HELPERS —
-
-def enable_create(
-    profile: gr.OAuthProfile | None,
-    oauth_token: gr.OAuthToken | None
-):
-    return gr.update(interactive=profile is not None)
-
-def enable_repo_actions(
-    repo_id: str,
-    profile: gr.OAuthProfile | None,
-    oauth_token: gr.OAuthToken | None
-):
-    return gr.update(interactive=bool(repo_id and profile and oauth_token))
-
-# — CORE ACTIONS —
-
-def create_space(
-    repo_name: str,
-    sdk: str,
-    profile: gr.OAuthProfile | None,
-    oauth_token: gr.OAuthToken | None
-) -> tuple[str, str, str]:
-    if not profile or not oauth_token:
-        return "", "⚠️ Please log in first.", "<p>No Space created yet.</p>"
-    repo_id = f"{profile.username}/{repo_name}"
-    create_repo(
-        repo_id=repo_id,
-        token=oauth_token.token,
-        exist_ok=True,
-        repo_type="space",
-        space_sdk=sdk
-    )
-    url    = f"https://huggingface.co/spaces/{repo_id}"
-    logmsg = f"✅ Space ready: {url} (SDK: {sdk})"
-    iframe = f'<iframe src="{url}" width="100%" height="500px"></iframe>'
-    return repo_id, logmsg, iframe
-
-def upload_file_to_space(
-    file,
-    path_in_repo: str,
-    repo_id: str,
-    profile: gr.OAuthProfile | None,
-    oauth_token: gr.OAuthToken | None
-) -> str:
-    if not profile or not oauth_token:
-        return "⚠️ Please log in first."
-    if not repo_id:
-        return "⚠️ Please create a Space first."
-    if not file:
-        return "⚠️ No file selected."
-    upload_file(
-        path_or_fileobj=file.name,
-        path_in_repo=path_in_repo,
-        repo_id=repo_id,
-        token=oauth_token.token,
-        repo_type="space"
-    )
-    return f"✅ Uploaded `{path_in_repo}` to `{repo_id}`"
-
-def _fetch_space_logs_level(repo_id: str, level: str) -> str:
-    # 1) Get SSE JWT
+# --- Function declarations for logs (behind the scenes) ---
+get_build_logs_decl = {
+    "name": "get_build_logs",
+    "description": "Fetches build logs for a Space",
+    "parameters": {
+        "type": "object",
+        "properties": {
+            "repo_id": {"type": "string"}
+        },
+        "required": ["repo_id"]
+    }
+}
+get_container_logs_decl = {
+    "name": "get_container_logs",
+    "description": "Fetches container logs for a Space",
+    "parameters": {
+        "type": "object",
+        "properties": {
+            "repo_id": {"type": "string"}
+        },
+        "required": ["repo_id"]
+    }
+}
+tools = [
+    types.Tool(function_declarations=[get_build_logs_decl, get_container_logs_decl])
+]
+
+# --- Core Hugging Face functions (unchanged) ---
+def _fetch_space_logs_level(repo_id: str, level: str):
     jwt_url = f"{constants.ENDPOINT}/api/spaces/{repo_id}/jwt"
     r = get_session().get(jwt_url, headers=build_hf_headers())
     hf_raise_for_status(r)
     jwt = r.json()["token"]
-    # 2) Stream logs
     logs_url = f"https://api.hf.space/v1/{repo_id}/logs/{level}"
-    lines = []
+    records = []
     with get_session().get(logs_url, headers=build_hf_headers(token=jwt), stream=True) as resp:
         hf_raise_for_status(resp)
         for raw in resp.iter_lines():
             if not raw.startswith(b"data: "):
                 continue
-            payload = raw[len(b"data: "):]
             try:
-                event = json.loads(payload.decode())
-            except json.JSONDecodeError:
-                continue
-            ts  = event.get("timestamp", "")
-            txt = event.get("data", "")
-            lines.append(f"[{ts}] {txt}")
-    return "\n".join(lines)
-
-def get_build_logs(
-    repo_id: str,
-    profile: gr.OAuthProfile | None,
-    oauth_token: gr.OAuthToken | None
-) -> str:
-    if not (profile and oauth_token and repo_id):
-        return "⚠️ Please log in and create a Space first."
-    return _fetch_space_logs_level(repo_id, "build")
-
-def get_container_logs(
-    repo_id: str,
-    profile: gr.OAuthProfile | None,
-    oauth_token: gr.OAuthToken | None
-) -> str:
-    if not (profile and oauth_token and repo_id):
-        return "⚠️ Please log in and create a Space first."
-    return _fetch_space_logs_level(repo_id, "run")
-
-# — BUILD THE UI —
-
-with gr.Blocks(title="HF OAuth + Space Manager with Logs") as demo:
-    gr.Markdown(
-        "## Sign in with Hugging Face + Manage Your Space\n\n"
-        "1. Sign in\n"
-        "2. Create a Space (Gradio/Streamlit)\n"
-        "3. Upload files to it\n"
-        "4. Fetch build and container logs\n\n"
-        "---"
-    )
-
-    # — LOGIN & MODEL LIST —
-    login_btn = gr.LoginButton(variant="huggingface", size="lg")
-    status_md = gr.Markdown("*Not logged in.*")
-    models_md = gr.Markdown()
-    demo.load(show_profile,         inputs=None, outputs=status_md)
-    login_btn.click(show_profile,   inputs=None, outputs=status_md)
-    demo.load(list_private_models,  inputs=None, outputs=models_md)
-    login_btn.click(list_private_models,
-                    inputs=None,     outputs=models_md)
-
-    # — CREATE SPACE —
-    repo_name    = gr.Textbox(label="New Space name", placeholder="my-space")
-    sdk_selector = gr.Radio(
-        choices=["gradio","streamlit"],
-        value="gradio",
-        label="Space template (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      = 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, sdk_selector],
-        outputs=[session_id, create_logs, preview]
-    )
-
-    # — 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)
-
-    demo.load(enable_repo_actions,
-              inputs=[session_id],
-              outputs=[upload_btn])
-    login_btn.click(enable_repo_actions,
-                    inputs=[session_id],
-                    outputs=[upload_btn])
-    session_id.change(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],
-        outputs=[upload_logs]
+                event = json.loads(raw[len(b"data: "):].decode())
+                records.append({
+                    "timestamp": event.get("timestamp"),
+                    "message": event.get("data")
+                })
+            except:
+                pass
+    return records
+
+# Space management (hidden behind scenes)
+def create_space_backend(username, token, repo_name, sdk):
+    repo_id = f"{username}/{repo_name}"
+    create_repo(
+        repo_id=repo_id,
+        token=token,
+        exist_ok=True,
+        repo_type="space",
+        space_sdk=sdk
     )
+    return repo_id
 
-    # — FETCH BUILD & CONTAINER LOGS —
-    build_logs_btn     = gr.Button("Get Build Logs", interactive=False)
-    container_logs_btn = gr.Button("Get Container Logs", interactive=False)
-    build_logs_md      = gr.Textbox(label="Build Logs", interactive=False, lines=10)
-    container_logs_md  = gr.Textbox(label="Container Logs", interactive=False, lines=10)
-
-    # enable both log buttons
-    for btn in (build_logs_btn, container_logs_btn):
-        demo.load(enable_repo_actions,
-                  inputs=[session_id],
-                  outputs=[btn])
-        login_btn.click(enable_repo_actions,
-                        inputs=[session_id],
-                        outputs=[btn])
-        session_id.change(enable_repo_actions,
-                          inputs=[session_id],
-                          outputs=[btn])
-
-    build_logs_btn.click(
-        fn=get_build_logs,
-        inputs=[session_id],
-        outputs=[build_logs_md]
+def upload_file_backend(repo_id, token, file_path, path_in_repo):
+    upload_file(
+        path_or_fileobj=file_path,
+        path_in_repo=path_in_repo,
+        repo_id=repo_id,
+        token=token,
+        repo_type="space"
     )
-    container_logs_btn.click(
-        fn=get_container_logs,
-        inputs=[session_id],
-        outputs=[container_logs_md]
+    return True
+
+# --- Chat and interaction ---
+chat = None
+
+def init_chat(repo_id, api_key):
+    global chat
+    os.environ["GEMINI_API_KEY"] = api_key
+    chat = client.chats.create(
+        model="gemini-2.5-flash-preview-04-17",
+        config=types.GenerateContentConfig(
+            system_instruction=system_instruction,
+            tools=tools,
+            temperature=0
+        )
     )
+    return {"success": True, "data": None, "message": f"Sandbox ready: {repo_id}"}
+
+def chatbot_respond(message, history, repo_id, api_key):
+    global chat
+    if chat is None:
+        init_chat(repo_id, api_key)
+    response = chat.send_message(message)
+    part = response.candidates[0].content.parts[0]
+    # Handle function calls
+    if part.function_call:
+        fn = part.function_call
+        args = json.loads(fn.args)
+        if fn.name == "get_build_logs":
+            result = _fetch_space_logs_level(repo_id, "build")
+        else:
+            result = _fetch_space_logs_level(repo_id, "run")
+        response2 = chat.send_message("", function_response={fn.name: result})
+        reply = response2.candidates[0].content.parts[0].text
+    else:
+        reply = part.text
+    history.append((message, reply))
+    return history
+
+# --- UI: Chatbot with Sidebar for Space Creation ---
+with gr.Blocks(title="HF Code Sandbox Chat") as demo:
+    # Top login button
+    with gr.Row():
+        login_btn = gr.LoginButton(variant="huggingface", label="Sign in with HF")
+        status = gr.Markdown("*Not signed in.*")
+        login_btn.click(
+            lambda profile: f"✅ Logged in as {profile.username}" if profile else "*Not signed in.*",
+            inputs=None,
+            outputs=status
+        )
+
+    with gr.Row():
+        # Sidebar
+        with gr.Column(scale=2):
+            gr.Markdown("### 🏗️ Create New Space Sandbox")
+            api_key = gr.Textbox(label="Gemini API Key", placeholder="sk-...", type="password")
+            repo_name = gr.Textbox(label="Space Name", placeholder="my-sandbox")
+            sdk_selector = gr.Radio(label="SDK", choices=["gradio", "streamlit"], value="gradio")
+            create_btn = gr.Button("Initialize Sandbox")
+            create_status = gr.JSON(label="Init Status")
+            create_btn.click(init_chat, inputs=[repo_name, api_key], outputs=create_status)
+            # hidden store for repo_id
+            repo_store = gr.Variable("")
+
+        # Chat area
+        with gr.Column(scale=8):
+            chatbot = gr.Chatbot()
+            user_input = gr.Textbox(show_label=False, placeholder="Ask the sandbox to write or debug code...")
+            user_input.submit(
+                chatbot_respond,
+                inputs=[user_input, chatbot, repo_store, api_key],
+                outputs=chatbot
+            )
 
 if __name__ == "__main__":
     demo.launch()