"""
Database setup for Graffiti Tracking & Removal
"""

import sys
from pathlib import Path
sys.path.append(str(Path(__file__).parent.parent.parent / "core"))

from database import VIMSDatabase


def setup():
    """Initialize database for graffiti-tracking."""
    db = VIMSDatabase("graffiti-tracking")
    db.create_schema(anomaly_classes=['new_graffiti', 'tagging', 'gang_symbols', 'hate_speech', 'property_damage'])
    print(f"Database initialized for Graffiti Tracking & Removal")


if __name__ == "__main__":
    setup()