from crewai import Task

def get_research_task(agent):
    return Task(
        description="""Research the given topic extensively using PMC and Google Scholar.
                      1. Search for at least 20 relevant scientific articles
                      2. For each article collect:
                         - Title
                         - Authors
                         - Abstract
                         - Key findings
                         - Publication year
                      3. Ensure articles are recent and from reputable sources
                      4. Organize the information in a structured format
                      5. Provide a brief summary of the collective findings""",
        agent=agent
    )

def get_writing_task(agent):
    return Task(
        description="""Write a comprehensive scientific paper based on the research provided.
                      The paper must follow this structure:
                      1. Introduction:
                         - Background of the topic
                         - Research gap
                         - Objectives
                      2. Materials and Methods:
                         - Research methodology
                         - Data collection process
                         - Analysis methods
                      3. Results (at least one page):
                         - Key findings from the literature
                         - Data synthesis
                         - Statistical analysis if applicable
                      4. Discussion (at least one page):
                         - Interpretation of results
                         - Comparison with existing literature
                         - Implications
                         - Limitations
                      5. Conclusion:
                         - Summary of findings
                         - Future research directions
                      6. References:
                         - All citations in APA format
                         - Minimum 20 references
                      
                      Important:
                      - Use proper academic writing style
                      - Include in-text citations
                      - Maintain logical flow between sections
                      - Support claims with evidence from the literature""",
        agent=agent
    )