feat: update fx parameter everytime a slider is moved

app.py

@@ -939,20 +939,22 @@ with gr.Blocks() as demo:
     random_button.click(
         chain_functions(
             lambda i: (torch.randn_like(mean).clip(SLIDER_MIN, SLIDER_MAX), i),
-            lambda args: (args[0], vec2fx(z2x(args[0])), args[1]),
-            lambda args: update_all(*args) + [args[0]],
+            lambda args: (args[0], z2x(args[0]), args[1]),
+            lambda args: [args[0], args[1], vec2fx(args[1]), args[2]],
+            lambda args: update_all(args[0], args[2], args[3]) + args[:2],
         ),
         inputs=extra_pc_dropdown,
-        outputs=update_all_outputs + [z],
+        outputs=update_all_outputs + [z, fx_params],
     )
     reset_button.click(
         # lambda: (lambda _: [0 for _ in range(NUMBER_OF_PCS + 1)])(z.zero_()),
         lambda: chain_functions(
             lambda _: torch.zeros_like(mean),
-            lambda z: (z, vec2fx(z2x(z))),
-            lambda args: update_all(args[0], args[1], NUMBER_OF_PCS) + [args[0]],
+            lambda z: (z, z2x(z)),
+            lambda args: [*args[:2], vec2fx(args[1])],
+            lambda args: update_all(args[0], args[2], NUMBER_OF_PCS) + args[:2],
         )(None),
-        outputs=update_all_outputs + [z],
+        outputs=update_all_outputs + [z, fx_params],
     )
 
     def update_z(z, s, i):
@@ -963,26 +965,34 @@ with gr.Blocks() as demo:
         slider.input(
             lambda *args, i=i: chain_functions(
                 lambda args: update_z(args[0], args[1], i),
-                lambda z: (z, vec2fx(z2x(z))),
-                lambda args: [args[0]]
-                + update_fx(args[1])
-                + update_plots(args[1])
-                + [model2json(args[1])],
+                lambda z: (z, z2x(z)),
+                lambda args: [args[0], args[1], vec2fx(args[1])],
+                lambda args: args[:2]
+                + update_fx(args[2])
+                + update_plots(args[2])
+                + [model2json(args[2])],
             )(args),
             inputs=[z, slider],
-            outputs=[z] + update_fx_outputs + update_plots_outputs + [json_output],
+            outputs=[z, fx_params]
+            + update_fx_outputs
+            + update_plots_outputs
+            + [json_output],
         )
     extra_slider.input(
         lambda *xs: chain_functions(
             lambda args: update_z(args[0], args[1], args[2]),
-            lambda z: (z, vec2fx(z2x(z))),
-            lambda args: [args[0]]
-            + update_fx(args[1])
-            + update_plots(args[1])
-            + [model2json(args[1])],
+            lambda z: (z, z2x(z)),
+            lambda args: [args[0], args[1], vec2fx(args[1])],
+            lambda args: args[:2]
+            + update_fx(args[2])
+            + update_plots(args[2])
+            + [model2json(args[2])],
         )(xs),
         inputs=[z, extra_slider, extra_pc_dropdown],
-        outputs=[z] + update_fx_outputs + update_plots_outputs + [json_output],
+        outputs=[z, fx_params]
+        + update_fx_outputs
+        + update_plots_outputs
+        + [json_output],
     )
 
     extra_pc_dropdown.input(