Update app.py

app.py

@@ -66,9 +66,8 @@ def handle_user_message(
     chat   = [{
         "role":"system",
         "content":(
-            f"You are an AI assistant that writes a HuggingFace Space using the "
-            f"{sdk_choice} SDK. After producing code, wait for logs. "
-            "If errors appear, fix them and return the full updated code."
+            f"You are an AI assistant writing a HuggingFace Space using the "
+            f"{sdk_choice} SDK. After producing code, wait for logs; if errors appear, fix them."
         )
     }] + history
 
@@ -84,16 +83,19 @@ def handle_user_message(
             tools.append(Tool(google_search=GoogleSearch()))
         cfg = GenerateContentConfig(tools=tools, response_modalities=["TEXT"])
 
-        resp    = client.models.generate_content(
+        resp = client.models.generate_content(
             model="gemini-2.5-flash-preview-04-17",
             contents=[m["content"] for m in chat],
             config=cfg
         )
-        code    = resp.text
+        code = resp.text
         chat.append({"role":"assistant","content":code})
 
-        # write files
-        with open(code_fn,   "w") as f: f.write(code)
+        # write code file
+        with open(code_fn, "w") as f:
+            f.write(code)
+
+        # write README.md
         with open(readme_fn, "w") as f:
             f.write(f"""---
 title: Wuhp Auto Space
@@ -108,10 +110,14 @@ pinned: false
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 """)
-        reqs = "pandas\n" + ("streamlit\n" if sdk_choice=="streamlit" else "gradio\n")
-        with open(reqs_fn, "w") as f: f.write(reqs)
 
-        # deploy
+        # write requirements.txt
+        base_reqs = "pandas\n"
+        extra     = "streamlit\n" if sdk_choice=="streamlit" else "gradio\n"
+        with open(reqs_fn, "w") as f:
+            f.write(base_reqs + extra)
+
+        # deploy all three files
         create_repo(
             repo_id=repo_id,
             token=oauth_token.token,
@@ -150,19 +156,20 @@ Check out the configuration reference at https://huggingface.co/docs/hub/spaces-
 
 # — BUILD THE UI —
 
-with gr.Blocks(title="HF Space Auto‑Builder (Gradio & Streamlit)") as demo:
+with gr.Blocks(title="HF Space Auto‑Builder") as demo:
     gr.Markdown("## Sign in + Auto‑Build Spaces\n\n"
                 "1. Sign in\n2. Enter your prompt\n3. Watch code, README, requirements, logs, and preview\n\n---")
 
     # LOGIN & MODEL LISTING
-    login_btn  = gr.LoginButton(variant="huggingface", size="lg")
-    status_md  = gr.Markdown("*Not logged in.*")
-    models_md  = gr.Markdown()
+    login_btn = gr.LoginButton(variant="huggingface", size="lg")
+    status_md = gr.Markdown("*Not logged in.*")
+    models_md = gr.Markdown()
 
-    demo.load(show_profile,        inputs=[login_btn], outputs=status_md)
-    demo.load(list_private_models, inputs=[login_btn], outputs=models_md)
-    login_btn.click(show_profile,        inputs=[login_btn], outputs=status_md)
-    login_btn.click(list_private_models, inputs=[login_btn], outputs=models_md)
+    # Implicitly injects profile (and token) when calling these
+    demo.load(show_profile,        inputs=None, outputs=status_md)
+    demo.load(list_private_models, inputs=None, outputs=models_md)
+    login_btn.click(show_profile,        inputs=None, outputs=status_md)
+    login_btn.click(list_private_models, inputs=None, outputs=models_md)
 
     # CONTROLS
     sdk_choice = gr.Radio(["gradio","streamlit"], value="gradio", label="SDK template")
@@ -178,9 +185,10 @@ with gr.Blocks(title="HF Space Auto‑Builder (Gradio & Streamlit)") as demo:
     run_box   = gr.Textbox(label="Run logs",   lines=5, interactive=False)
     preview   = gr.HTML("<p>No Space yet.</p>")
 
+    # Only user inputs here—profile/token are auto‑injected
     send_btn.click(
         fn=handle_user_message,
-        inputs=[chatbot, sdk_choice, api_key, grounding, login_btn],
+        inputs=[chatbot, sdk_choice, api_key, grounding],
         outputs=[chatbot, build_box, run_box, preview]
     )