mrleo1nid/ACE-Step
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add auto outputs folder creation in components.py

Browse Source 
Add the functionality to automatically create the outputs folder if it doesn't exist at line 99
This commit is contained in:
Samrat Barai
2025-05-13 16:35:12 +06:00
committed by GitHub
parent 777229b86c
commit e2614cd6ac
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
acestep/ui/components.py
+1
View File
@@ -96,6 +96,7 @@ def create_text2music_ui(
with gr.Row(equal_height=True): with gr.Row(equal_height=True):
curr_file_dir = os.path.dirname(__file__) curr_file_dir = os.path.dirname(__file__)
output_file_dir = os.path.join(curr_file_dir, "..", "..", "outputs") output_file_dir = os.path.join(curr_file_dir, "..", "..", "outputs")
if not os.path.isdir(output_file_dir): os.makedirs(output_file_dir)
json_files = [f for f in os.listdir(output_file_dir) if f.endswith('.json')] json_files = [f for f in os.listdir(output_file_dir) if f.endswith('.json')]
json_files.sort(reverse=True, key=lambda x: int(x.split('_')[1])) json_files.sort(reverse=True, key=lambda x: int(x.split('_')[1]))
output_files = gr.Dropdown(choices=json_files, label="Select previous generated input params", scale=9, interactive=True) output_files = gr.Dropdown(choices=json_files, label="Select previous generated input params", scale=9, interactive=True)