import os
from dotenv import load_dotenv

load_dotenv(dotenv_path=os.path.join(os.path.dirname(__file__), '..', '.env'))

from database import Base, engine, SessionLocal
import models
from crud import get_setting, set_setting

def init_db():
    Base.metadata.create_all(bind=engine)

    db = SessionLocal()
    if not get_setting(db, "ollama_url"):
        set_setting(db, "ollama_url", os.getenv("OLLAMA_URL", "http://localhost:11434"))
    if not get_setting(db, "default_model"):
        set_setting(db, "default_model", os.getenv("DEFAULT_MODEL", "llama3"))
    db.close()

    print("Database initialized.")

if __name__ == "__main__":
    init_db()