import os
import shutil

def mkdir(dirname):
    if not os.path.isdir(dirname):
        os.mkdir(dirname)

if __name__ == "__main__":


    types = ['fire', 'water', 'electric', 'rock', 'grass', 'ice', 'steel']
    animals = ['cat', 'dog', 'rat', 'rabbit', 'dragon', 'duck', 'turtle', 'butterfly', 'monkey', 'bee', 'fox', 'flower', 'horse', 'jellyfish', 'snake', 'Tyrannosaurus', 'dinosaur', 'fish', 'whale', 'bat', 'bear', 'deer', 'pig', 'eagle', 'chicken']


    mkdir('./output')
    for type_str in types:
        for animal in animals:
            mkdir('./output/{}_{}'.format(type_str, animal))
            for index in range(1):
                mkdir('./output/{}_{}/{}'.format(type_str, animal, index))
                prompt = "a {} type pokemon, it looks like a {}".format(type_str, animal)
                print("PROMPT: ", prompt)

                os.system('python dreamgaussian/main.py --config dreamgaussian/configs/text_mv.yaml prompt="{}" save_path=output'.format(prompt))
                os.system('python dreamgaussian/main2.py --config dreamgaussian/configs/text_mv.yaml prompt="{}" save_path=output'.format(prompt))

                for file in os.listdir('./logs'):
                    shutil.move('./logs/{}'.format(file), './output/{}_{}/{}/{}'.format(type_str, animal, index, file))