From 944e6ffce33e7e20ebeac50b626e49d718aa28c3 Mon Sep 17 00:00:00 2001 From: drbaph <84208527+Saganaki22@users.noreply.github.com> Date: Mon, 10 Jun 2024 18:02:56 +0100 Subject: [PATCH] Update gradio_app.py Update 0.4 Check README.md --- gradio_app.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/gradio_app.py b/gradio_app.py index ee34a65..ac9f558 100644 --- a/gradio_app.py +++ b/gradio_app.py @@ -66,7 +66,12 @@ def generate_audio(prompt, steps, cfg_scale, sigma_min, sigma_max, generation_ti save_path = os.path.join(output_folder, date_folder) os.makedirs(save_path, exist_ok=True) - # Generate a filename based on the prompt + # Set a maximum filename length (e.g., 50 characters) + max_length = 50 + if len(prompt) > max_length: + prompt = prompt[:max_length] + "_truncated" + + # Sanitize the prompt to create a safe filename filename = re.sub(r'\W+', '_', prompt) + ".mp3" full_path = os.path.join(save_path, filename)