Update backend.py

backend.py

@@ -140,7 +140,7 @@ def generate_with_model(role: str, prompt: str) -> str:
         print(f"Error during model generation for role {role}: {e}")
         return f'{{"error": "Failed to generate response: {str(e)}"}}'
 
-# ------------------------------ THE AGENT CHAIN EXECUTOR (REWRITTEN FOR RELIABILITY) ------------------------------
+# ------------------------------ THE AGENT CHAIN EXECUTOR ------------------------------
 def run_agent_chain(project_id, user_id, initial_prompt):
     project_dir = get_project_dir(user_id, project_id)
     log_entries = []
@@ -152,8 +152,7 @@ def run_agent_chain(project_id, user_id, initial_prompt):
         update_project_status(project_id, "running", logs="".join(log_entries))
 
     try:
-        log_step("SYSTEM", f"Initializing project directory...")
-        update_project_status(project_id, "running", logs="Agent team is assembling...")
+        log_step("SYSTEM", "Initializing project...")
 
         # 1. PLANNER
         log_step("PLANNER", "Analyzing user request and creating a project plan...")
@@ -161,12 +160,11 @@ def run_agent_chain(project_id, user_id, initial_prompt):
         plan_data = _extract_json(plan_response)
         if not plan_data or "files" not in plan_data: raise ValueError("Planner failed to create a valid JSON plan with a 'files' key.")
         
-        ## ROBUSTNESS FIX: Handle cases where the LLM returns [{"file": "path"}] instead of ["path"]
         if plan_data["files"] and isinstance(plan_data["files"][0], dict):
-            log_step("SYSTEM", "Planner returned a list of objects. Normalizing to a list of strings.")
+            log_step("SYSTEM", "Correcting malformed file list from Planner...")
             plan_data["files"] = [item["file"] for item in plan_data["files"] if "file" in item]
 
-        log_step("PLANNER", "Plan created successfully.", json.dumps(plan_data, indent=2))
+        log_step("PLANNER", "Plan created.", json.dumps(plan_data, indent=2))
         
         # 2. ARCHITECT
         log_step("ARCHITECT", "Creating initial file skeletons...")
@@ -193,11 +191,10 @@ def run_agent_chain(project_id, user_id, initial_prompt):
             if not code_content: continue
             review_prompt = f"Review this code from `{file_path}`:\n\n{code_content}"
             review_response = generate_with_model("reviewer", review_prompt)
-            review_data = _extract_json(review_response)
-            log_step("REVIEWER", f"Review of `{file_path}` complete.", json.dumps(review_data, indent=2))
+            log_step("REVIEWER", f"Review of `{file_path}` complete.", review_response)
 
         # 5. TESTER
-        log_step("TESTER", "Writing unit tests for all source code...")
+        log_step("TESTER", "Writing unit tests...")
         for file_path in source_files:
             code_content = read_file(project_dir, file_path)
             if not code_content: continue
@@ -221,7 +218,7 @@ def run_agent_chain(project_id, user_id, initial_prompt):
         log_step("SYSTEM", "Packaging project into a ZIP file...")
         zip_path = zip_project(project_dir, project_id)
         update_project_status(project_id, "completed", logs="".join(log_entries), zip_path=zip_path)
-        log_step("SYSTEM", "Project completed successfully!", f"Download available.")
+        log_step("SYSTEM", "Project completed successfully!")
 
     except Exception as e:
         tb_str = traceback.format_exc()