Upload 4 files

agents.py

@@ -0,0 +1,50 @@
+import pandas as pd
+from typing import List
+from pydantic import BaseModel, Field
+from crewai import Agent, Task, Crew
+
+class TaskEstimate(BaseModel):
+    task_name: str = Field(..., description="Name of the task")
+    estimated_time_hours: float = Field(..., description = "Estimated time to complete the task in hours")
+    required_resources: List[str] = Field(..., description = "List of resources required to complete the task")
+
+class Milestone(BaseModel):
+    milestone_name: str = Field(..., description = "Name of the milestone")
+    tasks: List[str] = Field(..., description = "List of task IDs associated with this milestone")
+
+class ProjectPlan(BaseModel):
+    tasks: List[TaskEstimate] = Field(..., description = "List of tasks with their estimates")
+    milestones: List[Milestone] = Field(..., description = "List of project milestones")
+
+class ProjectPlanner:
+    
+    def __init__(self, agents_config, tasks_config, llm):
+        
+        # Creating Agents
+        project_planning_agent = Agent(config = agents_config['project_planning_agent'], llm = llm)
+        estimation_agent = Agent(config = agents_config['estimation_agent'], llm = llm)
+        resource_allocation_agent = Agent(config = agents_config['resource_allocation_agent'], llm = llm)
+
+        # Creating Tasks
+        task_breakdown = Task(config = tasks_config['task_breakdown'], agent = project_planning_agent)
+        time_resource_estimation = Task(config = tasks_config['time_resource_estimation'], agent = estimation_agent)
+        resource_allocation = Task(config = tasks_config['resource_allocation'], agent = resource_allocation_agent,
+                                   output_pydantic = ProjectPlan # This is the structured output we want
+                                   )
+
+        # Creating Crew
+        self.crew = Crew(
+            agents = [project_planning_agent, estimation_agent, resource_allocation_agent],
+            tasks = [task_breakdown, time_resource_estimation, resource_allocation],
+            verbose = True
+            )
+    
+    def getPlanning(self, project_details):
+        
+        results = self.crew.kickoff(inputs = project_details)
+        task_breakdown = results.pydantic.model_dump()['tasks']
+        df_tasks = pd.DataFrame(task_breakdown)
+        milestones = results.pydantic.model_dump()['milestones']
+        df_milestones = pd.DataFrame(milestones)
+        
+        return df_tasks, df_milestones

app.py

