.wav or .mp3

app.py

@@ -82,20 +82,28 @@ tango.model.to(device_type)
 
 def check(
     prompt,
+    output_format,
     output_number,
     steps,
     guidance
 ):
     if prompt is None or prompt == "":
         raise gr.Error("Please provide a prompt input.")
+    if not output_format in ["wav", "mp3"]:
+        raise gr.Error("Please choose an allowed output format (.wav or .mp3).")
     if not output_number in [1, 2, 3]:
         raise gr.Error("Please ask for 1, 2 or 3 output files.")
 
-def update_display(output_number):
-    return [gr.update(visible = (2 <= output_number)), gr.update(visible = (output_number == 3))]
+def update_output(output_format, output_number):
+    return [
+        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(
     prompt,
+    output_format,
     output_number,
     steps,
     guidance
@@ -141,30 +149,34 @@ with gr.Blocks() as interface:
         """
     )
     input_text = gr.Textbox(label = "Prompt", value = "Snort of a horse", lines = 2, autofocus = True)
+    output_format = gr.Radio(label = "Output format", info = "The file you can dowload", ["mp3", "wav"], value = "mp3")
     with gr.Accordion("Advanced options", open = False):
         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)
 
-    submit = gr.Button("Generate 🚀", variant = "primary")
+    submit = gr.Button("🚀 Generate", variant = "primary")
 
-    output_audio_1 = gr.Audio(label = "Generated Audio #1/3")
-    output_audio_2 = gr.Audio(label = "Generated Audio #2/3")
-    output_audio_3 = gr.Audio(label = "Generated Audio #3/3")
+    output_audio_1 = gr.Audio(label = "Generated Audio #1/3", format = "mp3", autoplay = True)
+    output_audio_2 = gr.Audio(label = "Generated Audio #2/3", format = "mp3")
+    output_audio_3 = gr.Audio(label = "Generated Audio #3/3", format = "mp3")
     information = gr.Label(label = "Information")
 
     submit.click(fn = check, inputs = [
         input_text,
+        output_format,
         output_number,
         denoising_steps,
         guidance_scale
-    ], outputs = [], queue = False, show_progress = False).success(fn = update_display, inputs = [
+    ], outputs = [], queue = False, show_progress = False).success(fn = update_output, inputs = [
         output_number
     ], outputs = [
+        output_audio_1,
         output_audio_2,
         output_audio_3
     ], queue = False, show_progress = False).success(fn = text2audio, inputs = [
         input_text,
+        output_format,
         output_number,
         denoising_steps,
         guidance_scale
@@ -179,6 +191,7 @@ with gr.Blocks() as interface:
         fn = text2audio,
 	    inputs = [
             input_text,
+            output_format,
             output_number,
             denoising_steps,
             guidance_scale
@@ -190,11 +203,11 @@ with gr.Blocks() as interface:
             information
         ],
         examples = [
-                ["A hammer is hitting a wooden surface", 3, 100, 3],
-                ["Peaceful and calming ambient music with singing bowl and other instruments.", 3, 100, 3],
-                ["A man is speaking in a small room.", 3, 100, 3],
-                ["A female is speaking followed by footstep sound", 3, 100, 3],
-                ["Wooden table tapping sound followed by water pouring sound.", 3, 100, 3],
+                ["A hammer is hitting a wooden surface", "mp3", 3, 100, 3],
+                ["Peaceful and calming ambient music with singing bowl and other instruments.", "wav", 3, 100, 3],
+                ["A man is speaking in a small room.", "mp3", 2, 100, 3],
+                ["A female is speaking followed by footstep sound", "mp3", 1, 100, 3],
+                ["Wooden table tapping sound followed by water pouring sound.", "mp3", 3, 200, 3],
             ],
         cache_examples = "lazy",
     )