Added comprehensive prompt generation with role-specific templates

app.py

@@ -758,11 +758,182 @@ class EnhancedAIFlow(AIFlow):
         self.flow_graph = nx.DiGraph()
         self.contexts: Dict[AgentRole, AgentContext] = {}
         self.global_context = {}
+        self.requirements = ""
+    
+    def initialize_flow(self):
+        """Initialize the AI Flow with agent relationships and dependencies."""
+        # Create nodes for each agent role
+        for role in AgentRole:
+            self.flow_graph.add_node(role)
+            self.contexts[role] = AgentContext(
+                role=role,
+                state=FlowState.PENDING,
+                artifacts={},
+                dependencies=[],
+                feedback=[]
+            )
+        
+        # Define dependencies
+        dependencies = {
+            AgentRole.UI_DESIGNER: [AgentRole.ARCHITECT],
+            AgentRole.BACKEND_DEVELOPER: [AgentRole.ARCHITECT],
+            AgentRole.DATABASE_ENGINEER: [AgentRole.ARCHITECT, AgentRole.BACKEND_DEVELOPER],
+            AgentRole.SECURITY_EXPERT: [AgentRole.ARCHITECT, AgentRole.BACKEND_DEVELOPER],
+            AgentRole.QA_ENGINEER: [AgentRole.UI_DESIGNER, AgentRole.BACKEND_DEVELOPER],
+            AgentRole.DEVOPS_ENGINEER: [AgentRole.BACKEND_DEVELOPER, AgentRole.DATABASE_ENGINEER],
+            AgentRole.DOCUMENTATION_WRITER: [AgentRole.ARCHITECT, AgentRole.UI_DESIGNER, AgentRole.BACKEND_DEVELOPER]
+        }
+        
+        # Add edges based on dependencies
+        for role, deps in dependencies.items():
+            for dep in deps:
+                self.flow_graph.add_edge(dep, role)
+                self.contexts[role].dependencies.extend(deps)
+    
+    def _generate_prompt(self, role: AgentRole) -> str:
+        """Generate a prompt for an agent based on context and dependencies."""
+        try:
+            context = self.contexts[role]
+            dependencies_output = []
+            
+            # Gather outputs from dependencies
+            for dep_role in context.dependencies:
+                dep_context = self.contexts[dep_role]
+                if dep_context.state == FlowState.COMPLETED and "output" in dep_context.artifacts:
+                    dependencies_output.append(f"## {dep_role.value} Output:\n{dep_context.artifacts['output']}")
+            
+            # Build role-specific prompts
+            role_prompts = {
+                AgentRole.ARCHITECT: """You are a Software Architect. Design the high-level architecture for:
+Requirements:
+{requirements}
+
+Focus on:
+1. Overall system design
+2. Component interactions
+3. Technology stack
+4. Data flow
+5. Scalability considerations""",
+                
+                AgentRole.UI_DESIGNER: """You are a UI/UX Designer. Design the user interface for:
+Requirements:
+{requirements}
+
+Previous Designs:
+{dependencies}
+
+Focus on:
+1. User experience
+2. Component layout
+3. Responsive design
+4. Visual hierarchy
+5. Accessibility""",
+                
+                AgentRole.BACKEND_DEVELOPER: """You are a Backend Developer. Implement the server-side logic for:
+Requirements:
+{requirements}
+
+Architecture:
+{dependencies}
+
+Focus on:
+1. API design
+2. Business logic
+3. Data validation
+4. Error handling
+5. Performance""",
+                
+                AgentRole.DATABASE_ENGINEER: """You are a Database Engineer. Design the data layer for:
+Requirements:
+{requirements}
+
+System Context:
+{dependencies}
+
+Focus on:
+1. Schema design
+2. Data relationships
+3. Indexing strategy
+4. Query optimization
+5. Data integrity""",
+                
+                AgentRole.SECURITY_EXPERT: """You are a Security Expert. Review and enhance security for:
+Requirements:
+{requirements}
+
+System Context:
+{dependencies}
+
+Focus on:
+1. Authentication
+2. Authorization
+3. Data protection
+4. Security best practices
+5. Compliance requirements""",
+                
+                AgentRole.QA_ENGINEER: """You are a QA Engineer. Design the testing strategy for:
+Requirements:
+{requirements}
+
+Implementation Details:
+{dependencies}
+
+Focus on:
+1. Test coverage
+2. Test automation
+3. Edge cases
+4. Performance testing
+5. Security testing""",
+                
+                AgentRole.DEVOPS_ENGINEER: """You are a DevOps Engineer. Set up the deployment pipeline for:
+Requirements:
+{requirements}
+
+System Context:
+{dependencies}
+
+Focus on:
+1. CI/CD pipeline
+2. Infrastructure as code
+3. Monitoring
+4. Scaling
+5. Disaster recovery""",
+                
+                AgentRole.DOCUMENTATION_WRITER: """You are a Technical Writer. Create comprehensive documentation for:
+Requirements:
+{requirements}
+
+System Details:
+{dependencies}
+
+Focus on:
+1. Setup instructions
+2. API documentation
+3. User guides
+4. Architecture overview
+5. Troubleshooting guides"""
+            }
+            
+            # Get the base prompt for the role
+            base_prompt = role_prompts.get(role, "")
+            
+            # Format the prompt with requirements and dependencies
+            formatted_prompt = base_prompt.format(
+                requirements=self.requirements,
+                dependencies="\n\n".join(dependencies_output) if dependencies_output else "No previous context available."
+            )
+            
+            return formatted_prompt
+            
+        except Exception as e:
+            logger.error(f"Failed to generate prompt for {role}: {str(e)}")
+            raise
     
     async def execute_flow(self, requirements: str) -> str:
         """Execute the AI Flow and build the project."""
         try:
             # Initialize flow with requirements
+            self.requirements = requirements
             self.initialize_flow()
             
             # Extract app name