sample_reports.py

# sample_reports.py

# Raw incident reports
reports = {
    "r1": "Patron reports that John’s cup was left on the desk near the East Entrance.",
    "r2": "Riley’s mom called to ask about a previous incident report and wanted clarification.",
    "r3": "A teen was running around the Children’s Area and nearly bumped into other patrons.",
    "r4": "Two students left their backpacks blocking the main aisle near the computer stations.",
    "r5": "An argument broke out between Mark and another patron near the circulation desk.",
    "r6": "A patron spilled food in the Quiet Zone and did not clean it up.",
    "r7": "A child pushed open the emergency exit door near the Study Rooms.",
    "r8": "A man was found sleeping on the sofa in the Reading Lounge for over an hour.",
    "r9": "Staff noticed that someone carved initials into the wooden table near Window Bay 3.",
    "r10": "A woman was speaking loudly on her phone in the Quiet Study Area despite multiple warnings.",
}

# Gold labels for evaluation
gold_labels = {
    "r1": {
        "perpetrators": ["John"],
        "locations": ["East entrance", "desk"],
        "violations": ["Lost or misplaced item"],
    },
    "r2": {
        "perpetrators": ["Riley's mom"],
        "locations": [],
        "violations": ["Information request"],
    },
    "r3": {
        "perpetrators": ["Teen"],
        "locations": ["Children’s Area"],
        "violations": ["Disruptive behavior", "Safety risk"],
    },
    "r4": {
        "perpetrators": ["Students"],
        "locations": ["Main aisle", "Computer stations"],
        "violations": ["Obstructing walkways"],
    },
    "r5": {
        "perpetrators": ["Mark", "Another patron"],
        "locations": ["Circulation desk"],
        "violations": ["Verbal disturbance"],
    },
    "r6": {
        "perpetrators": ["Patron"],
        "locations": ["Quiet Zone"],
        "violations": ["Mess left behind"],
    },
    "r7": {
        "perpetrators": ["Child"],
        "locations": ["Emergency exit", "Study Rooms"],
        "violations": ["Unauthorized exit"],
    },
    "r8": {
        "perpetrators": ["Man"],
        "locations": ["Reading Lounge"],
        "violations": ["Sleeping in library"],
    },
    "r9": {
        "perpetrators": ["Unknown person"],
        "locations": ["Wooden table", "Window Bay 3"],
        "violations": ["Vandalism"],
    },
    "r10": {
        "perpetrators": ["Woman"],
        "locations": ["Quiet Study Area"],
        "violations": ["Noise disturbance", "Failure to comply with staff"],
    },
}