import psutil
import os


# Whisper settings
WHISPER_MODEL_NAME = "tiny"
WHISPER_DEVICE = "cpu"
WHISPER_COMPUTE_TYPE = "int8"

# Use token from Hugging Face Space secrets
PYANNOTE_AUTH_TOKEN = os.environ.get("HUGGINGFACE_API_KEY")

# Summarization settings
SUMMARIZER_MODEL = "facebook/bart-large-cnn"
SUMMARIZER_MAX_LENGTH = 150
SUMMARIZER_MIN_LENGTH = 50

# NLP settings
SPACY_MODEL = "en_core_web_sm"
CPU_THREADS = max(1, psutil.cpu_count(logical=False))

# Check that token is set
if not PYANNOTE_AUTH_TOKEN:
    raise ValueError("HUGGINGFACE_API_KEY is not set in environment variables")