@@ -0,0 +1,190 @@
+import streamlit as st
+import os
+import yaml
+from crewai import LLM
+from agents import *
+
+# Streamlit Page Config
+st.set_page_config(
+    page_title = "Project Planner", 
+    page_icon = "🛠️", 
+    layout = "wide",
+    initial_sidebar_state = "expanded")
+
+# Logo
+st.logo(
+    "https://cdn.prod.website-files.com/66cf2bfc3ed15b02da0ca770/66d07240057721394308addd_Logo%20(1).svg",
+    link = "https://www.crewai.com/",
+    size = "large"
+)
+
+col1, col2, col3 = st.columns([1, 9, 1])
+with col2:
+    # Title and description
+    st.title("AI Project Planner, powered by :red[CrewAI]")
+    st.markdown("Create an detailed project plan with resource allocation and a timeline for tasks and milestones using AI agents.")
+    
+# Sidebar
+with st.sidebar:
+    st.markdown("### ⚙️ Model API Configuration")
+    st.write("")
+    
+    model_options = [
+        "gpt-4o-mini",
+        "gpt-4o",
+        "o1",
+        "o1-mini", 
+        "o1-preview"
+        "o3-mini"
+    ]
+    
+    selected_model = st.selectbox("🤖 Select which LLM to use", model_options, key = "selected_model")
+    
+    with st.expander("🔑 API Keys", expanded = True):
+        
+        st.info("API keys are stored temporarily in memory and cleared when you close the browser.")
+        
+        openai_api_key = st.text_input(
+            "OpenAI API Key",
+            type = "password",
+            placeholder = "Enter your OpenAI API key",
+            help = "Enter your OpenAI API key"
+        )
+        
+        if openai_api_key:
+            os.environ["OPENAI_API_KEY"] = openai_api_key
+            
+        # serper_api_key = st.text_input(
+        #     "Serper API Key",
+        #     type = "password",
+        #     placeholder = "Enter your Serper API key",
+        #     help = "Enter your Serper API key for web search capabilities"
+        #     )
+        # if serper_api_key:
+        #     os.environ["SERPER_API_KEY"] = serper_api_key
+    
+    st.write("")
+    
+    with st.expander("ℹ️ About", expanded=False):
+        st.markdown(
+            """This Project Planner uses advanced AI Agents to help you:
+                - Strategically think and breakdown projects into actionable tasks and setting precise timelines.
+                - Provide highly accurate time, resource, and effort estimations for each task.
+                - Optimize the allocation of tasks for the project by balancing team members' skills, availability, and current workload.
+                
+                Choose your preferred model and enter the required API keys to get started.""")
+        
+if not os.environ.get("OPENAI_API_KEY"):
+    st.warning("⚠️ Please enter your OpenAI API key in the sidebar to get started")
+    st.stop()
+
+# if not os.environ.get("SERPER_API_KEY"):
+#     st.warning("⚠️ Please enter your Serper API key in the sidebar to get started")
+#     st.stop()
+
+# Define file paths for YAML configurations
+files = {
+    'agents': 'config/agents.yaml',
+    'tasks': 'config/tasks.yaml'
+}
+
+# Load configurations from YAML files
+configs = {}
+for config_type, file_path in files.items():
+    with open(file_path, 'r') as file:
+        configs[config_type] = yaml.safe_load(file)
+
+# Assign loaded configurations to specific variables
+agents_config = configs['agents']
+tasks_config = configs['tasks']
+
+llm = llm = LLM(model = f"openai/{selected_model}")
+planner_crew = ProjectPlanner(agents_config, tasks_config, llm)    
+
+# Create two columns for the input section
+input_col1, input_col2, input_col3 = st.columns([3, 3, 5])
+
+with input_col1:
+    project_topic = st.text_area(
+        "Project Topic",
+        height = 80,
+        placeholder = "Enter the project topic (eg Website, Hiring, Building)"                
+    )
+
+with input_col2:
+    industry = st.text_area(
+        "Industry",
+        height = 80,
+        placeholder = "Enter the industry (eg Technology, Finance, Construction)"                
+    )
+
+with input_col3:
+    objective = st.text_area(
+        "Project Objective",
+        height = 80,
+        placeholder = "Enter the project objective (eg Build a website for a small business)"                
+    )
+    
+input_col4, input_col5 = st.columns([3,2])
+
+with input_col4:
+    project_requirements = st.text_area(
+        "Project Requirements (Brief bullet points)",
+        height = 190,
+        placeholder = """Enter bullet points of project requirements.
+        eg: 
+        - Create a responsive design that works well on desktop and mobile devices
+        - Implement a modern, visually appealing user interface with a clean look
+        - Develop a user-friendly navigation system with intuitive menu structure
+        - Include an "About Us" page highlighting the company's history and values
+        - Design a "Services" page showcasing the business's offerings with descriptions"""
+    )
+    
+with input_col5:
+    team_members = st.text_area(
+        "Team Members",
+        height = 190,
+        placeholder = """Enter the Team Member names are their roles.
+        eg:
+        - John Doe (Project Manager)
+        - Jane Doe (Software Engineer)
+        - Bob Smith (Designer)
+        - Alice Johnson (QA Engineer)
+        - Tom Brown (QA Engineer)
+        """
+    )
+    
+generate_button = st.button("🚀 Plan My Project", use_container_width = False, type = "primary")
+
+if generate_button:
+    with st.spinner("Generating content... This may take a moment."):
+        try:
+            project_details = {
+                'project_type': project_topic,
+                'project_objectives': objective,
+                'industry': industry,
+                'team_members': team_members,
+                'project_requirements': project_requirements
+                }
+            df_tasks, df_milestones = planner_crew.getPlanning(project_details)
+
+        except Exception as e:
+            st.error(f"An error occurred: {str(e)}")
+    
+    st.markdown("## Task Breakdown:")
+    st.dataframe(df_tasks)
+    
+    st.divider()
+    
+    st.markdown("## Milestone Details")
+    st.dataframe(df_milestones)    
+    
+
+
+
+# Add footer
+st.divider()
+footer_col1, footer_col2, footer_col3 = st.columns([1, 2, 1])
+with footer_col2:
+    st.caption("Made with ❤️ using [CrewAI](https://crewai.com) and [Streamlit](https://streamlit.io)")
+    st.caption("By [Sharan Shyamsundar](http://sharan1712.github.io/)")

