configuration.py

import json
from typing import Literal


class Configuration:
    def __init__(self) -> None:
        with open("config/configuration.json") as file:
            config = json.loads(file.read())

        with open("config/config_thumbnail_with_text.json") as file:
            config_thumbnail_with_text = json.loads(file.read())

        self.output_dir = config["output_dir"]
        self.background_videos_dir = config["background_videos_dir"]
        self.tolerated_duration_offset = config["tolerated_duration_offset"]
        self.intro_header: str = config["intro_header"]
        self.intro_prompt: str = config["intro_prompt"]
        self.outro_prompt: str = config["outro_prompt"]
        self.video_title_prompt: str = config["video_title_prompt"]
        self.background_video_volume: float = config["background_video_volume"]
        

        self.init_video_text(config)

        self.init_openai(config)
        self.init_moviepy(config)

        self.init_thumbnail(config_thumbnail_with_text)

    def init_thumbnail(self, config_thumbnail_with_text):
        self.thumbnail_subreddit_icon_size: int = config_thumbnail_with_text[
            "thumbnail_subreddit_icon_size"
        ]
        self.thumbnail_subreddit_text_size: int = config_thumbnail_with_text[
            "thumbnail_subreddit_text_size"
        ]
        self.thumbnail_subreddit_text_color: str = config_thumbnail_with_text[
            "thumbnail_subreddit_text_color"
        ]
        self.thumbnail_subreddit_text_font: str = config_thumbnail_with_text[
            "thumbnail_subreddit_text_font"
        ]
        self.thumbnail_main_text_size: int = config_thumbnail_with_text[
            "thumbnail_main_text_size"
        ]
        self.thumbnail_main_text_color: str = config_thumbnail_with_text[
            "thumbnail_main_text_color"
        ]
        self.thumbnail_main_text_font: str = config_thumbnail_with_text[
            "thumbnail_main_text_font"
        ]
        self.thumbnail_like_icon_size: int = config_thumbnail_with_text[
            "thumbnail_like_icon_size"
        ]
        self.thumbnail_like_text_size: int = config_thumbnail_with_text[
            "thumbnail_like_text_size"
        ]
        self.thumbnail_like_text_color: str = config_thumbnail_with_text[
            "thumbnail_like_text_color"
        ]
        self.thumbnail_like_text_font: str = config_thumbnail_with_text[
            "thumbnail_like_text_font"
        ]
        self.thumbnail_element_margin: int = config_thumbnail_with_text[
            "thumbnail_element_margin"
        ]
        self.thumbnail_background_color: str = config_thumbnail_with_text[
            "thumbnail_background_color"
        ]
        self.thumbnail_side_offset: float = config_thumbnail_with_text[
            "thumbnail_side_offset"
        ]
        self.thumbnail_default_subreddit_icon: str = config_thumbnail_with_text[
            "thumbnail_default_subreddit_icon"
        ]
        self.thumbnail_upvotes_icon: str = config_thumbnail_with_text[
            "thumbnail_upvotes_icon"
        ]
        self.thumbnail_comments_icon: str = config_thumbnail_with_text[
            "thumbnail_comments_icon"
        ]

    def init_video_text(self, config):
        self.video_font: str = config["video_font"]
        self.video_font_color: str = config["video_font_color"]
        self.video_font_size: int = config["video_font_size"]
        self.video_font_stroke_width: int = config["video_font_stroke_width"]
        self.video_font_stroke_color: str = config["video_font_stroke_color"]

    def init_moviepy(self, config):
        self.video_fps: int = config["video_fps"]
        self.write_video_preset: str = config["write_video_preset"]
        self.num_threads: int = config["num_threads"]

    def init_openai(self, config):
        with open("config/secrets.json", "r") as file:
            secrets = json.loads(file.read())
            self.openai_api_key = secrets["openai_api_key"]

        self.audio_api: Literal["openai"] = config["audio_api"]
        if self.audio_api == "openai":
            self.audio_model: Literal["tts-1", "tts-1-hd"] = config["audio_model"]
            self.audio_voice: Literal[
                "alloy", "echo", "fable", "onyx", "nova", "shimmer"
            ] = config["audio_voice"]