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

Improve Docker output paths and non-Docker compatibility

Browse Source
This commit is contained in:
dannyzen
2025-05-14 15:19:33 -04:00
parent 39043aaa61
commit f95c91f072
2 changed files with 13 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
acestep/ui/components.py
+7 -2
View File
@@ -94,8 +94,13 @@ def create_text2music_ui(
):
with gr.Row(equal_height=True):
curr_file_dir = os.path.dirname(__file__)
output_file_dir = os.path.join(curr_file_dir, "..", "..", "outputs")
# curr_file_dir = os.path.dirname(__file__) # Original path logic
# output_file_dir = os.path.join(curr_file_dir, "..", "..", "outputs") # Original path logic
# Get base output directory from environment variable, defaulting to CWD-relative 'outputs'.
# This default (./outputs) is suitable for non-Docker local development.
# For Docker, the ACE_OUTPUT_DIR environment variable should be set (e.g., to /app/outputs).
_output_base_dir = os.environ.get("ACE_OUTPUT_DIR", "./outputs")
output_file_dir = os.path.join(_output_base_dir, "text2music")
if not os.path.isdir(output_file_dir):
os.makedirs(output_file_dir, exist_ok=True)
json_files = [f for f in os.listdir(output_file_dir) if f.endswith('.json')]