Update app.py

app.py

@@ -38,7 +38,10 @@ def handle_user_message(
     sdk_choice: str,
     gemini_api_key, grounding_enabled
 ):
-    # Initialize Gemini client + prime the conversation
+    # Ensure we actually have a profile & token
+    if hf_profile is None or hf_token is None:
+        return history + [("assistant", "⚠️ Please log in first.")], "", "", "<p>No Space yet.</p>"
+
     genai_client = genai.Client(api_key=gemini_api_key)
     chat = [{
         "role":"system",
@@ -52,34 +55,27 @@ def handle_user_message(
         chat.append({"role": role, "content": msg})
 
     filename = "app.py" if sdk_choice=="gradio" else "streamlit_app.py"
-
     build_logs = run_logs = ""
+
     for _ in range(5):
-        # --- build the tool list ---
+        # build tool list for grounding
         tools = []
         if grounding_enabled:
             tools.append(Tool(google_search=GoogleSearch()))
+        config = GenerateContentConfig(tools=tools, response_modalities=["TEXT"])
 
-        # --- configure Gemini properly ---
-        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=[c["content"] for c in chat],
             config=config
         )
-        ai_code = response.text
+        ai_code = resp.text
         chat.append({"role":"assistant", "content": ai_code})
 
-        # write out the code
+        # write & deploy
         with open(filename, "w") as f:
             f.write(ai_code)
 
-        # create/update the Space
         repo_id = f"{hf_profile.username}/{hf_profile.username}-auto-space"
         create_repo(
             repo_id=repo_id,
@@ -96,15 +92,12 @@ def handle_user_message(
             repo_type="space"
         )
 
-        # fetch logs
         build_logs = fetch_logs(repo_id, "build")
         run_logs   = fetch_logs(repo_id, "run")
 
-        # stop if no "ERROR" in either log
         if "ERROR" not in build_logs.upper() and "ERROR" not in run_logs.upper():
             break
 
-        # else, feed logs back into Gemini to repair
         chat.append({
             "role":"user",
             "content":(
@@ -115,7 +108,6 @@ def handle_user_message(
         })
         time.sleep(2)
 
-    # prepare the outputs
     new_history = [(h["role"], h["content"]) for h in chat if h["role"]!="system"]
     iframe = f'<iframe src="https://huggingface.co/spaces/{repo_id}" width="100%" height="500px"></iframe>'
     return new_history, build_logs, run_logs, iframe
@@ -126,36 +118,45 @@ with gr.Blocks(title="HF Space Auto‑Builder (Gradio & Streamlit)") as demo:
     with gr.Row():
         with gr.Column(scale=1):
             gr.Markdown("### Sidebar")
+            # hidden state holders for OAuthProfile and OAuthToken
+            profile_state = gr.State()
+            token_state   = gr.State()
+
             login_btn    = gr.LoginButton("huggingface", size="sm")
             login_status = gr.Markdown("*Not logged in.*")
-            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)
 
-            demo.load(lambda *args: f"Logged in as **{args[0].username}**" if args and args[0] else "*Not logged in.*",
-                      inputs=None, outputs=login_status)
+            # capture profile & token into state
+            login_btn.click(
+                lambda profile, oauth_token: (profile, oauth_token),
+                outputs=[profile_state, token_state]
+            )
+            # update status text
             login_btn.click(
-                lambda *args: f"Logged in as **{args[0].username}**" if args and args[0] else "*Not logged in.*",
-                inputs=None, outputs=login_status
+                lambda profile, oauth_token: f"Logged in as **{profile.username}**"
+                  if profile else "*Not logged in.*",
+                outputs=login_status
+            )
+
+            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)
 
         with gr.Column(scale=3):
-            chatbot    = gr.Chatbot()
-            user_input = gr.Textbox(placeholder="Your prompt here...", label="Prompt")
-            send_btn   = gr.Button("Send")
+            chatbot   = gr.Chatbot(type="messages")
+            user_in   = gr.Textbox(placeholder="Your prompt …", 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>")
+            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>")
 
             send_btn.click(
                 fn=handle_user_message,
                 inputs=[
-                    chatbot, login_status, login_btn,
+                    chatbot,
+                    profile_state, token_state,
                     sdk_choice, api_key, grounding
                 ],
                 outputs=[chatbot, build_box, run_box, preview]