Update utils/run_pipeline_and_save.py

utils/run_pipeline_and_save.py

@@ -1,3 +1,5 @@
+import os
+import json
 from agents import (
     product_manager_agent,
     project_manager_agent,
@@ -5,33 +7,39 @@ from agents import (
     software_engineer_agent,
     quality_assurance_agent,
 )
-import os
-import json
 
-os.makedirs("output", exist_ok=True)
+OUTPUT_DIR = "output"
+os.makedirs(OUTPUT_DIR, exist_ok=True)
 
-def run_pipeline_and_save(prompt):
-    log = []
+def run_pipeline_and_save(prompt: str):
+    chat_log = []
 
-    pm_out = product_manager_agent.run(prompt)
-    log.append({"Product Manager": pm_out})
+    # Step 1: Product Manager
+    pm_output = product_manager_agent.run(prompt)
+    chat_log.append({"Product Manager": pm_output})
 
-    proj_out = project_manager_agent.run(pm_out)
-    log.append({"Project Manager": proj_out})
+    # Step 2: Project Manager
+    proj_output = project_manager_agent.run(pm_output)
+    chat_log.append({"Project Manager": proj_output})
 
-    arch_out = software_architect_agent.run(proj_out)
-    log.append({"Software Architect": arch_out})
+    # Step 3: Software Architect
+    arch_output = software_architect_agent.run(proj_output)
+    chat_log.append({"Software Architect": arch_output})
 
-    eng_out = software_engineer_agent.run(arch_out)
-    log.append({"Software Engineer": eng_out})
+    # Step 4: Software Engineer
+    eng_output = software_engineer_agent.run(arch_output)
+    chat_log.append({"Software Engineer": eng_output})
 
-    qa_out = quality_assurance_agent.run(eng_out)
-    log.append({"QA": qa_out})
+    # Step 5: Quality Assurance
+    qa_output = quality_assurance_agent.run(eng_output)
+    chat_log.append({"Quality Assurance": qa_output})
 
-    with open("output/agent_log.json", "w") as f:
-        json.dump(log, f, indent=2)
+    # Save final UI output files
+    with open(os.path.join(OUTPUT_DIR, "index.html"), "w") as f:
+        f.write(eng_output)
 
-    with open("output/final_output.html", "w") as f:
-        f.write(eng_out)
+    # Save chat log
+    with open(os.path.join(OUTPUT_DIR, "agent_log.json"), "w") as f:
+        json.dump(chat_log, f, indent=2)
 
-    return log, eng_out
+    return chat_log, eng_output