from crewai import Task

from agents import get_data_analyst_agent, get_trading_strategy_agent, get_execution_agent, get_risk_management_agent

class Tasks:
    def get_data_analysis_task():
        # Task for Data Analyst Agent: Analyze Market Data
        data_analysis_task = Task(
            description=(
                "Continuously monitor and analyze market data for "
                "the selected stock ({stock_selection}). "
                "Use statistical modeling and machine learning to "
                "identify trends and predict market movements."
            ),
            expected_output=(
                "Insights and alerts about significant market "
                "opportunities or threats for {stock_selection}."
            ),
            agent=get_data_analyst_agent(),
        )
    
    def get_strategy_development_task():
        # Task for Trading Strategy Agent: Develop Trading Strategies
        strategy_development_task = Task(
            description=(
                "Develop and refine trading strategies based on "
                "the insights from the Data Analyst and "
                "user-defined risk tolerance ({risk_tolerance}). "
                "Consider trading preferences ({trading_strategy_preference})."
            ),
            expected_output=(
                "A set of potential trading strategies for {stock_selection} "
                "that align with the user's risk tolerance."
            ),
            agent=get_trading_strategy_agent(),
        )
    
    def get_execution_planning_task():
        # Task for Trade Advisor Agent: Plan Trade Execution
        execution_planning_task = Task(
            description=(
                "Analyze approved trading strategies to determine the "
                "best execution methods for {stock_selection}, "
                "considering current market conditions and optimal pricing."
            ),
            expected_output=(
                "Detailed execution plans suggesting how and when to "
                "execute trades for {stock_selection}."
            ),
            agent=get_execution_agent(),
        )
    
    def get_risk_assessment_task():
        # Task for Risk Advisor Agent: Assess Trading Risks
        return = Task(
            description=(
                "Evaluate the risks associated with the proposed trading "
                "strategies and execution plans for {stock_selection}. "
                "Provide a detailed analysis of potential risks "
                "and suggest mitigation strategies."
            ),
            expected_output=(
                "A comprehensive risk analysis report detailing potential "
                "risks and mitigation recommendations for {stock_selection}."
            ),
            agent=get_risk_management_agent(),
        )