Initial commit: local video captioning and translation tool

2026-06-08 13:24:00 -04:00
commit ed29467f47
7 changed files with 778 additions and 0 deletions
@@ -0,0 +1,41 @@
+import os
+from dotenv import load_dotenv
+
+# Load environment variables from .env
+load_dotenv()
+
+# API Keys and tokens
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+
+# Whisper ASR Settings
+DEFAULT_WHISPER_MODEL = "whisper-1"
+
+# LLM Translation settings
+DEFAULT_OPENAI_MODEL = "gpt-4o-mini"
+
+# Defaults
+DEFAULT_SOURCE_LANG = "en"
+DEFAULT_TARGET_LANG = "es"
+
+# Target translation languages map
+SUPPORTED_LANGUAGES = {
+    "en": "English",
+    "es": "Spanish",
+    "fr": "French",
+    "de": "German",
+    "it": "Italian",
+    "pt": "Portuguese",
+    "zh": "Chinese",
+    "ja": "Japanese",
+    "ar": "Arabic",
+    "ru": "Russian"
+}
+
+# Subtitle Visual Presets (ASS style rules for FFmpeg)
+SUBTITLE_PRESETS = {
+    "default": "FontName=Helvetica,FontSize=20,PrimaryColour=&H00FFFFFF,OutlineColour=&H00000000,BorderStyle=1,Outline=1,Shadow=1,MarginV=15",
+    "yellow": "FontName=Helvetica,FontSize=22,PrimaryColour=&H0000FFFF,OutlineColour=&H00000000,BorderStyle=1,Outline=1.5,Shadow=1,MarginV=15",
+    "box": "FontName=Helvetica,FontSize=20,PrimaryColour=&H00FFFFFF,BackColour=&H80000000,BorderStyle=3,Outline=0,Shadow=0,MarginV=15",
+    "clean": "FontName=Arial,FontSize=22,PrimaryColour=&H00FFFFFF,OutlineColour=&H00000000,BorderStyle=1,Outline=0.5,Shadow=0,MarginV=20"
+}
+DEFAULT_STYLE = "box"