from PIL import Image  # PIP install pillow

# Built-in libraries
import sys
import os
import re


def main():
    input_dir = os.path.abspath(sys.argv[1])  # 2nd Argument from command
    output_dir = os.path.abspath(sys.argv[2])  # 3rd Argument from command

    if not os.path.exists(output_dir):
        # check if the output dir doesn't exist, if so, create one
        os.mkdir(output_dir)

    if os.path.exists(input_dir):
        # loop through all files in input_dir
        for fname in os.listdir(input_dir):
            if fname.endswith('.jpg'):  # just get the files with .jpg extension
                image = Image.open(os.path.join(
                    input_dir, fname))  # Open image
                pattern = r'(.*)\.'  # set a regex pattern
                # Search for the pattern in the name of the file
                data = re.search(pattern, fname)
                # save the same name but in .png
                image.save(f'{output_dir}/{data.group(1)}.png')
    else:
        # print that the input dir doesn't exist if it actually doesn't.
        print(input_dir + 'does not exists')


if __name__ == '__main__':
    main()