feat: visualisation

app.py

@@ -155,6 +155,7 @@ to_fx_state_dict = lambda x: {
     k: v[0] if ndim_dict[k] == 0 else v for k, v in vec2dict(x).items()
 }
 
+
 meter = pyln.Meter(44100)
 
 
@@ -224,6 +225,9 @@ def inference(
     lr,
     progress=gr.Progress(track_tqdm=True),
 ):
+    # close all figures to avoid too many open figures
+    plt.close("all")
+
     device = Path("DEVICE.txt").read_text()
     if method == "Mean":
         return gaussian_params_dict[dataset][0].to(device)
@@ -474,7 +478,7 @@ def plot_t60(fx):
 def vec2fx(x):
     fx = deepcopy(global_fx)
     fx.load_state_dict(vec2dict(x), strict=False)
-    fx.apply(partial(clip_delay_eq_Q, Q=0.707))
+    # fx.apply(partial(clip_delay_eq_Q, Q=0.707))
     return fx
 
 
@@ -602,20 +606,57 @@ with gr.Blocks() as demo:
             interactive=True,
         )
 
+    _ = gr.Markdown("## Effect Parameters Visualisation")
+    with gr.Row():
+        peq_plot = gr.Plot(
+            plot_eq(global_fx), label="PEQ Frequency Response", elem_id="peq-plot"
+        )
+        comp_plot = gr.Plot(
+            plot_comp(global_fx), label="Compressor Curve", elem_id="comp-plot"
+        )
+
+    with gr.Row():
+        delay_plot = gr.Plot(
+            plot_delay(global_fx),
+            label="Delay Frequency Response",
+            elem_id="delay-plot",
+        )
+        reverb_plot = gr.Plot(
+            plot_reverb(global_fx),
+            label="Tone Correction PEQ",
+            elem_id="reverb-plot",
+            min_width=160,
+        )
+        t60_plot = gr.Plot(
+            plot_t60(global_fx), label="Decay Time", elem_id="t60-plot", min_width=160
+        )
+
+    _ = gr.Markdown("## Effect Settings JSON")
+    with gr.Row():
+        json_output = gr.JSON(
+            model2json(global_fx), label="Effect Settings", max_height=800, open=True
+        )
+
     process_button.click(
         chain_functions(
-            # lambda audio, ratio, x, *all_s: (
-            #     audio,
-            #     ratio,
-            #     # assign_fx_params(vec2fx(x), *all_s),
-            # ),
             lambda audio, approx, ratio, *args: (
                 audio,
                 approx,
                 ratio,
                 inference(audio, *args),
             ),
-            lambda audio, approx, ratio, vec: (*render(audio, approx, ratio, vec), vec),
+            lambda audio, approx, ratio, vec: (
+                vec2fx(vec),
+                *render(audio, approx, ratio, vec),
+                vec,
+            ),
+            lambda fx, *args: (
+                *args,
+                *map(
+                    lambda f: f(fx),
+                    [plot_eq, plot_comp, plot_delay, plot_reverb, plot_t60, model2json],
+                ),
+            ),
         ),
         inputs=[
             audio_input,
@@ -637,6 +678,12 @@ with gr.Blocks() as demo:
             direct_output,
             wet_output,
             fx_params,
+            peq_plot,
+            comp_plot,
+            delay_plot,
+            reverb_plot,
+            t60_plot,
+            json_output,
         ],
     )