fix examples

app.py

@@ -24,7 +24,31 @@ try:
     print("Available EGL libraries:", egl_libs)
 except Exception as e:
     print(f"Error finding libraries: {e}")
-    
+
+# Example videos
+example_videos = [
+    "./examples/000652_0_120.mp4",  # Replace with actual video paths
+    "./examples/000652_0_120.mp4",  # Replace with actual video paths
+    "./examples/000652_0_120.mp4",  # Replace with actual video paths
+    "./examples/000652_0_120.mp4",  # Replace with actual video paths
+]
+# Example videos
+example_keys = [
+    "000091",  # Replace with actual video paths
+    "000091",  # Replace with actual video paths
+    "000091",  # Replace with actual video paths
+    "000091",  # Replace with actual video paths
+]
+# Example videos
+example_texts = [
+    "need to use the opposite leg",  # Replace with actual video paths
+    "need to use the opposite leg2",  # Replace with actual video paths
+    "need to use the opposite leg3",  # Replace with actual video paths
+    "need to use the opposite leg4",  # Replace with actual video paths
+]
+example_video_outputs = [gr.Video(label=f"Example {i+1}", 
+                                  value=example_videos[i]) 
+                         for i in range(4)]
 
 class MotionEditor:
     def __init__(self):
@@ -226,8 +250,8 @@ def create_gradio_interface():
         random_key = random.choice(list(current_set.keys()))
         motion = current_set[random_key]['motion_a']
         text_annot = current_set[random_key]['annotation']
-        
-        return motion, text_annot, random_key, text_annot
+        return gr.update(value=motion, 
+                         visible=True), text_annot, random_key, text_annot
     
     def clear():
         return ""
@@ -243,6 +267,8 @@ def create_gradio_interface():
                 with gr.Row():
                     random_button = gr.Button("Random",  scale=0)
                     clear_button_retrieval = gr.Button("Clear", scale=0)
+            # Example videos grid with buttons
+
 
                 suggested_edit_text = gr.Textbox(
                     placeholder="Texts likely to edit the motion:",
@@ -259,8 +285,43 @@ def create_gradio_interface():
                 retrieved_video_output = gr.Video(
                     label="Retrieved Motion",
                     height=360,
-                    width=480
+                    width=480,
+                    visible=False  # Initially hidden
                 )
+                gr.Markdown("### Examples")
+                with gr.Row():
+                    # First example
+                    with gr.Column():
+                        gr.Video(value=example_videos[0], 
+                                 height=180,width=240,
+                                 label="Example 1")
+                        example_button1 = gr.Button("Select Example 1", 
+                                                    size='sm')
+                    
+                    # Second example
+                    with gr.Column(scale=2):
+                        gr.Video(value=example_videos[1], 
+                                height=180,width=240,
+                                 label="Example 2")
+                        example_button2 = gr.Button("Select Example 2", 
+                                                    size='sm',scale=1)
+                with gr.Row():
+
+                    # Third example
+                    with gr.Column():
+                        gr.Video(value=example_videos[2], 
+                                 height=180,width=240,
+                                 label="Example 3")
+                        example_button3 = gr.Button("Select Example 3",
+                                                    size='sm')
+                    
+                    # Fourth example
+                    with gr.Column():
+                        gr.Video(value=example_videos[3], 
+                                 height=180,width=240,
+                                 label="Example 4")
+                        example_button4 = gr.Button("Select Example 4",
+                                                    size='sm')
 
             with gr.Column(scale=5):
                 gr.HTML(WEB_target)
@@ -297,7 +358,60 @@ def create_gradio_interface():
                 input_text
             ]
         )
-        
+        # def load_example_video(example_path):
+        #     # motion = current_set[random_key]['motion_a']
+        #     # text_annot = current_set[random_key]['annotation']
+        #     import ipdb; ipdb.set_trace()
+        #     return gr.update(value=example_path, visible=True)
+        def load_example(example_video, example_key, example_text):
+            # Update all outputs
+            return (
+                gr.update(value=example_video, visible=True),  # Update video output
+                example_text,  # Update suggested edit text
+                example_key,  # Update random key state
+                example_text  # Update input text
+            )
+        example_button1.click(
+            fn=lambda: load_example(example_videos[0], example_keys[0], example_texts[0]),
+            inputs=None,
+            outputs=[
+                retrieved_video_output,
+                suggested_edit_text,
+                random_key_state,
+                input_text
+            ]
+        )
+        example_button2.click(
+            fn=lambda: load_example(example_videos[1], example_keys[1], example_texts[1]),
+            inputs=None,
+            outputs=[
+                retrieved_video_output,
+                suggested_edit_text,
+                random_key_state,
+                input_text
+            ]
+        )
+        example_button3.click(
+            fn=lambda: load_example(example_videos[2], example_keys[2], example_texts[2]),
+            inputs=None,
+            outputs=[
+                retrieved_video_output,
+                suggested_edit_text,
+                random_key_state,
+                input_text
+            ]
+        )
+        example_button4.click(
+            fn=lambda: load_example(example_videos[3], example_keys[3], example_texts[3]),
+            inputs=None,
+            outputs=[
+                retrieved_video_output,
+                suggested_edit_text,
+                random_key_state,
+                input_text
+            ]
+        )
+ 
         clear_button_edit.click(clear, outputs=input_text)
         clear_button_retrieval.click(clear, outputs=suggested_edit_text)