Update app.py

app.py

@@ -2,16 +2,9 @@ import gradio as gr
 import json, time
 from huggingface_hub import create_repo, upload_file, constants
 from huggingface_hub.utils import build_hf_headers, get_session, hf_raise_for_status
-from google import genai  # Gemini Python SDK
+from google import genai                 # Gemini Python SDK
 from google.genai.types import Tool, GenerateContentConfig, GoogleSearch
 
-# — USER INFO & MODEL LISTING (from your reference) —
-
-def show_profile(profile: gr.OAuthProfile | None) -> str:
-    if profile is None:
-        return "*Not logged in.*"
-    return f"✅ Logged in as **{profile.username}**"
-
 # — HELPERS FOR HF SPACE LOGS —
 
 def _get_space_jwt(repo_id: str):
@@ -30,9 +23,7 @@ def fetch_logs(repo_id: str, level: str):
             if raw.startswith(b"data: "):
                 try:
                     ev = json.loads(raw[len(b"data: "):].decode())
-                    ts  = ev.get("timestamp","")
-                    txt = ev.get("data","")
-                    lines.append(f"[{ts}] {txt}")
+                    lines.append(f"[{ev.get('timestamp','')}] {ev.get('data','')}")
                 except:
                     continue
     return "\n".join(lines)
@@ -51,10 +42,7 @@ def handle_user_message(
     if profile is None or oauth_token is None:
         return history + [{"role":"assistant","content":"⚠️ Please log in first."}], "", "", "<p>No Space yet.</p>"
 
-    # initialize Gemini
     genai_client = genai.Client(api_key=gemini_api_key)
-
-    # build the prompt history (including a system instruction)
     chat = [{
         "role":"system",
         "content":(
@@ -64,34 +52,58 @@ def handle_user_message(
         )
     }] + history
 
-    filename = "app.py" if sdk_choice=="gradio" else "streamlit_app.py"
-    build_logs = run_logs = ""
+    code_filename   = "app.py" if sdk_choice=="gradio" else "streamlit_app.py"
+    config_filename = "README.md"
+    reqs_filename   = "requirements.txt"
+    repo_id         = f"{profile.username}/{profile.username}-auto-space"
 
+    build_logs = run_logs = ""
     for _ in range(5):
-        # assemble tools
+        # build tools for grounding
         tools = []
         if grounding_enabled:
             tools.append(Tool(google_search=GoogleSearch()))
-
-        config = GenerateContentConfig(
-            tools=tools,
-            response_modalities=["TEXT"],
-        )
+        config = GenerateContentConfig(tools=tools, response_modalities=["TEXT"])
 
         # call Gemini
-        response = genai_client.models.generate_content(
+        resp = genai_client.models.generate_content(
             model="gemini-2.5-flash-preview-04-17",
             contents=[m["content"] for m in chat],
             config=config
         )
-        ai_code = response.text
+        ai_code = resp.text
         chat.append({"role":"assistant", "content": ai_code})
 
-        # write & deploy
-        with open(filename, "w") as f:
+        # write code file
+        with open(code_filename, "w") as f:
             f.write(ai_code)
 
-        repo_id = f"{profile.username}/{profile.username}-auto-space"
+        # write README.md with correct app_file
+        readme = f"""---
+title: Wuhp Auto Space
+emoji: 🐢
+colorFrom: red
+colorTo: pink
+sdk: {sdk_choice}
+sdk_version: 1.44.1
+app_file: {code_filename}
+pinned: false
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+"""
+        with open(config_filename, "w") as f:
+            f.write(readme)
+
+        # write requirements.txt
+        if sdk_choice == "streamlit":
+            reqs = "streamlit\npandas\n"
+        else:
+            reqs = "gradio\npandas\n"
+        with open(reqs_filename, "w") as f:
+            f.write(reqs)
+
+        # create/update Space and upload all three files
         create_repo(
             repo_id=repo_id,
             token=oauth_token.token,
@@ -99,23 +111,24 @@ def handle_user_message(
             repo_type="space",
             space_sdk=sdk_choice
         )
-        upload_file(
-            path_or_fileobj=filename,
-            path_in_repo=filename,
-            repo_id=repo_id,
-            token=oauth_token.token,
-            repo_type="space"
-        )
+        for fn in (code_filename, config_filename, reqs_filename):
+            upload_file(
+                path_or_fileobj=fn,
+                path_in_repo=fn,
+                repo_id=repo_id,
+                token=oauth_token.token,
+                repo_type="space"
+            )
 
-        # fetch build & run logs
+        # fetch logs
         build_logs = fetch_logs(repo_id, "build")
         run_logs   = fetch_logs(repo_id, "run")
 
-        # if no errors, stop looping
+        # stop if no errors
         if "ERROR" not in build_logs.upper() and "ERROR" not in run_logs.upper():
             break
 
-        # otherwise feed logs back to Gemini
+        # feed errors back
         chat.append({
             "role":"user",
             "content":(
@@ -133,39 +146,36 @@ def handle_user_message(
 
 # — BUILD THE UI —
 
+def show_profile(profile):
+    return f"✅ Logged in as **{profile.username}**" if profile else "*Not logged in.*"
+
 with gr.Blocks(title="HF Space Auto‑Builder (Gradio & Streamlit)") as demo:
-    gr.Markdown("## Sign in with Hugging Face + Auto‑Build Spaces\n\n"
-                "1. Sign in\n2. Enter your prompt\n3. Watch the code deploy & debug itself\n\n---")
+    gr.Markdown("## Sign in + Auto‑Build Spaces\n\n"
+                "1. Sign in\n2. Enter your prompt\n3. Watch the code, README, and requirements deploy & debug\n\n---")
 
-    # — LOGIN —
+    # LOGIN
     login_btn = gr.LoginButton(variant="huggingface", size="lg")
     status_md = gr.Markdown("*Not logged in.*")
-    # automatically show profile on load & login
     demo.load(show_profile,         inputs=None, outputs=status_md)
     login_btn.click(show_profile,   inputs=None, outputs=status_md)
 
-    # — SIDEBAR CONTROLS —
-    sdk_choice = gr.Radio(
-        choices=["gradio","streamlit"],
-        value="gradio",
-        label="SDK template"
-    )
-    api_key   = gr.Textbox(label="Gemini API Key", type="password")
-    grounding = gr.Checkbox(label="Enable grounding", value=False)
+    # CONTROLS
+    sdk_choice = gr.Radio(["gradio","streamlit"], value="gradio", label="SDK template")
+    api_key    = gr.Textbox(label="Gemini API Key", type="password")
+    grounding  = gr.Checkbox(label="Enable grounding", value=False)
 
-    # — CHAT INTERFACE & OUTPUTS —
+    # CHAT + OUTPUTS
     chatbot   = gr.Chatbot(type="messages")
-    user_in   = gr.Textbox(placeholder="e.g. Generate me a blurtest app…", label="Prompt")
+    user_in   = gr.Textbox(placeholder="e.g. Create me a CSV inspector…", label="Prompt")
     send_btn  = gr.Button("Send")
 
     build_box = gr.Textbox(label="Build logs", lines=5, interactive=False)
     run_box   = gr.Textbox(label="Run logs",   lines=5, interactive=False)
     preview   = gr.HTML("<p>No Space yet.</p>")
 
-    # wire up the Send button
     send_btn.click(
         fn=handle_user_message,
-        inputs=[chatbot, sdk_choice, api_key, grounding],
+        inputs=[chatbot, sdk_choice, api_key, grounding, gr.State(), gr.State()],
         outputs=[chatbot, build_box, run_box, preview]
     )