Update app.py

app.py

@@ -91,24 +91,44 @@ def get_random_video():
     video_path2 = get_video_path_local(model_name_2, type, prompt)
     return video_path1, video_path2, model_name_1, model_name_2, type, prompt
 
+def update_prompt_options(type):
+    return gr.Dropdown.update(choices=dimension[type])
+
+
+def display_videos(type, prompt, model_name_1, model_name_2):
+    video_path1 = get_video_path_local(model_name_1, type, prompt)
+    video_path2 = get_video_path_local(model_name_2, type, prompt)
+    return video_path1, video_path2
 
 
 with gr.Blocks() as interface:
+    type_output = gr.Dropdown(label="Type", choices=types, value=types[0])
+    prompt_output = gr.Dropdown(label="Prompt", choices=dimension[types[0]], value=dimension[types[0]][0])
+
+    with gr.Row():
+        random_button = gr.Button("🎲 Random 2 videos")
+        display_button = gr.Button("⏸ Play options")
+   
     with gr.Row():
         with gr.Column():
-            model_name_1_output = gr.Textbox(label="Model Name 1")
+            model_name_1_output = gr.Dropdown(label="Model Name 1", choices=model_names, value=model_names[0])
             video_output_1 = gr.Video(label="Video 1")
         with gr.Column():
-            model_name_2_output = gr.Textbox(label="Model Name 2")
+            model_name_2_output = gr.Dropdown(label="Model Name 2", choices=model_names, value=model_names[1])
             video_output_2 = gr.Video(label="Video 2")
     
-    type_output = gr.Textbox(label="Type")
-    prompt_output = gr.Textbox(label="Prompt")
+    type_output.change(fn=update_prompt_options, inputs=[type_output], outputs=[prompt_output])
 
-    display_button = gr.Button("Display Videos")
-    display_button.click(
+    
+    random_button.click(
         fn=get_random_video,
-        outputs=[video_output_1, video_output_2, type_output, prompt_output, model_name_1_output, model_name_2_output]
+        outputs=[video_output_1, video_output_2,model_name_1_output, model_name_2_output, type_output, prompt_output, ]
+    )
+
+    display_button.click(
+        fn=display_videos,
+        inputs=[type_output, prompt_output, model_name_1_output, model_name_2_output],
+        outputs=[video_output_1, video_output_2]
     )
 
 interface.launch()