def generate_agent_role_prompt(agent):
    """Generates the agent role prompt.
    Args: agent (str): The type of the agent.
    Returns: str: The agent role prompt.
    """
    prompts = {
        "Finance Agent": (
            "You are a seasoned finance analyst AI assistant. Your"
            " primary goal is to compose comprehensive, astute,"
            " impartial, and methodically arranged financial reports"
            " based on provided data and trends."
        ),
        "Travel Agent": (
            "You are a world-travelled AI tour guide assistant. Your"
            " main purpose is to draft engaging, insightful,"
            " unbiased, and well-structured travel reports on given"
            " locations, including history, attractions, and cultural"
            " insights."
        ),
        "Academic Research Agent": (
            "You are an AI academic research assistant. Your primary"
            " responsibility is to create thorough, academically"
            " rigorous, unbiased, and systematically organized"
            " reports on a given research topic, following the"
            " standards of scholarly work."
        ),
        "Default Agent": (
            "You are an AI critical thinker research assistant. Your"
            " sole purpose is to write well written, critically"
            " acclaimed, objective and structured reports on given"
            " text."
        ),
    }

    return prompts.get(agent, "No such agent")


def generate_report_prompt(question, research_summary):
    """Generates the report prompt for the given question and research summary.
    Args: question (str): The question to generate the report prompt for
            research_summary (str): The research summary to generate the report prompt for
    Returns: str: The report prompt for the given question and research summary
    """

    return (
        f'"""{research_summary}""" Using the above information,'
        f' answer the following question or topic: "{question}" in a'
        " detailed report -- The report should focus on the answer"
        " to the question, should be well structured, informative,"
        " in depth, with facts and numbers if available, a minimum"
        " of 1,200 words and with markdown syntax and apa format."
        " Write all source urls at the end of the report in apa"
        " format"
    )


def generate_search_queries_prompt(question):
    """Generates the search queries prompt for the given question.
    Args: question (str): The question to generate the search queries prompt for
    Returns: str: The search queries prompt for the given question
    """

    return (
        "Write 4 google search queries to search online that form an"
        f' objective opinion from the following: "{question}"You must'
        " respond with a list of strings in the following format:"
        ' ["query 1", "query 2", "query 3", "query 4"]'
    )


def generate_resource_report_prompt(question, research_summary):
    """Generates the resource report prompt for the given question and research summary.

    Args:
        question (str): The question to generate the resource report prompt for.
        research_summary (str): The research summary to generate the resource report prompt for.

    Returns:
        str: The resource report prompt for the given question and research summary.
    """
    return (
        f'"""{research_summary}""" Based on the above information,'
        " generate a bibliography recommendation report for the"
        f' following question or topic: "{question}". The report'
        " should provide a detailed analysis of each recommended"
        " resource, explaining how each source can contribute to"
        " finding answers to the research question. Focus on the"
        " relevance, reliability, and significance of each source."
        " Ensure that the report is well-structured, informative,"
        " in-depth, and follows Markdown syntax. Include relevant"
        " facts, figures, and numbers whenever available. The report"
        " should have a minimum length of 1,200 words."
    )


def generate_outline_report_prompt(question, research_summary):
    """Generates the outline report prompt for the given question and research summary.
    Args: question (str): The question to generate the outline report prompt for
            research_summary (str): The research summary to generate the outline report prompt for
    Returns: str: The outline report prompt for the given question and research summary
    """

    return (
        f'"""{research_summary}""" Using the above information,'
        " generate an outline for a research report in Markdown"
        f' syntax for the following question or topic: "{question}".'
        " The outline should provide a well-structured framework for"
        " the research report, including the main sections,"
        " subsections, and key points to be covered. The research"
        " report should be detailed, informative, in-depth, and a"
        " minimum of 1,200 words. Use appropriate Markdown syntax to"
        " format the outline and ensure readability."
    )


def generate_concepts_prompt(question, research_summary):
    """Generates the concepts prompt for the given question.
    Args: question (str): The question to generate the concepts prompt for
            research_summary (str): The research summary to generate the concepts prompt for
    Returns: str: The concepts prompt for the given question
    """

    return (
        f'"""{research_summary}""" Using the above information,'
        " generate a list of 5 main concepts to learn for a research"
        f' report on the following question or topic: "{question}".'
        " The outline should provide a well-structured frameworkYou"
        " must respond with a list of strings in the following"
        ' format: ["concepts 1", "concepts 2", "concepts 3",'
        ' "concepts 4, concepts 5"]'
    )


def generate_lesson_prompt(concept):
    """
    Generates the lesson prompt for the given question.
    Args:
        concept (str): The concept to generate the lesson prompt for.
    Returns:
        str: The lesson prompt for the given concept.
    """

    prompt = (
        f"generate a comprehensive lesson about {concept} in Markdown"
        f" syntax. This should include the definitionof {concept},"
        " its historical background and development, its"
        " applications or uses in differentfields, and notable"
        f" events or facts related to {concept}."
    )

    return prompt


def get_report_by_type(report_type):
    report_type_mapping = {
        "research_report": generate_report_prompt,
        "resource_report": generate_resource_report_prompt,
        "outline_report": generate_outline_report_prompt,
    }
    return report_type_mapping[report_type]