Session

app.py

@@ -110,6 +110,23 @@ os.makedirs(outputs_folder, exist_ok=True)
 
 input_image_debug_value = input_video_debug_value = prompt_debug_value = total_second_length_debug_value = None
 
+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.")
@@ -971,7 +988,7 @@ with block:
     gr.HTML(title_html)
     with gr.Row():
         with gr.Column():
-            generation_mode = gr.Radio([["Text-to-Video", "text"], ["Image-to-Video", "image"], ["Video Extension", "video"]], label="Generation mode", value = "image")
+            generation_mode = gr.Radio([["Text-to-Video", "text"], ["Image-to-Video", "image"], ["Video Extension", "video"]], elem_id="generation-mode", label="Generation mode", value = "image")
             text_to_video_hint = gr.HTML("I discourage to use the Text-to-Video feature. You should rather generate an image with Flux and use Image-to-Video. You will save time.")
             input_image = gr.Image(sources='upload', type="numpy", label="Image", height=320)
             input_video = gr.Video(sources='upload', label="Input Video", height=320)
@@ -1056,6 +1073,13 @@ with block:
     ], outputs = [end_button], queue = False, show_progress = False).success(fn=process_video, inputs=ips_video, outputs=[result_video, preview_image, progress_desc, progress_bar, start_button_video, end_button])
     end_button.click(fn=end_process)
 
+    generation_mode.change(fn = save_preferences, inputs = [
+        local_storage,
+        generation_mode,
+    ], outputs = [
+       local_storage
+    ])
+
     with gr.Row(elem_id="image_examples", visible=False):
         gr.Examples(
         examples = [
@@ -1194,4 +1218,13 @@ with block:
     ]
     )
 
+    # Load saved preferences when the page loads
+    block.load(
+        fn=load_preferences, inputs = [
+        local_storage
+    ], outputs = [
+       generation_mode
+    ]
+    )
+
 block.launch(mcp_server=True, ssr_mode=False)