Rahul-8799 commited on
Commit
b44375d
·
verified ·
1 Parent(s): db83d42

Update src/utils/run_pipeline_and_save.py

Browse files
Files changed (1) hide show
  1. src/utils/run_pipeline_and_save.py +6 -6
src/utils/run_pipeline_and_save.py CHANGED
@@ -3,9 +3,9 @@ import json
3
  from agents import (
4
  #product_manager_agent,
5
  #project_manager_agent,
6
- #software_architect_agent,
7
  #software_engineer_agent,
8
- quality_assurance_agent
9
  )
10
 
11
  from huggingface_hub import login
@@ -35,16 +35,16 @@ def run_pipeline_and_save(prompt: str):
35
  #conversation_log.append({"Project Manager": proj_mgmt_response})
36
 
37
  # Step 3: Software Architect proposes architecture
38
- #architect_response = software_architect_agent.run(proj_mgmt_response)
39
- #conversation_log.append({"Software Architect": architect_response})
40
 
41
  # Step 4: Software Engineer generates HTML/CSS/JS
42
  #engineer_response = software_engineer_agent.run(architect_response)
43
  #conversation_log.append({"Software Engineer": engineer_response})
44
 
45
  # Step 5: QA Engineer verifies and refines
46
- qa_response = quality_assurance_agent.run(engineer_response)
47
- conversation_log.append({"QA Engineer": qa_response})
48
 
49
  # Save the final HTML output
50
  os.makedirs("output", exist_ok=True)
 
3
  from agents import (
4
  #product_manager_agent,
5
  #project_manager_agent,
6
+ software_architect_agent,
7
  #software_engineer_agent,
8
+ #quality_assurance_agent
9
  )
10
 
11
  from huggingface_hub import login
 
35
  #conversation_log.append({"Project Manager": proj_mgmt_response})
36
 
37
  # Step 3: Software Architect proposes architecture
38
+ architect_response = software_architect_agent.run(proj_mgmt_response)
39
+ conversation_log.append({"Software Architect": architect_response})
40
 
41
  # Step 4: Software Engineer generates HTML/CSS/JS
42
  #engineer_response = software_engineer_agent.run(architect_response)
43
  #conversation_log.append({"Software Engineer": engineer_response})
44
 
45
  # Step 5: QA Engineer verifies and refines
46
+ #qa_response = quality_assurance_agent.run(engineer_response)
47
+ #conversation_log.append({"QA Engineer": qa_response})
48
 
49
  # Save the final HTML output
50
  os.makedirs("output", exist_ok=True)