make_compilation.py

from moviepy.editor import VideoFileClip, concatenate_videoclips
from moviepy.video.fx.resize import resize
import os
from os.path import isfile, join
import random
import shutil 
from collections import defaultdict
import json
from pyffmpeg import FFmpeg


ff = FFmpeg()
VideoFileClip.resize = resize

def extractAcc(filepath):
    try:
        s = filepath.split("/")[-1].split("-")
        acc = "-".join(s[1:(2+(len(s) - 4))])
        return acc
    except:
        return ""


# generateTimeRange converts float seconds to a range of form @MM:SS
def generateTimeRange(duration, clipDuration):
    preHour = int(duration / 60)
    preMin = int(duration % 60)
    preTime = str(preHour // 10) + str(preHour % 10) + ":" + str(preMin // 10) + str(preMin % 10)

    duration += clipDuration
    postHour = int(duration / 60)
    postMin = int(duration % 60)
    postTime = str(postHour // 10) + str(postHour % 10) + ":" + str(postMin // 10) + str(postMin % 10)

    #return "@" + preTime + " - " + "@" + postTime
    return "@" + preTime


# makeCompilation takes videos in a folder and creates a compilation with max length totalVidLength
def makeCompilation(path = "./",
                    introName = '',
                    outroName = '',
                    wmark = '',
                    totalVidLength = 10,
                    maxClipLength = 20,
                    minClipLength = 5,
                    outputFile = "output.mp4",
                    video_source_meta = {},
                    videoDirectory = "",
                    description_meta = "",
                    modeAM = "A"):

    downVideos = []
    seenLengths = defaultdict(list)
    #totalLength = 0
    duration = 0
    videos = []

    # Add intro video if included
    if introName != '':
        introVid = VideoFileClip("./" + introName)
        videos.append(introVid)
        timeStamp = generateTimeRange(duration, introVid.duration)
        duration += introVid.duration

    for fileName in os.listdir(path):
        filePath = join(path, fileName)

        if isfile(filePath) and fileName.endswith(".mp4"):

            if os.stat(filePath).st_size < 5000:
                continue

            # Destination path
            print("[i] ", filePath)
            clip = VideoFileClip(filePath)
            clip = clip.resize(width=1920)
            clip = clip.resize(height=1080)
            duration = clip.duration
            print("[i] " + fileName + " " + str(duration) + " Added")

            # add_video in min&max range or ignore errors
            def add_video(duration):
                downVideos.append(clip)
                seenLengths[duration].append(fileName)
                duration += clip.duration


            print("[i] ", duration, seenLengths, downVideos)

            if modeAM == "A":
                add_video(duration)
            elif modeAM == "M":
                if duration <= maxClipLength and duration >= minClipLength:
                    add_video(duration)
                else:
                    ignore_error = input("[Q] Do you want to ignore Errors in min max Total Video Length?(Y/n)").strip()
                    if ignore_error != "n":
                        pass
                    else:
                        add_video(duration)

            #Add automated description

            for k in range(len(os.listdir(path))):
                
                fileNameJ = fileName.split(".mp4")
                fileNameJSON = ''.join(fileNameJ) + ".json"

                acc = extractAcc(clip.filename)

                timeStamp = generateTimeRange((duration - clip.duration), clip.duration)
                video_source_meta[f"TimeStamps{k}"] = timeStamp + " : @" + acc + "\n"

                video_source_meta[f"profile{k}"] = "Instagram profile:" + "  instagram.com/" + acc +'\n'
                    
                #extract url & other information about video
                
                f = open(f"{videoDirectory}{fileNameJSON}", "r")
                json_d = json.loads(f.read())
                f.close()

                video_source_meta[f"vido_url{k}"] = "Video URL:" + "instagram.com/tv/" + json_d["shortcode"] + '\n'
                video_source_meta[f"Caption{k}"] = json_d["edge_media_to_caption"]["edges"][0]["node"]["text"] + '\n'

                description_meta = video_source_meta[f"TimeStamps{k}"] + video_source_meta[f"profile{k}"] + video_source_meta[f"vido_url{k}"] + video_source_meta[f"Caption{k}"] + '\n\n'

    print("[i] ", description_meta)

    with open(f"{videoDirectory}description.txt", 'a', encoding="utf-8") as dfile:
        dfile.write(description_meta)

    print("[i] Total Length: " + str(duration))

    # Create videos
    for clip in downVideos:
        #duration += clip.duration 
        videos.append(clip)

        if duration >= totalVidLength:
            # Just make one video
            break
    
    # Add outro vid
    if outroName != '':
        outroVid = VideoFileClip("./" + outroName)
        videos.append(outroVid)
    
    # Used Moviepy
    finalClip = concatenate_videoclips(videos, method="compose")

    audio_path = "/tmp/temoaudiofile.m4a"

    # Create compilation
    finalClip.write_videofile(outputFile, threads=8, temp_audiofile=audio_path, remove_temp=True, codec="libx264", audio_codec="aac")
    def watrmrk():
        print("[i] Adding Watermark")
        os.rename(outputFile, f"{outputFile}.tmp")
        ff.options(f"-i {outputFile}.tmp -i {wmark} -filter_complex overlay=1500:10 {outputFile}")
        os.remove(f"{outputFile}.tmp")
        print("[i] Watermark added")
    if modeAM == "M":
        add_waterMark = input(f"[Q] Do you want to add watermark {wmark}(Y/n):").strip()
        if add_waterMark.lower() == "n":
            print("[i] No watermark added")
        else:
            watrmrk()
    else:
        watrmrk()


if __name__ == "__main__":
    makeCompilation(path = "/home/kali/Documents/YOUTUBE/AutomatedChannel/Videos/Memes/",
                    introName = "intro_vid.mp4",
                    outroName = 'outro.mp4',
                    wmark = 'BotTuber.png',
                    totalVidLength = 10*60,
                    maxClipLength = 20,
                    outputFile = "outputseq.mp4",
                    video_source_meta = {},
                    videoDirectory = "",
                    description_meta = "",
                    modeAM = "A")