import sys, subprocess, time, os, shutil

# FFMPEG_PATH = "./ffmpeg/ffmpeg"

# text formatting
class cmdUtils:
    Purple = '\033[95m'
    Blue = '\033[94m'
    Cyan = '\033[96m'
    Green = '\033[92m'
    Tan = '\033[93m'
    Red = '\033[91m'
    Grey = '\033[0m' # Default
    White = '\033[1m'
    Sound = '\007'

# console clearer
def cls():
    os.system('cls' if os.name=='nt' else 'clear')

def encode_video(inputFile, outputPath):

	ffmpeg_cmd = FFMPEG_PATH
	ffmpeg_cmd += f' -y'
	ffmpeg_cmd += f' -i "classroom.jpg"' ### METTRE LE FICHIER D'ARRIÈRE PLAN ICI
	ffmpeg_cmd += f' -i "{inputFile}"'

	ffmpeg_cmd += f' -filter_complex "overlay=(W-w)/2:(H-h)/2"'

	ffmpeg_cmd += f' "{outputPath}"'

	print(ffmpeg_cmd)
	subprocess.call(ffmpeg_cmd)

if __name__ == "__main__":

	### Checking FFmpeg installation
	FFMPEG_PATH = shutil.which('ffmpeg') or './ffmpeg/ffmpeg'
	try:
	    subprocess.run([FFMPEG_PATH, '-version'], check=True)  # ensure ffmpeg is available
	except:
	    print(f'{cmdUtils.Red}Error: Couldn\'t find FFMPEG.{cmdUtils.Grey}')
	    input()
	    sys.exit(1)

	### Get dragndroped files and add file output suffix
	droppedFiles = sys.argv[1:]
	for inputFile in droppedFiles:
		print(f'{cmdUtils.Green}Detected input files : {inputFile}{cmdUtils.Grey}')
		outputPath = f'{inputFile.rsplit(".",1)[0]}__overlay.png'
		print(f'{cmdUtils.Green}Output file will be: {outputPath}{cmdUtils.Grey}')

		### Run main fun
		encode_video(inputFile, outputPath)

	### Success Message
	cls()
	print(cmdUtils.Sound)
	print(f"{cmdUtils.Green}Encoding succesful. This window will close after 10 seconds.{cmdUtils.Grey}")
	subprocess.call('timeout /t 10')