init

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 
 import sys
+import pandas as pd
 import os
 import argparse
 import random
@@ -517,6 +518,28 @@ def demo_update_w(mode):
         w_positions.append(gr.Slider())
     return w_positions
 
+def plot_update(*positions):
+    key_length = int(positions[-1])
+    frame_indices = positions[:key_length]
+    h_positions = positions[MAX_KEYS:MAX_KEYS+key_length]
+    h_positions_re = []
+    for i in h_positions:
+        h_positions_re.append(-i)
+    w_positions = positions[2*MAX_KEYS:2*MAX_KEYS+key_length]
+    traj_plot = gr.ScatterPlot(
+        value=pd.DataFrame({"x": w_positions, "y": h_positions_re, "frame": frame_indices}),
+        x="x",
+        y="y",
+        color='frame',
+        x_lim= [-0.05, 1.05],
+        y_lim= [-1.05, 0.05],
+        label="Trajectory",
+        width=512,
+        height=320,
+    )
+    return traj_plot
+
+
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown(
@@ -608,6 +631,22 @@ with gr.Blocks(css=css) as demo:
         dropdown_demo.change(demo_update_w, dropdown_demo, w_positions)
         radio_mode.change(mode_update, radio_mode, [row_demo, row_diy])
 
+        traj_plot = gr.ScatterPlot(
+            label="Trajectory",
+            width=512,
+            height=320,
+        )
+
+        h_positions[0].change(plot_update, frame_indices + h_positions + w_positions + [dropdown_diy], traj_plot)
+        h_positions[1].change(plot_update, frame_indices + h_positions + w_positions + [dropdown_diy], traj_plot)
+        h_positions[2].change(plot_update, frame_indices + h_positions + w_positions + [dropdown_diy], traj_plot)
+        h_positions[3].change(plot_update, frame_indices + h_positions + w_positions + [dropdown_diy], traj_plot)
+        h_positions[4].change(plot_update, frame_indices + h_positions + w_positions + [dropdown_diy], traj_plot)
+        w_positions[0].change(plot_update, frame_indices + h_positions + w_positions + [dropdown_diy], traj_plot)
+        w_positions[1].change(plot_update, frame_indices + h_positions + w_positions + [dropdown_diy], traj_plot)
+        w_positions[2].change(plot_update, frame_indices + h_positions + w_positions + [dropdown_diy], traj_plot)
+        w_positions[3].change(plot_update, frame_indices + h_positions + w_positions + [dropdown_diy], traj_plot)
+        w_positions[4].change(plot_update, frame_indices + h_positions + w_positions + [dropdown_diy], traj_plot)
 
         with gr.Row():
             with gr.Accordion('Useful FreeTraj Parameters (feel free to adjust these parameters based on your prompt): ', open=True):