Add model2json function to serialize effect settings as JSON

app.py

@@ -140,6 +140,29 @@ def get_important_pcs(n=10, **kwargs):
     return sliders
 
 
+def model2json():
+    fx_names = ["PK1", "PK2", "LS", "HS", "LP", "HP", "DRC"]
+    results = {k: v.toJSON() for k, v in zip(fx_names, fx)} | {
+        "Panner": fx[7].pan.toJSON()
+    }
+    spatial_fx = {
+        "DLY": fx[7].effects[0].toJSON() | {"LP": fx[7].effects[0].eq.toJSON()},
+        "FDN": fx[7].effects[1].toJSON()
+        | {
+            "Tone correction PEQ": {
+                k: v.toJSON() for k, v in zip(fx_names[:4], fx[7].effects[1].eq)
+            }
+        },
+        "Cross Send (dB)": fx[7].params.sends_0.log10().mul(20).item(),
+    }
+    return json.dumps(
+        {
+            "Direct": results,
+            "Sends": spatial_fx,
+        }
+    )
+
+
 with gr.Blocks() as demo:
     gr.Markdown(
         title_md,
@@ -177,15 +200,16 @@ with gr.Blocks() as demo:
             audio_output = gr.Audio(
                 type="numpy", label="Output Audio", interactive=False
             )
+            json_output = gr.JSON(label="Effect Settings", max_height=800, open=True)
 
     render_button.click(
-        inference,
+        lambda *args: (lambda x: (x, model2json()))(inference(*args)),
         inputs=[
             audio_input,
             random_rest_checkbox,
         ]
         + sliders,
-        outputs=audio_output,
+        outputs=[audio_output, json_output],
     )
 
     random_button.click(