app/config.py

# app/config.py
import os
from dotenv import load_dotenv

load_dotenv()

HF_TOKEN = os.getenv("HF_TOKEN", "")

MODEL_A_LIST = [
    "tiiuae/falcon-7b-instruct",
    "HuggingFaceH4/zephyr-7b-alpha",
    "EleutherAI/pythia-2.8b-deduped",
    "google/flan-t5-base"
]

MODEL_B_LIST = [
    "google/flan-t5-large",
    "google/flan-t5-base",
    "tiiuae/falcon-3b-instruct"
]

MODEL_C_LIST = [
    "tiiuae/falcon-7b-instruct",
    "google/flan-t5-large"
]

MODEL_D_LIST = [
    "tiiuae/falcon-7b-instruct",
    "HuggingFaceH4/zephyr-7b-alpha"
]


SECRET_KEY = os.getenv("SECRET_KEY", "change_this_secret")
ACCESS_TOKEN_EXPIRE_MINUTES = int(os.getenv("ACCESS_TOKEN_EXPIRE_MINUTES", "120"))

DB_PATH = os.getenv("DB_PATH", "app/db/gate_ce_multi.db")