main.py

#!/bin/python

#basic Python libraries
import random
import os

#joke Api module
import joke_gen as test

#non pre installed packages
try:
	import audioread
except:
	os.system("pip install -r requirements.txt")

#code begins:
	
class create:
	def __init__(self,img):
		self.test = test
		self.img = img
		self.start = 0
		self.end = 0
		self.iterator1 = 1
	def createAudio(self,aud1,aud2):
		self.test.get()
		self.audio_1 = aud1
		self.audio_2 = aud2
		self.test = test
	def createText(self):
 		self.subtitle = ""
 		self.quote = [test.data["setup"],"Reply: "+test.data["delivery"]]
 		for self.i in self.quote:
 			if len(str(self.start)) == 1:
 				self.start = f"0{self.start}"
 			if len(str(self.end)) == 1:
 				self.end = f"0{self.end}"
 			s = f"""{self.iterator1}
00:00:{self.start},000 --> 00:00:{self.end},000
{self.i}
"""
 			self.subtitle = self.subtitle + s
 			self.start = int(self.start)
 			self.end = int(self.end)
 			self.start = self.end
 			self.end = self.end + self.duration_aud_2 +1
 			self.iterator1 += 1
 		self.file = open("subtitle.srt", "w") # create and open for writing
 		self.file.write(self.subtitle)
 		self.file.close() #to change file access modes

	def preProduce(self):
		self.duration_aud_1 = audioread.audio_open(f"process/{self.audio_1}").duration
		self.duration_aud_2 = audioread.audio_open(f"process/{self.audio_2}").duration
		print(self.duration_aud_2 + self.duration_aud_1)
		#change some
		self.end = self.duration_aud_1


	def createFile(self):
		os.system(f"""ffmpeg -loop 1 -y -i input/{self.img} -i additional/overlay.png -filter_complex "[1]scale=    iw/2:-1[b];[0:v][b] overlay" -c:v libx264 -t {self.duration_aud_2 + self.duration_aud_1 + 1} -pix_fmt yuv420p -vf scale=1080:1920 process/out-cache.avi""")
		os.system(f"""ffmpeg -y -i "concat:process/{self.audio_1}|process/{self.audio_2}" -i process/out-cache.avi -vf "subtitles=subtitle.srt:force_style='Alignment=10,Fontsize=18,PrimaryColour=&H0xFAEBD7&,FontName=DejaVu Serif'" -c:a copy out-main.avi""")


def generate():
    n = 1 						#n = random.randint(0,any number of inputs)
    video = create(f"input.jpg")			#video = create(f"input{n}.jpg") #add n to get a random image from input folder:)
    print(" Creating audio...   ",end="")
    video.createAudio("audio.mp3","audio2.mp3")
    print("DONE \n Preproduce...    ", end="")
    video.preProduce()
    print("DONE \n Generating subtitles...  ", end="")
    video.createText()
    print("DONE \n Generating video...  ",end="")
    video.createFile()
    print("DONE \n\n\n")
    print("Video file saved as out-main.avi")

generate()