Update app.py

app.py

@@ -91,8 +91,11 @@ 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 update_prompt_options(type, value=None):
+    if value:
+        return gr.update(choices=dimension[type], value=value if dimension[type] else None)
+    else:    
+        return gr.update(choices=dimension[type], value=dimension[type][0] if dimension[type] else None)
 
 
 def display_videos(type, prompt, model_name_1, model_name_2):
@@ -117,13 +120,15 @@ with gr.Blocks() as interface:
             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.change(fn=update_prompt_options, inputs=[type_output], outputs=[prompt_output])
+    type_output.change(fn=update_prompt_options, inputs=[type_output,None], outputs=[prompt_output])
 
     
     random_button.click(
         fn=get_random_video,
         outputs=[video_output_1, video_output_2,model_name_1_output, model_name_2_output, type_output, prompt_output, ]
-    )
+    ).then(fn=update_prompt_options,
+          inputs=[type_output, prompt_output],
+          outputs=[prompt_output])
 
     display_button.click(
         fn=display_videos,