| | import os |
| | from datetime import datetime |
| | from mutagen.wave import WAVE |
| | from mutagen.id3._frames import * |
| |
|
| | def create_filename(path, seed, name, extension): |
| | now = datetime.now() |
| | date_str =now.strftime("%m-%d-%Y") |
| | outputs_folder = os.path.join(os.getcwd(), path) |
| | if not os.path.exists(outputs_folder): |
| | os.makedirs(outputs_folder) |
| |
|
| | sub_folder = os.path.join(outputs_folder, date_str) |
| | if not os.path.exists(sub_folder): |
| | os.makedirs(sub_folder) |
| |
|
| | time_str = now.strftime("%H-%M-%S") |
| | if seed == None: |
| | file_name = f"{name}_{time_str}{extension}" |
| | else: |
| | file_name = f"{name}_{time_str}_s{seed}{extension}" |
| | return os.path.join(sub_folder, file_name) |
| |
|
| |
|
| | def add_id3_tag(filename, text, speakername, seed): |
| | audio = WAVE(filename) |
| | if speakername == None: |
| | speakername = "Unconditional" |
| |
|
| | |
| | audio["TIT2"] = TIT2(encoding=3, text=text[:60]) |
| | audio["TPE1"] = TPE1(encoding=3, text=f"Voice {speakername} using Seed={seed}") |
| | audio["TPUB"] = TPUB(encoding=3, text="Bark by Suno AI") |
| | audio["COMMENT"] = COMM(encoding=3, text="Generated with Bark GUI - Text-Prompted Generative Audio Model. Visit https://github.com/C0untFloyd/bark-gui") |
| | audio.save() |
| |
|