ver 1.5

app.py

@@ -2,6 +2,7 @@ import os
 import re
 import io
 import torch
+import shutil
 import requests
 import torchaudio
 import numpy as np
@@ -207,10 +208,21 @@ def srt_to_audio_multi(srt_files):
                 output_paths.append(result)
     
     if invalid_files:
-        raise ValueError(f"Invalid SRT files: {', '.join(invalid_files)}")
+        raise gr.Warning(f"Invalid SRT files: {', '.join(invalid_files)}")
     
     return output_paths
 
+def download_all(outputs):
+    # If no outputs, return None
+    if not outputs:
+        raise gr.Warning("No files available for download.")
+    
+    zip_path = os.path.join(cache_dir, "all_outputs.zip")
+    with shutil.ZipFile(zip_path, 'w') as zipf:
+        for file_path in outputs:
+            zipf.write(file_path, os.path.basename(file_path))
+    return zip_path
+
 # Initialize model
 model = Model(
     model_name=default_model_name, 
@@ -229,12 +241,24 @@ with gr.Blocks(css=css) as demo:
         elem_id="title",
     )
     with gr.Row(elem_id="container"):
-        inp = gr.File(label="Upload SRT files", file_count="multiple", type="filepath")
-        out = gr.File(label="Generated Audio Files", file_count="multiple", type="filepath")
+        inp = gr.File(
+            label="Upload SRT files", 
+            file_count="multiple", 
+            type="filepath", 
+            height=600
+        )
+        out = gr.File(
+            label="Generated Audio Files",
+            file_count="multiple",
+            type="filepath",
+            height=600
+        )
     
     btn = gr.Button("Generate")
+    download_btn = gr.Button("Download All")
 
     btn.click(fn=srt_to_audio_multi, inputs=inp, outputs=out)
+    download_btn.click(fn=download_all, inputs=out, outputs="file")
 
 if __name__ == "__main__":
     demo.launch()