from utils.database import get_db
from data.repository.annotator_repo import AnnotatorRepo
from utils.logger import Logger

log = Logger()

try:
    with get_db() as session:
        repo = AnnotatorRepo(session)

        # ---------------------- Add new user ---------------------- #
        new_user = repo.add_new_user("navid", "123")
        log.info(
            f"User created successfully (id={new_user.id}, username='{new_user.name}')"
        )

        new_user = repo.add_new_user("vargha", "111")
        log.info(
            f"User created successfully (id={new_user.id}, username='{new_user.name}')"
        )

except Exception as exc:
    log.error(f"An error occurred in user operations: {exc}")