Update app.py

app.py

@@ -213,7 +213,6 @@ def run_music(
 
     for p in text_items:
         t0 = time.time()
-        # Параметры генерации держим минимальными и совместимыми
         out = pipe(p, forward_params={"do_sample": bool(do_sample)})
         dt = time.time() - t0
 
@@ -261,6 +260,17 @@ music_description_md = (
     """
 )
 
+def run_tts_ui(prompts_text, split_lines, model_choice):
+    files, _, last = run_tts(prompts_text, split_lines, model_choice)
+    samples_update = gr.update(choices=files, value=(last or (files[-1] if files else None)))
+    return files, (last or None), samples_update
+
+def run_music_ui(prompts_text, split_lines, model_name, do_sample):
+    files, _, last = run_music(prompts_text, split_lines, model_name, do_sample)
+    samples_update = gr.update(choices=files, value=(last or (files[-1] if files else None)))
+    return files, (last or None), samples_update
+
+
 with gr.Blocks(title="Speech & Music Bench") as demo:
     gr.Markdown("# 🎙️🪄 Speech & Music Bench")
 
@@ -285,16 +295,26 @@ with gr.Blocks(title="Speech & Music Bench") as demo:
         run_btn_tts = gr.Button("Сгенерировать речь", variant="primary")
 
         with gr.Row():
-            files_tts = gr.Files(label="Файлы .wav для скачивания")
+            files_tts = gr.Files(label="Файлы .wav для скачивания")  
         with gr.Row():
-            df_out_tts = gr.Dataframe(label="Таблица результатов", interactive=False)
+            samples_tts = gr.Dropdown(
+                label="Все сгенерённые семплы (выберите для прослушивания)",
+                choices=[],
+                allow_custom_value=False
+            )
         with gr.Row():
-            preview_tts = gr.Audio(label="Предпросмотр последнего семпла", autoplay=False)
+            preview_tts = gr.Audio(label="Предпросмотр выбранного семпла", autoplay=False)
 
         run_btn_tts.click(
-            fn=run_tts,
+            fn=run_tts_ui,
             inputs=[prompts_tts, split_lines_tts, model_choice],
-            outputs=[files_tts, df_out_tts, preview_tts],
+            outputs=[files_tts, preview_tts, samples_tts],
+        )
+
+        samples_tts.change(
+            fn=lambda p: gr.update(value=p),
+            inputs=samples_tts,
+            outputs=preview_tts,
         )
 
     with gr.Tab("🎵 Music"):
@@ -321,15 +341,26 @@ with gr.Blocks(title="Speech & Music Bench") as demo:
         with gr.Row():
             files_music = gr.Files(label="Файлы .wav для скачивания")
         with gr.Row():
-            df_out_music = gr.Dataframe(label="Таблица результатов", interactive=False)
+            samples_music = gr.Dropdown(
+                label="Все сгенерённые треки (выберите для прослушивания)",
+                choices=[],
+                allow_custom_value=False
+            )
         with gr.Row():
-            preview_music = gr.Audio(label="Предпросмотр последнего трека", autoplay=False)
+            preview_music = gr.Audio(label="Предпросмотр выбранного трека", autoplay=False)
 
         run_btn_music.click(
-            fn=run_music,
+            fn=run_music_ui,
             inputs=[prompts_music, split_lines_music, music_model, do_sample],
-            outputs=[files_music, df_out_music, preview_music],
+            outputs=[files_music, preview_music, samples_music],
         )
 
+        samples_music.change(
+            fn=lambda p: gr.update(value=p),
+            inputs=samples_music,
+            outputs=preview_music,
+        )
+
+
 if __name__ == "__main__":
     demo.launch()