examples.py

"""Built-in demo tasks — shown as Gradio examples."""

from pathlib import Path

from config import DATA_DIR

DEMO_DATASETS: dict[str, Path] = {
    "Sales (CSV)": DATA_DIR / "sales.csv",
    "Employees (CSV)": DATA_DIR / "employees.csv",
    "Students (CSV)": DATA_DIR / "students.csv",
}

DEMO_EXAMPLES: list[list[str]] = [
    [
        "Sales (CSV)",
        "Which product had the highest total revenue across all months?",
    ],
    [
        "Sales (CSV)",
        "How many units of Widget A were sold in total?",
    ],
    [
        "Employees (CSV)",
        "What is the average salary in the Engineering department?",
    ],
    [
        "Employees (CSV)",
        "Which city has the highest average salary?",
    ],
    [
        "Students (CSV)",
        "What percentage of students passed math (pass_math=True)?",
    ],
    [
        "Students (CSV)",
        "Who is the student with the highest average score across math, reading, and writing?",
    ],
]