Cria novos components

video_render/config.py

@@ -0,0 +1,103 @@
+from __future__ import annotations
+
+import os
+from dataclasses import dataclass
+from pathlib import Path
+
+
+BASE_DIR = Path(__file__).resolve().parent.parent
+VIDEOS_ROOT = BASE_DIR / "videos"
+OUTPUTS_ROOT = BASE_DIR / "outputs"
+TEMP_ROOT = BASE_DIR / "temp"
+
+
+@dataclass(frozen=True)
+class RabbitMQSettings:
+    host: str = os.environ.get("RABBITMQ_HOST", "rabbitmq")
+    port: int = int(os.environ.get("RABBITMQ_PORT", 5672))
+    user: str = os.environ.get("RABBITMQ_USER", "admin")
+    password: str = os.environ.get("RABBITMQ_PASS", "")
+    consume_queue: str = os.environ.get("RABBITMQ_QUEUE", "to-render")
+    publish_queue: str = os.environ.get("RABBITMQ_UPLOAD_QUEUE", "to-upload")
+    prefetch_count: int = int(os.environ.get("RABBITMQ_PREFETCH", 1))
+    heartbeat: int = int(os.environ.get("RABBITMQ_HEARTBEAT", 60))
+    blocked_timeout: int = int(os.environ.get("RABBITMQ_BLOCKED_TIMEOUT", 300))
+
+
+@dataclass(frozen=True)
+class GeminiSettings:
+    api_key: str = os.environ.get("GEMINI_API_KEY", "")
+    model: str = os.environ.get("GEMINI_MODEL", "gemini-1.5-pro-latest")
+    safety_settings: str | None = os.environ.get("GEMINI_SAFETY_SETTINGS")
+    temperature: float = float(os.environ.get("GEMINI_TEMPERATURE", 0.2))
+    top_k: int | None = (
+        int(os.environ["GEMINI_TOP_K"]) if os.environ.get("GEMINI_TOP_K") else None
+    )
+    top_p: float | None = (
+        float(os.environ["GEMINI_TOP_P"]) if os.environ.get("GEMINI_TOP_P") else None
+    )
+    prompt_path: str = os.environ.get("GEMINI_PROMPT_PATH", "prompts/generate.txt")
+
+
+@dataclass(frozen=True)
+class OpenRouterSettings:
+    api_key: str = os.environ.get("OPENROUTER_API_KEY", "")
+    model: str = os.environ.get(
+        "OPENROUTER_MODEL", "anthropic/claude-3-haiku:beta"
+    )
+    temperature: float = float(os.environ.get("OPENROUTER_TEMPERATURE", 0.6))
+    max_output_tokens: int = int(os.environ.get("OPENROUTER_MAX_OUTPUT_TOKENS", 256))
+
+
+@dataclass(frozen=True)
+class WhisperSettings:
+    model_size: str = os.environ.get("FASTER_WHISPER_MODEL_SIZE", "medium")
+    device: str | None = os.environ.get("FASTER_WHISPER_DEVICE")
+    compute_type: str | None = os.environ.get("FASTER_WHISPER_COMPUTE_TYPE")
+    download_root: Path = Path(
+        os.environ.get("FASTER_WHISPER_DOWNLOAD_ROOT", str(BASE_DIR / ".whisper"))
+    )
+
+
+@dataclass(frozen=True)
+class RenderingSettings:
+    frame_width: int = int(os.environ.get("RENDER_WIDTH", 1080))
+    frame_height: int = int(os.environ.get("RENDER_HEIGHT", 1920))
+    fps: int = int(os.environ.get("RENDER_FPS", 30))
+    video_codec: str = os.environ.get("RENDER_CODEC", "libx264")
+    audio_codec: str = os.environ.get("RENDER_AUDIO_CODEC", "aac")
+    bitrate: str = os.environ.get("RENDER_BITRATE", "5000k")
+    preset: str = os.environ.get("RENDER_PRESET", "faster")
+    highlight_color: str = os.environ.get("SUBTITLE_HIGHLIGHT_COLOR", "#FFD200")
+    base_color: str = os.environ.get("SUBTITLE_BASE_COLOR", "#FFFFFF")
+    font_path: Path = Path(os.environ.get("RENDER_FONT_PATH", "./Montserrat.ttf"))
+    title_font_size: int = int(os.environ.get("RENDER_TITLE_FONT_SIZE", 110))
+    subtitle_font_size: int = int(os.environ.get("RENDER_SUBTITLE_FONT_SIZE", 64))
+    caption_min_words: int = int(os.environ.get("CAPTION_MIN_WORDS", 3))
+    caption_max_words: int = int(os.environ.get("CAPTION_MAX_WORDS", 4))
+
+
+@dataclass(frozen=True)
+class Settings:
+    rabbitmq: RabbitMQSettings = RabbitMQSettings()
+    gemini: GeminiSettings = GeminiSettings()
+    openrouter: OpenRouterSettings = OpenRouterSettings()
+    whisper: WhisperSettings = WhisperSettings()
+    rendering: RenderingSettings = RenderingSettings()
+
+    videos_dir: Path = VIDEOS_ROOT
+    outputs_dir: Path = OUTPUTS_ROOT
+    temp_dir: Path = TEMP_ROOT
+
+
+def load_settings() -> Settings:
+    settings = Settings()
+
+    if not settings.rabbitmq.password:
+        raise RuntimeError("RABBITMQ_PASS must be provided")
+
+    settings.videos_dir.mkdir(parents=True, exist_ok=True)
+    settings.outputs_dir.mkdir(parents=True, exist_ok=True)
+    settings.temp_dir.mkdir(parents=True, exist_ok=True)
+
+    return settings