Format choice

app.py

@@ -92,10 +92,11 @@ def check(
     if not output_number in [1, 2, 3]:
         raise gr.Error("Please ask for 1, 2 or 3 output files.")
 
-def update_output(output_number):
+def update_output(output_format, output_number):
     return [
-        gr.update(visible = (2 <= output_number)),
-        gr.update(visible = (output_number == 3))
+        gr.update(format = output_format),
+        gr.update(format = output_format, visible = (2 <= output_number)),
+        gr.update(format = output_format, visible = (output_number == 3))
     ]
 
 def text2audio(
@@ -147,6 +148,7 @@ with gr.Blocks() as interface:
     )
     input_text = gr.Textbox(label = "Prompt", value = "Snort of a horse", lines = 2, autofocus = True)
     with gr.Accordion("Advanced options", open = False):
+        output_format = gr.Radio(label = "Output format", info = "The file you can dowload", choices = ["mp3", "wav"], value = "wav")
         output_number = gr.Slider(label = "Number of generations", info = "1, 2 or 3 output files", minimum = 1, maximum = 3, value = 3, step = 1, interactive = True)
         denoising_steps = gr.Slider(label = "Steps", info = "lower=faster & variant, higher=audio quality & similar", minimum = 100, maximum = 200, value = 100, step = 1, interactive = True)
         guidance_scale = gr.Slider(label = "Guidance Scale", info = "lower=audio quality, higher=follow the prompt", minimum = 1, maximum = 10, value = 3, step = 0.1, interactive = True)
@@ -164,8 +166,10 @@ with gr.Blocks() as interface:
         denoising_steps,
         guidance_scale
     ], outputs = [], queue = False, show_progress = False).success(fn = update_output, inputs = [
+        output_format,
         output_number
     ], outputs = [
+        output_audio_1,
         output_audio_2,
         output_audio_3
     ], queue = False, show_progress = False).success(fn = text2audio, inputs = [