Update app.py

app.py

@@ -1,180 +1,54 @@
-import os
-import json
 import uuid
-from typing import Any, Dict, Tuple
-
 import gradio as gr
-from huggingface_hub import create_repo, HfApi
-from google import genai
-from google.genai import types
-from google.genai.types import Tool, GoogleSearch
-
-# -----------------------------------------------------------------------------
-# Configuration
-# -----------------------------------------------------------------------------
-MODEL_ID = "gemini-2.5-flash-preview-04-17"
-WORKSPACE_DIR = "workspace"
-SYSTEM_INSTRUCTION = (
-    "You are a helpful coding assistant that scaffolds a complete Hugging Face Space app. "
-    "Based on the user's request, decide between Gradio or Streamlit (whichever fits best), "
-    "and respond with exactly one JSON object with keys:\n"
-    "  • \"framework\": either \"gradio\" or \"streamlit\"\n"
-    "  • \"files\": a map of relative file paths to file contents\n"
-    "  • \"message\": a human‑readable summary\n"
-    "Do not include extra text or markdown."
-)
-
-state_store: Dict[str, Dict[str, Any]] = {}
-
-def start_app(
-    gemini_key: str,
-    hf_token: str,
-    hf_username: str,
-    repo_name: str
-) -> Dict[str, Any]:
-    os.makedirs(WORKSPACE_DIR, exist_ok=True)
-    client = genai.Client(api_key=gemini_key)
-    config = types.GenerateContentConfig(system_instruction=SYSTEM_INSTRUCTION)
-    tools = [Tool(google_search=GoogleSearch())]
-    chat = client.chats.create(model=MODEL_ID, config=config, tools=tools)
-    local_path = os.path.join(WORKSPACE_DIR, repo_name)
-    os.makedirs(local_path, exist_ok=True)
-    return {
-        "chat": chat,
-        "hf_token": hf_token,
-        "hf_username": hf_username,
-        "repo_name": repo_name,
-        "created": False,
-        "repo_id": None,
-        "local_path": local_path,
-        "embed_url": None,
-        "logs": [f"Initialized workspace at {WORKSPACE_DIR}/{repo_name}."]
-    }
-
-def handle_message(user_msg: str, state: Dict[str, Any]) -> Tuple[str, Dict[str, Any]]:
-    chat = state["chat"]
-    logs = state.setdefault("logs", [])
-    logs.append(f"> **User**: {user_msg}")
-    resp = chat.send_message(user_msg)
-    logs.append("Received response from Gemini.")
-
-    try:
-        data = json.loads(resp.text)
-        framework = data["framework"]
-        files = data.get("files", {})
-        reply = data.get("message", "")
-    except Exception:
-        logs.append("⚠️ Failed to parse assistant JSON.\n" + resp.text)
-        return "⚠️ Parsing error. Check logs.", state
+from huggingface_hub import create_repo
 
-    if not state["created"]:
-        full_repo = f"{state['hf_username']}/{state['repo_name']}"
-        create_repo(
-            repo_id=full_repo,
-            token=state["hf_token"],
-            exist_ok=True,
-            repo_type="space",
-            space_sdk=framework,
-        )
-        state.update({
-            "created": True,
-            "repo_id": full_repo,
-            "embed_url": f"https://huggingface.co/spaces/{full_repo}",
-        })
-
-    if files:
-        logs.append(f"Writing {len(files)} file(s): {list(files)}")
-        for relpath, content in files.items():
-            dest = os.path.join(state["local_path"], relpath)
-            os.makedirs(os.path.dirname(dest), exist_ok=True)
-            with open(dest, "w", encoding="utf‑8") as fp:
-                fp.write(content)
-
-    HfApi(token=state["hf_token"]).upload_folder(
-        folder_path=state["local_path"],
-        repo_id=state["repo_id"],
-        repo_type="space",
-    )
-    logs.append("Snapshot upload complete.")
-    return reply, state
+# Store created spaces' info
+state_store = {}
 
-with gr.Blocks(title="Gemini → HF Space scaffolder") as demo:
-    # 1) OAuth button + status
-    login_btn = gr.LoginButton(variant="huggingface", size="lg")  # displays "Sign in with Hugging Face"
+with gr.Blocks(title="HF Space Creator") as demo:
+    # 1) OAuth button + login status
+    login_btn = gr.LoginButton(variant="huggingface", size="lg")
     status_md = gr.Markdown("*Not logged in.*")
 
-    # 2) show_profile called on load & login/logout
     def show_profile(profile: gr.OAuthProfile | None) -> str:
         return "*Not logged in.*" if profile is None else f"Logged in as **{profile.username}**"
 
