Front and back launch ok

app.py

@@ -16,23 +16,20 @@ def generate_audio(descriptions: str):
     if model is None:
         model = AudioGen.get_pretrained('facebook/audiogen-medium')
         model.set_generation_params(duration=10) 
-
-    safe_name = "_".join(descriptions.split())
-    output_path = os.path.join(OUTPUT_DIR, safe_name)
-
+    if not os.path.exists('audio_files'):
+        os.makedirs('audio_files')
+    
     wav = model.generate([descriptions]) 
-    audio_write(
-        output_path,
-        wav[0].cpu(),
-        model.sample_rate,
-        strategy="loudness",
-        loudness_compressor=True,
-        add_suffix=False,
-    )
-
-    final_path = f"{output_path}.wav"
-    print(f"Generated audio for '{descriptions}' -> {final_path}")
-    return final_path
+    results = []
+
+    for idx, one_wav in enumerate(wav):
+        filename = f'{descriptions}.wav'
+        file_path = os.path.join('audio_files', filename)
+        audio_write(file_path, one_wav.cpu(), model.sample_rate, strategy="loudness", loudness_compressor=True, add_suffix=False)
+        print(f"Generated audio for '{descriptions}'")
+        results.append(file_path)
+    
+    return results[0]
 
 with gr.Blocks() as demo:
     gr.Markdown("# AudioGen Demo")