Rename utils/run_pipeline_and_save.py to utils/langgraph_pipeline.py

utils/langgraph_pipeline.py

@@ -0,0 +1,39 @@
+from langgraph.graph import StateGraph
+from agents import (
+    product_manager_agent,
+    project_manager_agent,
+    software_architect_agent,
+    software_engineer_agent,
+    quality_assurance_agent,
+)
+
+def run_pipeline_and_save(prompt: str):
+    state = {"prompt": prompt, "chat_log": []}
+
+    graph = StateGraph()
+
+    graph.add_node("PM", lambda s: product_manager_agent.run(s))
+    graph.add_node("ProjM", lambda s: project_manager_agent.run(s))
+    graph.add_node("Architect", lambda s: software_architect_agent.run(s))
+    graph.add_node("Engineer", lambda s: software_engineer_agent.run(s))
+    graph.add_node("QA", lambda s: quality_assurance_agent.run(s))
+
+    graph.set_entry_point("PM")
+    graph.add_edge("PM", "ProjM")
+    graph.add_edge("ProjM", "Architect")
+    graph.add_edge("Architect", "Engineer")
+    graph.add_edge("Engineer", "QA")
+
+    graph.set_finish_point("QA")
+
+    final_state = graph.execute(state)
+
+    # Save outputs
+    with open("output/index.html", "w") as f:
+        f.write(final_state.get("final_output", ""))
+
+    with open("output/agent_log.json", "w") as f:
+        import json
+        json.dump(final_state["chat_log"], f, indent=2)
+
+    return final_state["chat_log"], final_state.get("final_output", "")

utils/run_pipeline_and_save.py

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