Display and information

app.py

@@ -90,24 +90,40 @@ def check(
     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 = True), gr.update(visible = (2 <= output_number)), gr.update(visible = (output_number == 3))]
+
 def text2audio(
     prompt,
     output_number,
     steps,
     guidance
 ):
+    start = time.time()
     output_wave = tango.generate(prompt, steps, guidance, output_number)
     output_wave_1 = gr.make_waveform((16000, output_wave[0]))
     output_wave_2 = gr.make_waveform((16000, output_wave[1])) if (2 <= output_number) else None
     output_wave_3 = gr.make_waveform((16000, output_wave[2])) if (output_number == 3) else None
-    return [output_wave_1, output_wave_2, output_wave_3]
+
+    end = time.time()
+    secondes = int(end - start)
+    minutes = secondes // 60
+    secondes = secondes - (minutes * 60)
+    hours = minutes // 60
+    minutes = minutes - (hours * 60)
+    return [
+        output_wave_1,
+        output_wave_2,
+        output_wave_3,
+        "Start again to get a different result. The output have been generated in " + str(hours) + " h, " + str(minutes) + " min, " + str(secondes) + " sec."
+    ]
 
 # Gradio interface
 with gr.Blocks() as interface:
     gr.Markdown("""
         <p style="text-align: center;">
         <b><big><big><big>Text-to-Audio</big></big></big></b>
-        <br/>Generates 10 second audio file, freely, without account, without watermark, that you can download.
+        <br/>Generates 10 second of sound effect from description, freely, without account, without watermark, that you can download.
         </p>
         <br/>
         <br/>
@@ -134,13 +150,20 @@ with gr.Blocks() as interface:
     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")
+    information = gr.Label(label = "Information")
 
     submit.click(fn = check, inputs = [
         input_text,
         output_number,
         denoising_steps,
         guidance_scale
-    ], outputs = [], queue = False, show_progress = False).success(fn = text2audio, inputs = [
+    ], outputs = [], queue = False, show_progress = False).success(fn = update_display, 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_number,
         denoising_steps,
@@ -148,7 +171,8 @@ with gr.Blocks() as interface:
     ], outputs = [
         output_audio_1,
         output_audio_2,
-        output_audio_3
+        output_audio_3,
+        information
     ], scroll_to_output = True)
 
     gr.Examples(
@@ -162,7 +186,8 @@ with gr.Blocks() as interface:
 	    outputs = [
             output_audio_1,
             output_audio_2,
-            output_audio_3
+            output_audio_3,
+            information
         ],
         examples = [
                 ["A hammer is hitting a wooden surface", 1, 100, 3],