import os
import json

OPENAI_API_KEY: str = os.environ.get("OPENAI_API_KEY", "")

if not OPENAI_API_KEY:
    print("You Should Set OPENAI_API_KEY Enviroment.")
    exit(-1)

config = {
    "USE_MODELSCOPE": "0",
    "translators": [
        {
            "name": "openai",
            "envs": {
                # 设置环境变量
                "OPENAI_BASE_URL": "https://megatrump-oneapi.hf.space/v1",
                "OPENAI_API_KEY": OPENAI_API_KEY,
                "OPENAI_MODEL": "gpt-4o-mini",
            }
        }
    ],
    "ENABLED_SERVICES": [
        "OpenAI",
    ],
    "HIDDEN_GRADIO_DETAILS": True,
}

with open("config.json", "w") as f:
    json.dump(config, f)