Spaces:
Running
Running
| import os | |
| import logging | |
| from openai import OpenAI | |
| try: | |
| from dotenv import load_dotenv | |
| load_dotenv() | |
| except ImportError: | |
| pass | |
| class Envs: | |
| def __init__(self): | |
| self.hf_token = os.getenv("HF_TOKEN") | |
| self.api_key = os.getenv("API_KEY") | |
| self.end_task = os.getenv("END_TASK") | |
| self.end_lang = os.getenv("END_LANG") | |
| self.model_agent = os.getenv("MODEL_AGENT") | |
| self.model_mllm = os.getenv("MODEL_MLLM") | |
| self.model_loc = os.getenv("MODEL_LOC") | |
| # Only initialize OpenAI client if we have the required env vars | |
| if self.end_lang and self.api_key: | |
| self.client = OpenAI(base_url=self.end_lang, api_key=self.api_key) | |
| else: | |
| self.client = None | |
| print("WARNING: OpenAI client not initialized due to missing environment variables") | |
| self.debug = os.getenv("DEBUG", "1").lower() in ("true", "1", "yes") | |
| self.fps = int(os.getenv("FPS")) | |
| env = Envs() | |
| logger = logging.getLogger('copilot') | |
| logger.setLevel(logging.DEBUG if env.debug else logging.INFO) | |
| logger.addHandler(logging.StreamHandler()) |