"""
Database setup for Urban Vegetation Management
"""

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 vegetation-management."""
    db = VIMSDatabase("vegetation-management")
    db.create_schema(anomaly_classes=['overgrown_branches', 'dead_tree', 'invasive_species', 'root_damage', 'fallen_branches'])
    print(f"Database initialized for Urban Vegetation Management")


if __name__ == "__main__":
    setup()