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/)")