Update app.py

app.py

@@ -3,6 +3,7 @@ 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.genai.types import Tool, GenerateContentConfig, GoogleSearch
 
 # — HELPERS FOR HF SPACE LOGS —
 
@@ -22,7 +23,8 @@ 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","")
+                    ts  = ev.get("timestamp","")
+                    txt = ev.get("data","")
                     lines.append(f"[{ts}] {txt}")
                 except:
                     continue
@@ -36,7 +38,7 @@ def handle_user_message(
     sdk_choice: str,
     gemini_api_key, grounding_enabled
 ):
-    # prep Gemini client + chat
+    # Initialize Gemini client + prime the conversation
     genai_client = genai.Client(api_key=gemini_api_key)
     chat = [{
         "role":"system",
@@ -46,39 +48,51 @@ def handle_user_message(
             "If errors appear, fix them and return the full updated code."
         )
     }]
-    # include past chat
     for role, msg in history:
         chat.append({"role": role, "content": msg})
 
     filename = "app.py" if sdk_choice=="gradio" else "streamlit_app.py"
 
-    # iterate up to 5 times to resolve errors
+    build_logs = run_logs = ""
     for _ in range(5):
+        # --- build the tool list ---
+        tools = []
+        if grounding_enabled:
+            tools.append(Tool(google_search=GoogleSearch()))
+
+        # --- configure Gemini properly ---
+        config = GenerateContentConfig(
+            tools=tools,
+            response_modalities=["TEXT"],
+        )
+
+        # --- call Gemini ---
         response = genai_client.models.generate_content(
             model="gemini-2.5-flash-preview-04-17",
             contents=[c["content"] for c in chat],
-            config={
-                "tools": [],
-                "google_search": grounding_enabled
-            }
+            config=config
         )
         ai_code = response.text
         chat.append({"role":"assistant", "content": ai_code})
 
-        # write and deploy
+        # write out the code
         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, token=hf_token.token,
-            exist_ok=True, repo_type="space",
+            repo_id=repo_id,
+            token=hf_token.token,
+            exist_ok=True,
+            repo_type="space",
             space_sdk=sdk_choice
         )
         upload_file(
             path_or_fileobj=filename,
             path_in_repo=filename,
-            repo_id=repo_id, token=hf_token.token,
+            repo_id=repo_id,
+            token=hf_token.token,
             repo_type="space"
         )
 
@@ -86,10 +100,11 @@ def handle_user_message(
         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
 
-        # feed errors back
+        # else, feed logs back into Gemini to repair
         chat.append({
             "role":"user",
             "content":(
@@ -100,7 +115,7 @@ def handle_user_message(
         })
         time.sleep(2)
 
-    # prepare outputs
+    # 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
@@ -121,10 +136,10 @@ with gr.Blocks(title="HF Space Auto‑Builder (Gradio & Streamlit)") as demo:
             api_key      = gr.Textbox(label="Gemini API Key", type="password")
             grounding    = gr.Checkbox(label="Enable grounding", value=False)
 
-            demo.load(lambda p: f"Logged in as **{p.username}**" if p else "*Not logged in.*",
+            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)
             login_btn.click(
-                lambda p: f"Logged in as **{p.username}**" if p else "*Not logged in.*",
+                lambda *args: f"Logged in as **{args[0].username}**" if args and args[0] else "*Not logged in.*",
                 inputs=None, outputs=login_status
             )