import os

def main():
    # Obtém o diretório atual onde o script Python e as imagens estão
    current_directory = os.getcwd()

    # Lista todos os arquivos no diretório atual
    files = os.listdir(current_directory)

    # Filtra apenas os arquivos que terminam com '.png'
    png_files = [file for file in files if file.endswith('.png')]

    # Escreve os nomes dos arquivos em um bloco de notas
    with open("lista_fotos.txt", "w") as file:
        for png in png_files:
            file.write(png + "\n")

    print("Lista de arquivos .png salva em 'lista_fotos.txt'")

if __name__ == "__main__":
    main()