t2m / src /config /config.py
thanhkt's picture
Upload 75 files
9b5ca29 verified
import os
from dotenv import load_dotenv
# Load environment variables from .env file
load_dotenv()
class Config:
OUTPUT_DIR = "output"
THEOREMS_PATH = os.path.join("data", "easy_20.json")
CONTEXT_LEARNING_PATH = "data/context_learning"
CHROMA_DB_PATH = "data/rag/chroma_db"
MANIM_DOCS_PATH = "data/rag/manim_docs"
EMBEDDING_MODEL = "hf:ibm-granite/granite-embedding-30m-english"
# Kokoro TTS configurations
KOKORO_MODEL_PATH = os.getenv('KOKORO_MODEL_PATH')
KOKORO_VOICES_PATH = os.getenv('KOKORO_VOICES_PATH')
KOKORO_DEFAULT_VOICE = os.getenv('KOKORO_DEFAULT_VOICE')
KOKORO_DEFAULT_SPEED = float(os.getenv('KOKORO_DEFAULT_SPEED', '1.0'))
KOKORO_DEFAULT_LANG = os.getenv('KOKORO_DEFAULT_LANG')