add example

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 
 import sys
+import random
 import pandas as pd
 import os
 import argparse
@@ -519,7 +520,14 @@ def demo_update_w(mode):
     return w_positions
 
 def plot_update(*positions):
-    key_length = int(positions[-1])
+    if type(positions[-1]) != int:
+        traj_plot = gr.ScatterPlot(
+            label="Trajectory",
+            width=512,
+            height=320,
+        )
+        return traj_plot
+    key_length = positions[-1]
     frame_indices = positions[:key_length]
     h_positions = positions[MAX_KEYS:MAX_KEYS+key_length]
     h_positions_re = []
@@ -565,127 +573,128 @@ with gr.Blocks(css=css) as demo:
             video_result = gr.Video(label="Video Output")
             video_result_bbox = gr.Video(label="Video Output with BBox")
 
-        with gr.Row():
-            prompt_in = gr.Textbox(label="Prompt", placeholder="A corgi running on the grassland on the grassland.", scale = 3)
-            target_indices = gr.Textbox(label="Target Indices", placeholder="1,2", scale = 1)
-
-        with gr.Row():
-            radio_mode = gr.Radio(label='Trajectory Mode', choices = ['demo', 'diy', 'ori'], scale = 1)
-            height_ratio = gr.Slider(label='Height Ratio of BBox',
-                              minimum=0.2,
-                              maximum=0.4,
-                              step=0.01,
-                              value=0.3,
-                              scale = 1)
-            width_ratio = gr.Slider(label='Width Ratio of BBox',
-                              minimum=0.2,
-                              maximum=0.4,
-                              step=0.01,
-                              value=0.3, 
-                              scale = 1)
-          
-        with gr.Row(visible=False) as row_demo:
-            dropdown_demo = gr.Dropdown(
-                label="Demo Trajectory",
-                choices= ['topleft->bottomright', 'bottomleft->topright', 'topleft->bottomleft->bottomright', 'bottomright->topright->topleft', '"V"', '"^"', 'left->right->left->right', 'triangle']
-            )
-            
-        with gr.Row(visible=False) as row_diy:
-            dropdown_diy = gr.Dropdown(
-                label="Number of keyframes",
-                choices=range(2, MAX_KEYS+1),
-            )
+        with gr.Group():
+            with gr.Row():
+                prompt_in = gr.Textbox(label="Prompt", placeholder="A corgi running on the grassland on the grassland.", scale = 3)
+                target_indices = gr.Textbox(label="Target Indices", placeholder="1,2", scale = 1)
+
+            with gr.Row():
+                radio_mode = gr.Radio(label='Trajectory Mode', choices = ['demo', 'diy', 'ori'], scale = 1)
+                height_ratio = gr.Slider(label='Height Ratio of BBox',
+                                minimum=0.2,
+                                maximum=0.4,
+                                step=0.01,
+                                value=0.3,
+                                scale = 1)
+                width_ratio = gr.Slider(label='Width Ratio of BBox',
+                                minimum=0.2,
+                                maximum=0.4,
+                                step=0.01,
+                                value=0.3, 
+                                scale = 1)
             
-        for i in range(MAX_KEYS):
-            with gr.Row(visible=False) as row:
-                text = f"Keyframe #{i}"
-                text = gr.HTML(text, visible=True)
-                frame_ids = gr.Textbox(
-                    None,
-                    label=f"Frame Indices #{i}",
-                    visible=True,
-                    interactive=True,
-                    scale=1
+            with gr.Row(visible=False) as row_demo:
+                dropdown_demo = gr.Dropdown(
+                    label="Demo Trajectory",
+                    choices= ['topleft->bottomright', 'bottomleft->topright', 'topleft->bottomleft->bottomright', 'bottomright->topright->topleft', '"V"', '"^"', 'left->right->left->right', 'triangle']
                 )
-                h_position = gr.Slider(label='Position in Height',
-                    minimum=0.0,
-                    maximum=1.0,
-                    step=0.01,
-                    scale=1)
-                w_position = gr.Slider(label='Position in Width',
-                    minimum=0.0,
-                    maximum=1.0,
-                    step=0.01,
-                    scale=1)
                 
-            frame_indices.append(frame_ids)
-            h_positions.append(h_position)
-            w_positions.append(w_position)
-            keyframes.append(row)
-
-
-        dropdown_demo.change(demo_update, dropdown_demo, dropdown_diy)
-        dropdown_diy.change(keyframe_update, dropdown_diy, keyframes)
-        dropdown_demo.change(demo_update_frame, dropdown_demo, frame_indices)
-        dropdown_demo.change(demo_update_h, dropdown_demo, h_positions)
-        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):
-                with gr.Row():
-                    ddim_edit = gr.Slider(label='Editing Steps (larger for better control while losing some quality)',
-                             minimum=0,
-                             maximum=12,
-                             step=1,
-                             value=6)
-                    seed = gr.Slider(label='Random Seed',
-                             minimum=0,
-                             maximum=10000,
-                             step=1,
-                             value=123)
+            with gr.Row(visible=False) as row_diy:
+                dropdown_diy = gr.Dropdown(
+                    label="Number of keyframes",
+                    choices=range(2, MAX_KEYS+1),
+                )
+                
+            for i in range(MAX_KEYS):
+                with gr.Row(visible=False) as row:
+                    text = f"Keyframe #{i}"
+                    text = gr.HTML(text, visible=True)
+                    frame_ids = gr.Textbox(
+                        None,
+                        label=f"Frame Indices #{i}",
+                        visible=True,
+                        interactive=True,
+                        scale=1
+                    )
+                    h_position = gr.Slider(label='Position in Height',
+                        minimum=0.0,
+                        maximum=1.0,
+                        step=0.01,
+                        scale=1)
+                    w_position = gr.Slider(label='Position in Width',
+                        minimum=0.0,
+                        maximum=1.0,
+                        step=0.01,
+                        scale=1)
                     
