Spaces:
Runtime error
Runtime error
File size: 2,681 Bytes
2a28594 da9ac99 2a28594 9b3b414 2a28594 9b3b414 2a28594 da9ac99 2a28594 9b3b414 2a28594 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 |
from datetime import datetime
from os import getenv
from pathlib import Path
from dotenv import load_dotenv
from loguru import logger
from requests import get
load_dotenv()
SERVER_URL: str = getenv(key="SERVER_URL", default="127.0.0.1")
SERVER_PORT: int = int(getenv(key="SERVER_PORT", default="7860"))
CURRENT_DATE: str = datetime.now().strftime(format="%Y-%m-%d_%H-%M-%S")
MCP_VOICE_GENERATOR: str = getenv(
key="MCP_VOICE_GENERATOR", default="http://localhost:8001/"
)
MCP_VIDEO_GENERATOR: str = getenv(
key="MCP_VIDEO_GENERATOR", default="http://localhost:8002/"
)
VECTOR_DATABASE_NAME: str = getenv(
key="VECTOR_DATABASE_NAME", default="chattr"
)
DOCKER_MODEL_RUNNER_URL: str = getenv(
key="DOCKER_MODEL_RUNNER_URL", default="http://127.0.0.1:12434/engines/v1"
)
DOCKER_MODEL_RUNNER_MODEL_NAME: str = getenv(
key="DOCKER_MODEL_RUNNER_MODEL_NAME",
default="ai/qwen3:0.6B-Q4_0",
)
GROQ_URL: str = getenv(
key="MODEL_URL", default="https://api.groq.com/openai/v1"
)
GROQ_MODEL_NAME: str = getenv(key="GROQ_MODEL_NAME", default="llama3-70b-8192")
BASE_DIR: Path = Path.cwd()
ASSETS_DIR: Path = BASE_DIR / "assets"
LOG_DIR: Path = BASE_DIR / "logs"
IMAGE_DIR: Path = ASSETS_DIR / "image"
AUDIO_DIR: Path = ASSETS_DIR / "audio"
VIDEO_DIR: Path = ASSETS_DIR / "video"
LOG_FILE_PATH: Path = LOG_DIR / f"{CURRENT_DATE}.log"
AUDIO_FILE_PATH: Path = AUDIO_DIR / f"{CURRENT_DATE}.wav"
VIDEO_FILE_PATH: Path = VIDEO_DIR / f"{CURRENT_DATE}.mp4"
ASSETS_DIR.mkdir(exist_ok=True)
IMAGE_DIR.mkdir(exist_ok=True)
AUDIO_DIR.mkdir(exist_ok=True)
VIDEO_DIR.mkdir(exist_ok=True)
LOG_DIR.mkdir(exist_ok=True)
MODEL_URL: str = (
DOCKER_MODEL_RUNNER_URL
if get(DOCKER_MODEL_RUNNER_URL, timeout=10).status_code == 200
else GROQ_URL
)
MODEL_NAME: str = (
DOCKER_MODEL_RUNNER_MODEL_NAME
if MODEL_URL == DOCKER_MODEL_RUNNER_URL
else GROQ_MODEL_NAME
)
MODEL_API_KEY: str = (
"not-needed"
if MODEL_URL == DOCKER_MODEL_RUNNER_URL
else getenv("GROQ_API_KEY")
)
MODEL_TEMPERATURE: float = float(getenv(key="MODEL_TEMPERATURE", default=0.0))
logger.add(
sink=LOG_FILE_PATH,
format="{time:YYYY-MM-DD at HH:mm:ss} | {level} | {message}",
colorize=True,
)
logger.info(f"Current date: {CURRENT_DATE}")
logger.info(f"Base directory: {BASE_DIR}")
logger.info(f"Assets directory: {ASSETS_DIR}")
logger.info(f"Log directory: {LOG_DIR}")
logger.info(f"Audio file path: {AUDIO_FILE_PATH}")
logger.info(f"Log file path: {LOG_FILE_PATH}")
logger.info(f"Model URL is going to be used is {MODEL_URL}")
logger.info(f"Model name is going to be used is {MODEL_NAME}")
logger.info(f"Model temperature is going to be used is {MODEL_TEMPERATURE}")
|