import os
from dotenv import load_dotenv

load_dotenv()


class ImapConfig:
    IMAP_SERVER = os.getenv("IMAP_SERVER", "imap.yandex.ru")
    IMAP_PORT = int(os.getenv("IMAP_PORT", "993"))
    IMAP_USER = os.getenv("IMAP_USER", "")
    IMAP_PASSWORD = os.getenv("IMAP_PASSWORD", "")
    IMAP_FOLDER = os.getenv("IMAP_FOLDER", "запись разговоров")

    DIRECTUS_URL = os.getenv("DIRECTUS_URL", "http://localhost:8055")
    DIRECTUS_TOKEN = os.getenv("DIRECTUS_TOKEN", "")

    WHISPER_MODEL_SIZE = os.getenv("WHISPER_MODEL_SIZE", "turbo")
    WHISPER_DEVICE = os.getenv("WHISPER_DEVICE", "cuda")
    WHISPER_COMPUTE_TYPE = os.getenv("WHISPER_COMPUTE_TYPE", "float16")

    TEMP_DIR = os.path.join(os.path.dirname(os.path.dirname(__file__)), "temp")


cfg = ImapConfig()