-        with gr.Row():
-            with gr.Accordion('Useless FreeTraj Parameters (mostly no need to adjust): ', open=False):
-                with gr.Row():
-                    ddim_steps = gr.Slider(label='DDIM Steps',
-                              minimum=5,
-                              maximum=200,
-                              step=1,
-                              value=50)
-                    unconditional_guidance_scale = gr.Slider(label='Unconditional Guidance Scale',
-                              minimum=1.0,
-                              maximum=20.0,
-                              step=0.1,
-                              value=12.0)
-                with gr.Row():
-                    video_fps = gr.Slider(label='Video FPS (larger for quicker motion)',
-                              minimum=8,
-                              maximum=36,
-                              step=4,
-                              value=16)
-                    save_fps = gr.Slider(label='Save FPS',
-                              minimum=1,
-                              maximum=30,
-                              step=1,
-                              value=10)
+                frame_indices.append(frame_ids)
+                h_positions.append(h_position)
+                w_positions.append(w_position)
+                keyframes.append(row)
+
+
+            dropdown_demo.change(demo_update, dropdown_demo, dropdown_diy)
+            dropdown_diy.change(keyframe_update, dropdown_diy, keyframes)
+            dropdown_demo.change(demo_update_frame, dropdown_demo, frame_indices)
+            dropdown_demo.change(demo_update_h, dropdown_demo, h_positions)
+            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):
+                    with gr.Row():
+                        ddim_edit = gr.Slider(label='Editing Steps (larger for better control while losing some quality)',
+                                minimum=0,
+                                maximum=12,
+                                step=1,
+                                value=6)
+                        seed = gr.Slider(label='Random Seed',
+                                minimum=0,
+                                maximum=10000,
+                                step=1,
+                                value=123)
+                        
+            with gr.Row():
+                with gr.Accordion('Useless FreeTraj Parameters (mostly no need to adjust): ', open=False):
+                    with gr.Row():
+                        ddim_steps = gr.Slider(label='DDIM Steps',
+                                minimum=5,
+                                maximum=200,
+                                step=1,
+                                value=50)
+                        unconditional_guidance_scale = gr.Slider(label='Unconditional Guidance Scale',
+                                minimum=1.0,
+                                maximum=20.0,
+                                step=0.1,
+                                value=12.0)
+                    with gr.Row():
+                        video_fps = gr.Slider(label='Video FPS (larger for quicker motion)',
+                                minimum=8,
+                                maximum=36,
+                                step=4,
+                                value=16)
+                        save_fps = gr.Slider(label='Save FPS',
+                                minimum=1,
+                                maximum=30,
+                                step=1,
+                                value=10)
                 
         with gr.Row():
             submit_btn = gr.Button("Generate", variant='primary')
@@ -693,6 +702,12 @@ with gr.Blocks(css=css) as demo:
         with gr.Row():
             gr.Examples(label='Sample Prompts', examples=examples, 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])
 
+        demo_list = ['0026_0_0.4_0.4.gif', '0047_1_0.4_0.3.gif', '0051_1_0.4_0.4.gif']
+        demo_pick = random.randint(0, len(demo_list) - 1)
+        with gr.Row():
+            for i in range(len(demo_list)):
+                gr.Image(show_label = False, show_download_button = False, value='assets/' + demo_list[i])
+
         with gr.Row():
             gr.Markdown(
                 """