Update app.py

app.py

@@ -94,20 +94,38 @@ async def create_demo():
                 voice_dropdown = gr.Dropdown(choices=[""] + list(voices.keys()), label="Select Voice", value="")
                 rate_slider = gr.Slider(minimum=-50, maximum=50, value=0, label="Rate Adjustment (%)", step=1)
                 pitch_slider = gr.Slider(minimum=-20, maximum=20, value=0, label="Pitch Adjustment (Hz)", step=1)
+                
+                generate_button = gr.Button("Generate Audio and Subtitles", variant="primary")
 
             with gr.Column():
-                generate_button = gr.Button("Generate Audio and Subtitles", variant="primary")
+                output_audio = gr.Audio(label="Generated Audio", type="filepath")
+                audio_download_button = gr.Button("Download Audio")
 
-        output_audio = gr.Audio(label="Generated Audio", type="filepath")
-        output_srt = gr.File(label="Generated SRT", file_count="single")
-        warning_msg = gr.Markdown(label="Warning", visible=False)
+                output_srt = gr.File(label="Generated SRT", file_count="single")
+                srt_download_button = gr.Button("Download SRT")
+                
+                warning_msg = gr.Markdown(label="Warning", visible=False)
 
         generate_button.click(
             fn=tts_interface,
             inputs=[text_input, voice_dropdown, rate_slider, pitch_slider],
             outputs=[output_audio, output_srt, warning_msg]
         )
-    
+        
+        audio_download_button.click(
+            fn=lambda audio: audio,
+            inputs=output_audio,
+            outputs=None,
+            _js="(function(audio) { if(audio) { window.open(audio); } })"
+        )
+
+        srt_download_button.click(
+            fn=lambda srt: srt,
+            inputs=output_srt,
+            outputs=None,
+            _js="(function(srt) { if(srt) { window.open(srt); } })"
+        )
+
     return demo
 
 # 运行应用