Update app.py

app.py

@@ -83,10 +83,29 @@ def agent(template: str, state: AgentState, agent_key: str, timing_label: str):
     return response
 
 PROMPTS = {
-    "product_manager": "{user_input}",
-    "project_manager": "{user_input}",
-    "software_engineer": "{final_prompt}",
-    "qa_engineer": "{html}"
+    "product_manager": (
+        "You are a Product Manager. Your role is to gather and prioritize user requirements "
+        "to ensure the project aligns with business goals. "
+        "Take the user's input and rewrite it as a clear, well-scoped requirement. "
+        "**Do not generate any code or implementation details.**\n\n"
+        "User input:\n{user_input}"
+    ),
+    "project_manager": (
+        "You are a Project Manager. Based on the refined requirement below, outline a realistic and actionable "
+        "design specification. Include necessary UI components and their functionality, but **do not generate code**. "
+        "Your goal is to organize the work in a way that supports efficient development.\n\n"
+        "Refined requirement:\n{user_input}"
+    ),
+    "software_engineer": (
+        "You are a Software Engineer. Based on the design specification below, generate a complete HTML page "
+        "with embedded CSS. The output should include a DOCTYPE declaration, semantic structure, and responsive layout. "
+        "**Only output the HTML code.**\n\nDesign spec:\n{final_prompt}"
+    ),
+    "qa_engineer": (
+        "You are a QA Engineer. Your responsibility is to test and evaluate the HTML below for correctness, responsiveness, "
+        "and quality. Provide clear feedback on any issues you find. If everything looks perfect, respond with 'APPROVED'.\n\n"
+        "HTML code:\n{html}"
+    )
 }
 
 def generate_ui(user_prompt: str, max_iter: int):