-    # Trigger on page load and when login button is clicked
+    # Update status on page load and after click
     demo.load(fn=show_profile, inputs=None, outputs=[status_md])
     login_btn.click(fn=show_profile, inputs=None, outputs=[status_md])
 
-    with gr.Row():
-        with gr.Column(scale=1):
-            gemini_key   = gr.Textbox(label="Gemini API Key", type="password")
-            repo_name    = gr.Textbox(label="New App (repo) name")
-            session_id   = gr.Textbox(value="", visible=False)
-            start_btn    = gr.Button("Start a new app")
-        with gr.Column(scale=3):
-            chatbot        = gr.Chatbot(type="messages", value=[])
-            logs_display   = gr.Textbox(label="Operation Logs", interactive=False, lines=8)
-            preview_iframe = gr.HTML("<p>No deployed app yet.</p>")
-            user_msg       = gr.Textbox(label="Your message")
-            send_btn       = gr.Button("Send", interactive=False)
+    # 2) Repo creation UI
+    repo_name = gr.Textbox(label="New Space name", placeholder="your-repo-name")
+    create_btn = gr.Button("Create Space", interactive=False)
+    session_id = gr.Textbox(value="", visible=False)
+    logs = gr.Textbox(label="Logs", interactive=False, lines=5)
+    preview_iframe = gr.HTML("<p>No Space created yet.</p>")
 
-    # 3) on_start: triggered by "Start a new app"
-    def on_start(
-        evt,
-        gemini_key: str,
-        repo_name: str,
-        oauth_token: gr.OAuthToken | None
-    ):
-        if oauth_token is None:
-            return gr.Error("Please *Sign in with Hugging Face* first."), "", "", gr.update(interactive=False)
-        hf_username = oauth_token.user_info["username"]
-        new_id = str(uuid.uuid4())
-        state_store[new_id] = start_app(gemini_key, oauth_token.token, hf_username, repo_name)
-        logs = "\n".join(state_store[new_id]["logs"])
-        return new_id, logs, "<p>Awaiting first instruction…</p>", gr.update(interactive=True)
-
-    start_btn.click(
-        on_start,
-        inputs=[gemini_key, repo_name, login_btn],  # pass login_btn to provide OAuthToken
-        outputs=[session_id, logs_display, preview_iframe, send_btn],
-    )
-
-    # 4) Handle user messages
-    def on_send(
-        msg: str,
-        chat_history: list[dict],
-        sess_id: str
-    ) -> Tuple[list[dict], str, str, str]:
-        if not sess_id or sess_id not in state_store:
-            return [], sess_id, "", ""
-        reply, new_state = handle_message(msg, state_store[sess_id])
-        state_store[sess_id] = new_state
-
-        history = chat_history or []
-        history.append({"role": "user", "content": msg})
-        history.append({"role": "assistant", "content": reply})
-
-        logs = "\n".join(new_state["logs"])
-        embed = ""
-        if new_state.get("embed_url"):
-            embed = f'<iframe src="{new_state["embed_url"]}" width="100%" height="500px"></iframe>'
-        return history, sess_id, logs, embed
+    # Enable create button only after login
+    def enable_create(profile: gr.OAuthProfile | None):
+        return gr.update(interactive=(profile is not None))
+    login_btn.click(fn=enable_create, inputs=None, outputs=[create_btn])
 
-    send_btn.click(
-        on_send,
-        inputs=[user_msg, chatbot, session_id],
-        outputs=[chatbot, session_id, logs_display, preview_iframe],
-    )
-    user_msg.submit(
-        on_send,
-        inputs=[user_msg, chatbot, session_id],
-        outputs=[chatbot, session_id, logs_display, preview_iframe],
+    def create_space(name: str, oauth_token: gr.OAuthToken | None):
+        if oauth_token is None:
+            return "", "Please sign in first.", "<p>No Space created yet.</p>"
+        username = oauth_token.user_info["username"]
+        repo_id = f"{username}/{name}"
+        create_repo(
+            repo_id=repo_id,
+            token=oauth_token.token,
+            exist_ok=True,
+            repo_type="space"
+        )
+        url = f"https://huggingface.co/spaces/{repo_id}"
+        logs = f"Created or found Space: {url}"
+        iframe = f'<iframe src="{url}" width="100%" height="500px"></iframe>'
+        return repo_id, logs, iframe
+
+    create_btn.click(
+        fn=create_space,
+        inputs=[repo_name, login_btn],
+        outputs=[session_id, logs, preview_iframe]
     )
 
 if __name__ == "__main__":