YouTube Shorts Automation

def generate_video(self, tts_instance: TTS) -> str:
    # Generate the Topic
    self.generate_topic()

    # Generate the Script
    self.generate_script()

    # Generate the Metadata
    self.generate_metadata()

    # Generate the Image Prompts
    self.generate_prompts()

    # Generate the Images
    for prompt in self.image_prompts:
        self.generate_image(prompt)

    # Generate the TTS
    self.generate_script_to_speech(tts_instance)

    # Combine everything
    path = self.combine()

    return path

def generate_script(self) -> str:
    sentence_length = get_script_sentence_length()
    prompt = f"""
    Generate a script for a video in {sentence_length} sentences, depending on the subject of the video.

    The script is to be returned as a string with the specified number of paragraphs.

    Here is an example of a string:
    "This is an example string."

    Do not under any circumstance reference this prompt in your response.

    Get straight to the point, don't start with unnecessary things like, "welcome to this video".

    Obviously, the script should be related to the subject of the video.
    
    YOU MUST NOT EXCEED THE {sentence_length} SENTENCES LIMIT. MAKE SURE THE {sentence_length} SENTENCES ARE SHORT.
    YOU MUST NOT INCLUDE ANY TYPE OF MARKDOWN OR FORMATTING IN THE SCRIPT, NEVER USE A TITLE.
    YOU MUST WRITE THE SCRIPT IN THE LANGUAGE SPECIFIED IN [LANGUAGE].
    
    Subject: {self.subject}
    Language: {self.language}
    """
    completion = self.generate_response(prompt)
    return completion

def combine(self) -> str:
    combined_image_path = os.path.join(ROOT_DIR, ".mp", str(uuid4()) + ".mp4")
    threads = get_threads()
    tts_clip = AudioFileClip(self.tts_path)
    max_duration = tts_clip.duration
    req_dur = max_duration / len(self.images)

    # Make a generator that returns a TextClip when called
    generator = lambda txt: TextClip(
        txt,
        font=os.path.join(get_fonts_dir(), get_font()),
        fontsize=100,
        color="#FFFF00",
        stroke_color="black",
        stroke_width=5,
        size=(1080, 1920),
        method="caption",
    )

def generate_subtitles(self, audio_path: str) -> str:
    provider = str(get_stt_provider() or "local_whisper").lower()

    if provider == "local_whisper":
        return self.generate_subtitles_local_whisper(audio_path)

    if provider == "third_party_assemblyai":
        return self.generate_subtitles_assemblyai(audio_path)

def upload_video(self) -> bool:
    try:
        self.get_channel_id()

        driver = self.browser

        # Go to youtube.com/upload
        driver.get("https://www.youtube.com/upload")

        # Set video file
        FILE_PICKER_TAG = "ytcp-uploads-file-picker"
        file_picker = driver.find_element(By.TAG_NAME, FILE_PICKER_TAG)
        INPUT_TAG = "input"
        file_input = file_picker.find_element(By.TAG_NAME, INPUT_TAG)
        file_input.send_keys(self.video_path)

        # Wait for upload to finish
        time.sleep(5)

        # Set title
        textboxes = driver.find_elements(By.ID, YOUTUBE_TEXTBOX_ID)

        title_el = textboxes[0]
        description_el = textboxes[-1]

        title_el.click()
        time.sleep(1)
        title_el.clear()
        title_el.send_keys(self.metadata["title"])

        # Set description
        time.sleep(10)
        description_el.click()
        time.sleep(0.5)
        description_el.clear()
        description_el.send_keys(self.metadata["description"])

user_input = int(question("Select an Option: "))

cron_script_path = os.path.join(ROOT_DIR, "src", "cron.py")
command = ["python", cron_script_path, "youtube", selected_account['id'], get_active_model()]

def job():
    subprocess.run(command)

if user_input == 1:
    # Upload Once
    schedule.every(1).day.do(job)
    success("Set up CRON Job.")
elif user_input == 2:
    # Upload Twice a day
    schedule.every().day.at("10:00").do(job)
    schedule.every().day.at("16:00").do(job)
    success("Set up CRON Job.")