config/agents.yaml

@@ -0,0 +1,49 @@
+project_planning_agent:
+  role: >
+    The Ultimate Project Planner
+  goal: >
+    To meticulously break down the {project_type} project into
+    actionable tasks, ensuring no detail is overlooked, and setting
+    precise timelines that align with the {project_objectives}.
+  backstory: >
+    As a veteran project manager, you’ve led numerous successful
+    projects, particularly in {industry}. Your keen eye for detail
+    and strategic thinking have always ensured that projects are
+    delivered on time and within scope. Now, you're tasked with
+    planning the next groundbreaking {project_type} project.
+  allow_delegation: false
+  verbose: true
+
+estimation_agent:
+  role: >
+    Expert Estimation Analyst
+  goal: >
+    Provide highly accurate time, resource, and effort estimations
+    for each task in the {project_type} project to ensure it is
+    delivered efficiently and on budget.
+  backstory: >
+    You are the go-to expert for project estimation in {industry}.
+    With a wealth of experience and access to vast historical data,
+    you can predict the resources required for any task with
+    remarkable accuracy.
+    Your precision ensures that the {project_type} project remains
+    feasible and avoids unnecessary delays or budget overruns.
+  allow_delegation: false
+  verbose: true
+
+resource_allocation_agent:
+  role: >
+    Resource Allocation Strategist
+  goal: >
+    Optimize the allocation of tasks for the {project_type} project
+    by balancing team members' skills, availability, and current
+    workload to maximize efficiency and project success.
+  backstory: >
+    With a deep understanding of team dynamics and resource
+    management in {industry}, you have a track record of ensuring
+    that the right person is always assigned to the right task.
+    Your strategic thinking ensures that the {project_type} project
+    team is utilized to its full potential without overburdening
+    any individual.
+  allow_delegation: false
+  verbose: true

config/tasks.yaml

@@ -0,0 +1,47 @@
+task_breakdown:
+  description: >
+    Carefully analyze the project_requirements for the {project_type}
+    project and break them down into individual tasks. Define each
+    task's scope in detail, set achievable timelines, and ensure that
+    all dependencies are accounted for:
+
+    {project_requirements}
+
+
+    Team members:
+
+    {team_members}
+  expected_output: >
+    A comprehensive list of tasks with detailed descriptions, timelines,
+    dependencies, and deliverables. Your final output MUST include a
+    Gantt chart or similar timeline visualization specific to the
+    {project_type} project.
+
+time_resource_estimation:
+  description: >
+    Thoroughly evaluate each task in the {project_type} project to
+    estimate the time, resources, and effort required.
+    Use historical data, task complexity, and available resources to
+    provide a realistic estimation for each task.
+  expected_output: >
+    A detailed estimation report outlining the time, resources, and
+    effort required for each task in the {project_type} project.
+    Your final report MUST include a summary of any risks or
+    uncertainties associated with the estimations.
+
+resource_allocation:
+  description: >
+    Strategically allocate tasks for the {project_type} project to
+    team members based on their skills, availability, and current
+    workload. Ensure that each task is assigned to the most suitable
+    team member and that the workload is evenly distributed.
+
+
+    Team members:
+
+    {team_members}
+  expected_output: >
+    A resource allocation chart showing which team members are
+    responsible for each task in the {project_type} project, along with
+    start and end dates. Your final output MUST also include a summary
+    explaining the rationale behind each allocation decision.