briefing_service.py

from datetime import datetime, timedelta

from app.ai_service import generate_briefing
from app.notion_service import get_all_tasks_debug


def deduplicate_tasks(tasks):
    unique = {}

    for task in tasks:
        props = task["properties"]

        try:
            raw_title = props["Name"]["title"][0]["text"]["content"]
        except Exception:
            continue

        clean = raw_title.lower().strip()
        clean = clean.replace("semester ii", "")
        clean = clean.replace("(1mh201cc25)", "")
        clean = clean.replace("-", " ")
        clean = " ".join(clean.split())

        try:
            due_date = props["Due Date"]["date"]["start"]
        except Exception:
            continue

        if clean not in unique or due_date < unique[clean]["raw_due_date"]:
            unique[clean] = {
                "title": raw_title,
                "raw_due_date": due_date,
            }

    return list(unique.values())


def clean_title(title):
    if not title:
        return None

    title = title.strip()
    title = " ".join(title.split())
    words = []
    for word in title.split():
        if word.isupper() and len(word) <= 5:
            words.append(word)
        else:
            words.append(word.capitalize())
    title = " ".join(words)

    bad_keywords = ["event", "venue", "time", "schedule"]

    title_lower = title.lower()
    if any(word in title_lower for word in bad_keywords) and len(title.split()) > 6:
        return None

    return title


def format_date(date_str):
    try:
        date = datetime.fromisoformat(date_str)
        return date.strftime("%b %d")
    except Exception:
        return date_str


def categorize_tasks(raw_tasks):
    today = datetime.today().date()
    tasks = deduplicate_tasks(raw_tasks)

    today_tasks = []
    upcoming_tasks = []
    overdue_tasks = []

    for task in tasks:
        title = clean_title(task["title"])
        if not title:
            continue

        try:
            due_date = datetime.fromisoformat(task["raw_due_date"]).date()
        except Exception:
            continue

        task_data = {
            "title": title,
            "due_date": format_date(task["raw_due_date"]),
            "raw_due_date": task["raw_due_date"],
        }

        if due_date == today:
            today_tasks.append(task_data)
        elif due_date > today:
            upcoming_tasks.append(task_data)
        else:
            overdue_tasks.append(task_data)

    upcoming_tasks.sort(key=lambda x: x["raw_due_date"])
    overdue_tasks.sort(key=lambda x: x["raw_due_date"])
    today_tasks.sort(key=lambda x: x["raw_due_date"])
    upcoming_tasks = upcoming_tasks[:5]

    return today_tasks, upcoming_tasks, overdue_tasks


def generate_insights(today_tasks, upcoming_tasks, overdue_tasks):
    insights = []

    if len(overdue_tasks) > 0:
        insights.append(f"You have {len(overdue_tasks)} overdue task(s). Clear them first.")

    if len(today_tasks) > 2:
        insights.append("Busy day ahead. Prioritize wisely.")
    elif len(today_tasks) == 0:
        insights.append("No tasks due today. Great chance to get ahead.")

    date_count = {}
    for task in upcoming_tasks:
        date = task["due_date"]
        date_count[date] = date_count.get(date, 0) + 1

    for date, count in date_count.items():
        if count >= 3:
            insights.append(f"Heavy workload on {date} ({count} tasks). Start early.")

    tomorrow = datetime.today().date() + timedelta(days=1)

    for task in upcoming_tasks:
        due = datetime.fromisoformat(task["raw_due_date"]).date()
        if due == tomorrow:
            insights.append(f"'{task['title']}' is due tomorrow. Start now.")

    return insights


def build_demo_tasks():
    today = datetime.today().date()

    def make_task(title, due_date):
        return {
            "properties": {
                "Name": {
                    "title": [{"text": {"content": title}}],
                },
                "Due Date": {
                    "date": {"start": due_date.isoformat()},
                },
            }
        }

    return [
        make_task("Operating Systems Viva Prep", today - timedelta(days=2)),
        make_task("DBMS Assignment Final Edit", today - timedelta(days=1)),
        make_task("Computer Networks Lab Record", today),
        make_task("AI Quiz Revision", today),
        make_task("Data Structures Worksheet", today + timedelta(days=1)),
        make_task("Math Tutorial Submission", today + timedelta(days=3)),
        make_task("Mini Project Checkpoint", today + timedelta(days=5)),
    ]


def run_briefing(use_demo_data=False):
    print("Morning Briefing\n")

    tasks = build_demo_tasks() if use_demo_data else get_all_tasks_debug()

    if not tasks:
        print("No tasks found.")
        return

    today_tasks, upcoming_tasks, overdue_tasks = categorize_tasks(tasks)

    print(
        f"Summary: Today={len(today_tasks)}, "
        f"Upcoming={len(upcoming_tasks)}, "
        f"Overdue={len(overdue_tasks)}"
    )

    insights = generate_insights(today_tasks, upcoming_tasks, overdue_tasks)
    briefing = generate_briefing(today_tasks, upcoming_tasks, overdue_tasks)

    print("\nYour Day\n")

    if insights:
        print("Insights\n")
        for insight in insights:
            print(f"- {insight}")
        print("\n----------------------\n")

    print(briefing)