File size: 2,977 Bytes
c6abe34 | 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 84 85 86 87 88 89 90 91 92 93 94 95 | """
Configuration settings for the Basketball Analysis API.
Uses Pydantic Settings for environment variable management with type validation.
"""
from functools import lru_cache
from typing import Optional
from pydantic_settings import BaseSettings, SettingsConfigDict
class Settings(BaseSettings):
"""Application configuration loaded from environment variables."""
model_config = SettingsConfigDict(
env_file=".env",
env_file_encoding="utf-8",
case_sensitive=False,
extra="ignore"
)
# Application settings
app_name: str = "Basketball Analysis API"
app_version: str = "1.0.0"
debug: bool = False
default_rate_limit: str = "100 per minute"
# Server settings
host: str = "0.0.0.0"
port: int = 8000
request_timeout_seconds: int = 120
# Supabase configuration
supabase_url: Optional[str] = None
supabase_key: Optional[str] = None
supabase_service_key: Optional[str] = None
# JWT settings
jwt_secret: str = "your-super-secret-key-change-in-production"
jwt_algorithm: str = "HS256"
jwt_expiration_minutes: int = 60 * 24 # 24 hours
# CORS (comma-separated list). Example: "http://localhost:5173,https://yourapp.com"
cors_origins: str = ""
# File storage
upload_dir: str = "./uploads"
max_upload_size_mb: int = 500
allowed_video_extensions: str = "mp4,avi,mov,mkv"
serve_uploads_in_debug: bool = True
# GPU settings
gpu_enabled: bool = True
cuda_device: int = 0
# Model paths (relative to backend root)
team_model_path: str = "models/player_detector.pt"
personal_model_path: str = "models/player_detector.pt"
# Independent detector models trained on NBL dataset
player_detector_path: str = "models/player_detector.pt"
ball_detector_path: str = "models/ball_detector.pt"
court_keypoint_detector_path: str = "models/court_keypoint_detector.pt"
pose_model_path: str = "models/yolov8n-pose.pt"
# Swish-Vision specific models
swish_ball_rim_model: str = "models/swish_ball_rim.pt"
swish_pose_model: str = "models/swish_pose.pt"
# Processing settings
batch_size: int = 20
detection_confidence: float = 0.5
@property
def allowed_extensions_list(self) -> list[str]:
"""Get allowed video extensions as a list."""
return [ext.strip().lower() for ext in self.allowed_video_extensions.split(",")]
@property
def cors_origins_list(self) -> list[str]:
"""Get CORS origins as a list."""
if not self.cors_origins:
return []
return [o.strip() for o in self.cors_origins.split(",") if o.strip()]
@property
def max_upload_size_bytes(self) -> int:
"""Get max upload size in bytes."""
return self.max_upload_size_mb * 1024 * 1024
@lru_cache
def get_settings() -> Settings:
"""Get cached settings instance."""
return Settings()
|