videoscript.py

from datetime import datetime
from moviepy.editor import AudioFileClip
import voiceover, re

MAX_WORDS_PER_COMMENT = 300
MIN_COMMENTS_FOR_FINISH = 4
MIN_DURATION = 30
MAX_DURATION = 59

class VideoScript:
    title = ""
    fileName = ""
    titleSCFile = ""
    url = ""
    totalDuration = 0
    frames = []
    read_comments = True

    def __init__(self, url, title, fileId, read_comments=True) -> None:
        self.read_comments = read_comments
        if not read_comments:
            global MAX_WORDS_PER_COMMENT, MIN_COMMENTS_FOR_FINISH, MIN_DURATION, MAX_DURATION
            MAX_WORDS_PER_COMMENT = 9999
            MIN_COMMENTS_FOR_FINISH = 0
            MIN_DURATION = 0
            MAX_DURATION = 9999
        self.fileName = f"{datetime.today().strftime('%Y-%m-%d')}-{fileId}"
        self.url = url
        self.title = title
        self.titleAudioClip = self.__createVoiceOver("title", title)

    def canBeFinished(self) -> bool:
        return (len(self.frames) > 0) and (self.totalDuration > MIN_DURATION)

    def canQuickFinish(self) -> bool:
        return (len(self.frames) >= MIN_COMMENTS_FOR_FINISH) and (self.totalDuration > MIN_DURATION)

    def addCommentScene(self, text, commentId) -> None:
        wordCount = len(text.split())
        if (wordCount > MAX_WORDS_PER_COMMENT):
            return True
        frame = ScreenshotScene(text, commentId)
        frame.audioClip = self.__createVoiceOver(commentId, text)
        if (frame.audioClip == None):
            return True
        self.frames.append(frame)

    def addStoryScene(self, text, paragraph_number) -> None:
        frame = ScreenshotScene(text, paragraph_number)
        frame.audioClip = self.__createVoiceOver(paragraph_number, text)
        if (frame.audioClip == None):
            return True
        self.frames.append(frame)

    def getDuration(self):
        return self.totalDuration

    def getFileName(self):
        return self.fileName
    
    def clean_text(self, text):
        url_pattern = re.compile(r'https?://\S+|www\.\S+')
        cleaned_text = url_pattern.sub('', text)
        cleaned_text = cleaned_text.lower().replace("aita", "am I the Ay hole", 10)
        cleaned_text = cleaned_text.lower().replace("aitah", "am I the Ay hole", 10)
        cleaned_text = cleaned_text.lower().replace("wibta", "would I be the Ay hole", 10)
        cleaned_text = cleaned_text.lower().replace("wibtah", "would I be the Ay hole", 10)
        cleaned_text = cleaned_text.lower().replace("tifu", "today I f'd up", 10)
        return cleaned_text

    def __createVoiceOver(self, name, text):
        text = self.clean_text(text)
        script_path = f"Scripts/{self.fileName}"
        with open(script_path, 'w') as file:
            file.write(text)
        
        special_voice = False
        if len(text) < 10 and name != "title":
            special_voice = True

        file_path = voiceover.create_voice_over(f"{self.fileName}-{name}", script_path, special_voice, self.read_comments)
        print(f"Created voice over: {file_path}")
        audioClip = AudioFileClip(file_path)
        if (self.totalDuration + audioClip.duration > MAX_DURATION):
            return None
        self.totalDuration += audioClip.duration
        return audioClip


class ScreenshotScene:
    text = ""
    screenShotFile = ""
    commentId = ""

    def __init__(self, text, commentId) -> None:
        self.text = text
        self.commentId = commentId