ninth commit

app.py

@@ -169,16 +169,23 @@ def main(input, mix_coefficient):
         args, our_model, device, track_audio, track_name, meter, augmented_gain
     )
 
-    # if args.save_mixed_output:
     orig_audio = int16_to_float32(orig_audio)
     track_lufs = meter.integrated_loudness(orig_audio.T)
     augmented_gain = args.save_output_loudnorm - track_lufs
     orig_audio = orig_audio * db2linear(augmented_gain, eps=0.0)
 
+    prl_mix_out = orig_audio.T * mix_coefficient + estimates.T * (1 - mix_coefficient)
+    prl_mix_out = prl_mix_out * 32768
+    prl_mix_out = prl_mix_out.astype(np.int16)
+    estimates = estimates.T * 32768
+    estimates = estimates.astype(np.int16)
+    orig_audio = orig_audio.T * 32768
+    orig_audio = orig_audio.astype(np.int16)
+
     return (
-        (sr, estimates.T),
-        (sr, orig_audio.T),
-        (sr, orig_audio.T * mix_coefficient + estimates.T * (1 - mix_coefficient)),
+        (sr, estimates),
+        (sr, orig_audio),
+        (sr, prl_mix_out),
     )
 
 
@@ -214,12 +221,15 @@ with gr.Blocks() as demo:
                 btn = gr.Button("De-limit")
         with gr.Column():
             with gr.Box():
-                loud_norm_input = gr.Audio(label="Loudness Normalized Input (-14LUFS)")
+                loud_norm_input = gr.Audio(label="Loudness Normalized Input (-14LUFS)",
+                                           show_download_button=True)
             with gr.Box():
-                output_audio = gr.Audio(label="De-limiter Output")
+                output_audio = gr.Audio(label="De-limiter Output",
+                                        show_download_button=True,)
             with gr.Box():
                 output_audio_parallel = gr.Audio(
-                    label="Parallel Mix of the Input and its De-limiter Output"
+                    label="Parallel Mix of the Input and its De-limiter Output",
+                    show_download_button=True,
                 )
                 slider = gr.Slider(
                     minimum=0,
@@ -268,16 +278,6 @@ with gr.Blocks() as demo:
                     ],
                     outputs=plot,
                 )
-                # slider.release(
-                #     waveform_plot,
-                #     inputs=[
-                #         loud_norm_input,
-                #         output_audio,
-                #         output_audio_parallel,
-                #         slider_plot_x,
-                #         slider_plot_y,
-                #     ],
-                #     outputs=plot,
-                # )
+                
 if __name__ == "__main__":
     demo.launch(debug=True)