Update display when the page loads

app.py

@@ -114,26 +114,6 @@ default_local_storage = {
         "generation-mode": "image",
     }
 
-def save_preferences(preferences, value):
-    preferences["generation-mode"] = value
-    return preferences
-
-def load_preferences(saved_prefs):
-    saved_prefs = init_preferences(saved_prefs)
-    return saved_prefs["generation-mode"]
-
-def init_preferences(saved_prefs):
-    if saved_prefs is None:
-        saved_prefs = default_local_storage
-    return saved_prefs
-
-def check_parameters(generation_mode, input_image, input_video):
-    if generation_mode == "image" and input_image is None:
-        raise gr.Error("Please provide an image to extend.")
-    if generation_mode == "video" and input_video is None:
-        raise gr.Error("Please provide a video to extend.")
-    return [gr.update(interactive=True)]
-
 @spaces.GPU()
 @torch.no_grad()
 def video_encode(video_path, resolution, no_resize, vae, vae_batch_size=16, device="cuda", width=None, height=None):
@@ -1064,6 +1044,26 @@ with block:
     ips = [input_image, final_prompt, generation_mode, n_prompt, randomize_seed, seed, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, use_teacache, mp4_crf]
     ips_video = [input_video, final_prompt, n_prompt, randomize_seed, seed, batch, resolution, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, use_teacache, no_resize, mp4_crf, num_clean_frames, vae_batch]
 
+    def save_preferences(preferences, value):
+        preferences["generation-mode"] = value
+        return preferences
+    
+    def load_preferences(saved_prefs):
+        saved_prefs = init_preferences(saved_prefs)
+        return saved_prefs["generation-mode"]
+    
+    def init_preferences(saved_prefs):
+        if saved_prefs is None:
+            saved_prefs = default_local_storage
+        return saved_prefs
+    
+    def check_parameters(generation_mode, input_image, input_video):
+        if generation_mode == "image" and input_image is None:
+            raise gr.Error("Please provide an image to extend.")
+        if generation_mode == "video" and input_video is None:
+            raise gr.Error("Please provide a video to extend.")
+        return [gr.update(interactive=True)]
+
     prompt_number.change(fn=handle_prompt_number_change, inputs=[], outputs=[])
     timeless_prompt.change(fn=handle_timeless_prompt_change, inputs=[timeless_prompt], outputs=[final_prompt])
     start_button.click(fn = check_parameters, inputs = [
@@ -1210,6 +1210,15 @@ with block:
         outputs=[]
     )
 
+    # Update display when the page loads
+    block.load(
+        fn=handle_generation_mode_change, inputs = [
+        generation_mode
+    ], outputs = [
+       text_to_video_hint, input_image, input_video, start_button, start_button_video, no_resize, batch, resolution, num_clean_frames, vae_batch
+    ]
+    )
+
     # Load saved preferences when the page loads
     block.load(
         fn=load_preferences, inputs = [