add example

app.py

@@ -1,8 +1,10 @@
-import gradio as gr
-
 import sys
+import gradio as gr
 import random
 import pandas as pd
+import numpy as np
+import matplotlib.pyplot as plt
+
 import os
 import argparse
 import random
@@ -19,6 +21,7 @@ from funcs import (
     load_model_checkpoint,
 )
 from utils.utils import instantiate_from_config
+from utils.utils_freetraj import plan_path
 
 MAX_KEYS = 5
 
@@ -521,29 +524,25 @@ def demo_update_w(mode):
 
 def plot_update(*positions):
     if type(positions[-1]) != int:
-        traj_plot = gr.ScatterPlot(
-            label="Trajectory",
-            width=512,
-            height=320,
+        traj_plot = gr.Plot(
+            label="Trajectory"
         )
         return traj_plot
     key_length = positions[-1]
     frame_indices = positions[:key_length]
+    frame_indices = [int(i) for i in frame_indices]
     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],
+    frame_indices, h_positions, w_positions = zip(*sorted(zip(frame_indices, h_positions, w_positions)))
+    plt.cla()
+    plt.xlim(0, 1)
+    plt.ylim(0, 1)
+    plt.gca().invert_yaxis()
+    plt.gca().xaxis.tick_top()
+    plt.plot(w_positions, h_positions, linestyle='-', marker = 'o', markerfacecolor='r')
+    traj_plot = gr.Plot(
         label="Trajectory",
-        width=512,
-        height=320,
+        value = plt
     )
     return traj_plot
 
@@ -640,10 +639,8 @@ 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,
+            traj_plot = gr.Plot(
+                label="Trajectory"
             )
 
             h_positions[0].change(plot_update, frame_indices + h_positions + w_positions + [dropdown_diy], traj_plot)
@@ -731,6 +728,6 @@ with gr.Blocks(css=css) as demo:
     submit_btn.click(fn=infer,
             inputs=[prompt_in, target_indices, ddim_edit, seed, ddim_steps, unconditional_guidance_scale, video_fps, save_fps, height_ratio, width_ratio, radio_mode, dropdown_diy, *frame_indices, *h_positions, *w_positions],
             outputs=[video_result, video_result_bbox],
-            api_name="zrscp")
+            api_name="freetraj")
 
-demo.queue(max_size=12).launch(show_api=True)
+demo.queue(max_size=8).launch(show_api=True)