73 lines
1.9 KiB
Python
73 lines
1.9 KiB
Python
|
from pathlib import Path
|
||
|
|
from typing import Optional
|
||
|
|
|
||
|
|
from pydantic_settings import BaseSettings, SettingsConfigDict
|
||
|
|
|
||
|
|
|
||
|
|
class Settings(BaseSettings):
|
||
|
|
model_config = SettingsConfigDict(
|
||
|
|
env_file=".env",
|
||
|
|
env_file_encoding="utf-8",
|
||
|
|
case_sensitive=False,
|
||
|
|
extra="ignore",
|
||
|
|
)
|
||
|
|
|
||
|
|
env: str = "development"
|
||
|
|
host: str = "0.0.0.0"
|
||
|
|
port: int = 8085
|
||
|
|
log_level: str = "INFO"
|
||
|
|
api_v1_prefix: str = "/api"
|
||
|
|
project_name: str = "BotModels API"
|
||
|
|
version: str = "2.0.0"
|
||
|
|
api_key: str = "change-me"
|
||
|
|
|
||
|
|
# External Providers for Speech (Optional)
|
||
|
|
groq_api_key: Optional[str] = None
|
||
|
|
openai_api_key: Optional[str] = None
|
||
|
|
|
||
|
|
# Image generation model
|
||
|
|
image_model_path: str = "./models/stable-diffusion-v1-5"
|
||
|
|
image_steps: int = 4
|
||
|
|
image_width: int = 512
|
||
|
|
image_height: int = 512
|
||
|
|
image_gpu_layers: int = 20
|
||
|
|
image_batch_size: int = 1
|
||
|
|
|
||
|
|
# Video generation model
|
||
|
|
video_model_path: str = "./models/zeroscope-v2"
|
||
|
|
video_frames: int = 24
|
||
|
|
video_fps: int = 8
|
||
|
|
video_width: int = 320
|
||
|
|
video_height: int = 576
|
||
|
|
video_gpu_layers: int = 15
|
||
|
|
video_batch_size: int = 1
|
||
|
|
|
||
|
|
# Speech/TTS model
|
||
|
|
speech_model_path: str = "./models/tts"
|
||
|
|
|
||
|
|
# Vision model (BLIP2 for captioning)
|
||
|
|
vision_model_path: str = "./models/blip2"
|
||
|
|
|
||
|
|
# Whisper model for speech-to-text
|
||
|
|
whisper_model_path: str = "./models/whisper"
|
||
|
|
|
||
|
|
# Real-time Audio model for speech-to-speech
|
||
|
|
realtime_audio_model_path: str = "./models/realtime_audio"
|
||
|
|
|
||
|
|
# Device configuration
|
||
|
|
device: str = "cuda"
|
||
|
|
|
||
|
|
# Output directory for generated files
|
||
|
|
output_dir: Path = Path("./outputs")
|
||
|
|
|
||
|
|
@property
|
||
|
|
def is_production(self) -> bool:
|
||
|
|
return self.env == "production"
|
||
|
|
|
||
|
|
|
||
|
|
settings = Settings()
|
||
|
|
settings.output_dir.mkdir(parents=True, exist_ok=True)
|
||
|
|
(settings.output_dir / "images").mkdir(exist_ok=True)
|
||
|
|
(settings.output_dir / "videos").mkdir(exist_ok=True)
|
||
|
|
(settings.output_dir / "audio").mkdir(exist_ok=True)
|