import os
from sqlmodel import create_engine, SQLModel, Session

DATA_DIR = "data"
os.makedirs(DATA_DIR, exist_ok=True)

engine = create_engine(f"sqlite:///{DATA_DIR}/database.db")
SQLModel.metadata.create_all(engine)

def get_session():
    with Session(engine) as session:
        yield session