main.py

import math
import sys
from uuid import uuid4
from image_fetcher import fetch_images
from script_generator import generate_script
from subtitle_generator import generate_subtitles
from video_maker import burn_subtitles_into_video, create_video
from voice_generator import text_to_speech


def generate_video(topic: str):
    video_id = uuid4()
    audio_filepath = f'output/audio/{video_id}.mp3'
    video_filepath = f'output/video/{video_id}.mp4'
    subtitles_filepath = f'output/subtitles/{video_id}.srt'
    final_output_filepath = f'output/final/{video_id}.mp4'
    seconds_per_image = 4

    print('Generating script...')
    try:
        script = generate_script(topic=topic)
        video_title: str = script['title']
        video_script: str = script['content']
        keywords = script['keywords']
        print(video_title)
        print(video_script)
        print(keywords)
        if not video_script or not keywords:
            print("Error generating script")
            sys.exit()

    except Exception as error:
        print("Error generating script:", error)
        sys.exit()

    # Generate voice-over narration using text-to-speech
    print("Generating voice-over narration...")
    tts_audio = text_to_speech(
        text=video_script, audio_filepath=audio_filepath)
    # Compute number of images required to fit voiceover duration, given that each image is displayed for x seconds
    number_of_images_required: int = math.ceil(
        tts_audio.duration / seconds_per_image)

    # Fetch images related to generated keywords
    print("Fetching images...")
    image_urls = fetch_images(
        search_terms=keywords, number_of_images=number_of_images_required)

    if not image_urls:
        print('No images found')
        sys.exit()

    print("Creating video...")
    create_video(image_urls=image_urls, seconds_per_image=seconds_per_image,
                 audio_filepath=audio_filepath, video_filepath=video_filepath)

    # Generate subtitles based on audio file
    print("Generating subtitles...")
    generate_subtitles(audio_filepath, subtitles_filepath)

    # Burn subtitles into video
    print("Burning subtitles...")
    burn_subtitles_into_video(
        video_filepath=video_filepath, subtitles_filepath=subtitles_filepath, final_output_filepath=final_output_filepath)


def main():
    topic_word_limit = 5
    while True:
        topic = input('Enter a topic: ')
        if (len(topic.split(' ')) <= topic_word_limit):
            break
        print(f'Topic must be {topic_word_limit} or less')
    generate_video(topic=topic)


if __name__ == '__main__':
    main()