Update app.py

app.py

@@ -79,6 +79,25 @@ css = """
         }
 """
 
+download_audio_js = """
+() =>{{
+    let root = document.querySelector("body > gradio-app");
+    if (root.shadowRoot != null)
+        root = root.shadowRoot;
+    let audio = root.querySelector("#{audio_id}").querySelector("audio");
+    if (audio == undefined)
+        return;
+    audio = audio.src;
+    let oA = document.createElement("a");
+    oA.download = Math.floor(Math.random()*100000000)+'.wav';
+    oA.href = audio;
+    document.body.appendChild(oA);
+    oA.click();
+    oA.remove();
+}}
+"""
+
+
 if __name__ == '__main__':
     models_tts = []
     name = '砂狼シロコ(ブルアカ) TTS'
@@ -132,6 +151,8 @@ if __name__ == '__main__':
                 tts_submit = gr.Button("Generate", variant="primary")
                 tts_output1 = gr.Textbox(label="Output Message")
                 tts_output2 = gr.Audio(label="Output Audio")
+                download = gr.Button("Download Audio")
+                download.click(None, [], [], _js=download_audio_js.format(audio_id=f"tts-audio{i}"))
                 tts_submit.click(tts_fn, [tts_input1, tts_input2, tts_input3, phoneme_input],
                                     [tts_output1, tts_output2])
                 to_phoneme_btn.click(to_phoneme_fn, [tts_input1], [tts_input1])