Update app.py

app.py

@@ -82,7 +82,7 @@ def check_iframe(url: str, timeout: int = 5) -> bool:
     except:
         return False
 
-# — CORE LOOP —
+# — CORE LOOP WITH TWO AGENTS —
 
 def handle_user_message(
     history,
@@ -97,21 +97,33 @@ def handle_user_message(
     if not profile or not oauth_token:
         return history + [{"role":"assistant","content":"⚠️ Please log in first."}], "", "", "<p>No Space yet.</p>"
 
-    client = genai.Client(api_key=gemini_api_key)
-    code_fn   = "app.py" if sdk_choice=="gradio" else "streamlit_app.py"
-    repo_id   = f"{profile.username}/{profile.username}-auto-space"
+    client     = genai.Client(api_key=gemini_api_key)
+    code_fn    = "app.py" if sdk_choice=="gradio" else "streamlit_app.py"
+    repo_id    = f"{profile.username}/{profile.username}-auto-space"
     iframe_url = f"https://huggingface.co/spaces/{repo_id}"
 
-    system_msg = {
+    # — SYSTEM PROMPTS —  
+    system_code = {
         "role":"system",
         "content":(
-            f"You are an AI assistant that scaffolds a complete HuggingFace Space using the "
-            f"{sdk_choice} SDK. **Respond with exactly one markdown code block** containing the "
-            f"entire contents of `{code_fn}`—no extra commentary or version hints. After that, "
-            "wait for logs; if errors appear, fix the code and re-output the full code block."
+            "You are **Code‑Gen Agent**, a proactive AI developer. Your sole responsibility is to author "
+            f"and correct the entire `{code_fn}` file in a single markdown code block—no extra commentary. "
+            "You have permission to edit files, push updates to the HF Space, and optimize code. "
+            "After each push, await build & run logs before making further changes."
         )
     }
-    chat = [system_msg] + history
+    system_debug = {
+        "role":"system",
+        "content":(
+            "You are **Debug Agent**, a meticulous code reviewer. You can read all files, logs, and the app "
+            "preview, but you **cannot** modify or push code. Your task is to analyze the latest code + logs "
+            "and return concise, actionable feedback or “All clear.”"
+        )
+    }
+
+    # initialize each agent’s conversation
+    code_chat  = [system_code] + history[:]
+    debug_chat = [system_debug] + history[:]
 
     build_logs = run_logs = ""
     backoff = 1
@@ -125,28 +137,28 @@ def handle_user_message(
             max_output_tokens=max_output_tokens,
         )
 
-        # generate code
-        resp = client.models.generate_content(
+        # --- 1) Code‑Gen generates or updates code ---
+        resp_code = client.models.generate_content(
             model="gemini-2.5-flash-preview-04-17",
-            contents=[m["content"] for m in chat],
+            contents=[m["content"] for m in code_chat],
             config=cfg
         )
-        code = extract_code(resp.text)
-        chat.append({"role":"assistant","content":code})
+        code = extract_code(resp_code.text)
+        code_chat.append({"role":"assistant","content":code})
+        debug_chat.append({"role":"assistant","content":code})
 
-        # local syntax check
+        # quick syntax check
         try:
             compile(code, code_fn, "exec")
         except SyntaxError as e:
-            chat.append({
+            code_chat.append({
                 "role":"user",
-                "content": f"SyntaxError caught: {e}. Please output the corrected `{code_fn}` code block only."
+                "content": f"SyntaxError caught: {e}. Please correct `{code_fn}` only."
             })
-            time.sleep(backoff)
-            backoff = min(backoff*2, 30)
+            time.sleep(backoff); backoff = min(backoff*2, 30)
             continue
 
-        # write files
+        # write & push to HF Space
         sdk_version = get_sdk_version(sdk_choice)
         files = {
             code_fn: code,
@@ -158,8 +170,6 @@ sdk_version: {sdk_version}
 app_file: {code_fn}
 pinned: false
 ---
-
-See https://huggingface.co/docs/hub/spaces-config-reference
 """,
             "requirements.txt": "pandas\n" + ("streamlit\n" if sdk_choice=="streamlit" else "gradio\n")
         }
@@ -167,46 +177,59 @@ See https://huggingface.co/docs/hub/spaces-config-reference
             with open(fn, "w") as f:
                 f.write(content)
 
-        # push to HF
         create_repo(repo_id=repo_id, token=oauth_token.token,
                     exist_ok=True, repo_type="space", space_sdk=sdk_choice)
         for fn in files:
             upload_file(
-                path_or_fileobj=fn,
-                path_in_repo=fn,
-                repo_id=repo_id,
-                token=oauth_token.token,
+                path_or_fileobj=fn, path_in_repo=fn,
+                repo_id=repo_id, token=oauth_token.token,
                 repo_type="space"
             )
 
         # fetch logs
         build_logs = fetch_logs(repo_id, "build")
         run_logs   = fetch_logs(repo_id, "run")
-        errors = classify_errors(build_logs + "\n" + run_logs)
+        err_types  = classify_errors(build_logs + "\n" + run_logs)
+
+        # --- 2) Debug‑Agent reviews code & logs ---
+        debug_input = (
+            f"🏷 **Attempt {attempt}**\n"
+            f"Error types: {err_types}\n\n"
+            f"**Build logs:**\n{build_logs}\n\n"
+            f"**Run logs:**\n{run_logs}\n\n"
+            "If there are no errors, reply “All clear.” Otherwise, list your recommended changes."
+        )
+        debug_chat.append({"role":"user","content":debug_input})
+        resp_debug = client.models.generate_content(
+            model="gemini-2.5-flash-preview-04-17",
+            contents=[m["content"] for m in debug_chat],
+            config=cfg
+        )
+        feedback = resp_debug.text.strip()
+        debug_chat.append({"role":"assistant","content":feedback})
 
+        # check for success
         if "ERROR" not in build_logs.upper() and \
            "ERROR" not in run_logs.upper() and \
            check_iframe(iframe_url):
             break
 
-        chat.append({
+        # feed debug feedback back to Code‑Gen
+        code_chat.append({
             "role":"user",
-            "content":(
-                f"Attempt {attempt} had {errors} errors.\n"
-                f"Build logs:\n{build_logs}\n\n"
-                f"Run logs:\n{run_logs}\n\n"
-                f"Please output the full corrected `{code_fn}` code block only, "
-                f"and ensure the iframe at {iframe_url} returns HTTP 200."
-            )
+            "content": f"🔧 Debug feedback:\n{feedback}\nPlease output the full corrected `{code_fn}` code block only."
         })
-        time.sleep(backoff)
-        backoff = min(backoff*2, 30)
+        time.sleep(backoff); backoff = min(backoff*2, 30)
 
-    messages = [{"role":m["role"],"content":m["content"]} for m in chat if m["role"]!="system"]
+    # prepare UI outputs
+    messages = [
+        {"role": m["role"], "content": m["content"]}
+        for m in code_chat if m["role"] != "system"
+    ]
     iframe_html = (
         f'<iframe src="{iframe_url}" width="100%" height="500px"></iframe>'
-        + ("" if check_iframe(iframe_url) else
-           "<p style='color:red;'>⚠️ iframe not responding.</p>")
+        + ("" if check_iframe(iframe_url)
+           else "<p style='color:red;'>⚠️ iframe not responding.</p>")
     )
     return messages, build_logs, run_logs, iframe_html
 
@@ -249,9 +272,9 @@ with gr.Blocks(title="HF Space Auto‑Builder") as demo:
 
     refresh_btn = gr.Button("Refresh Logs")
     refresh_btn.click(
-        fn=lambda profile,token: (
-            fetch_logs(f"{profile.username}/{profile.username}-auto-space","build"),
-            fetch_logs(f"{profile.username}/{profile.username}-auto-space","run")
+        fn=lambda profile, token: (
+            fetch_logs(f"{profile.username}/{profile.username}-auto-space", "build"),
+            fetch_logs(f"{profile.username}/{profile.username}-auto-space", "run")
         ),
         outputs=[build_box, run_box]
     )