Add secret gate and limit default controls

app.py

@@ -38,6 +38,14 @@ import aoti
 os.environ["TOKENIZERS_PARALLELISM"] = "true"
 warnings.filterwarnings("ignore")
 IS_ZERO_GPU = bool(os.getenv("SPACES_ZERO_GPU"))
+SECRET_KEY = os.getenv("SECRET_KEY", "")
+_REQUIRE_SECRET_RAW = os.getenv("REQUIRE_SECRET")
+if _REQUIRE_SECRET_RAW is None:
+    # When SECRET_KEY is configured as a Space secret, require users to enter it.
+    REQUIRE_SECRET = bool(SECRET_KEY)
+else:
+    # REQUIRE_SECRET=0 keeps the Space public even if a key exists for testing.
+    REQUIRE_SECRET = _REQUIRE_SECRET_RAW.strip().lower() not in {"0", "false", "no", "off"}
 
 # if IS_ZERO_GPU:
 #     print("Loading...")
@@ -523,6 +531,7 @@ def run_inference(
 
 
 def generate_video(
+    secret_key,
     input_image,
     last_image,
     prompt,
@@ -547,11 +556,12 @@ def generate_video(
     prompt and parameters. It uses an FP8 qunatized Wan 2.2 14B Image-to-Video model in with Lightning LoRA
     for fast generation in 4-8 steps.
     Args:
+        secret_key (str): Optional access key checked when REQUIRE_SECRET is enabled.
         input_image (PIL.Image): The input image to animate. Will be resized to target dimensions.
         last_image (PIL.Image, optional): The optional last image for the video.
         prompt (str): Text prompt describing the desired animation or motion.
         steps (int, optional): Number of inference steps. More steps = higher quality but slower.
-            Defaults to 4. Range: 1-30.
+            Defaults to 4. Range: 1-8.
         negative_prompt (str, optional): Negative prompt to avoid unwanted elements.
             Defaults to default_negative_prompt (contains unwanted visual artifacts).
         duration_seconds (float, optional): Duration of the generated video in seconds.
@@ -578,7 +588,7 @@ def generate_video(
             - video_path (str): Path for the file download component. Attempt to avoid reconversion in video component.
             - current_seed (int): The seed used for generation.
     Raises:
-        gr.Error: If input_image is None (no image uploaded).
+        gr.Error: If the access key is invalid, or input_image is None (no image uploaded).
     Note:
         - Frame count is calculated as duration_seconds * FIXED_FPS (24)
         - Output dimensions are adjusted to be multiples of MOD_VALUE (32)
@@ -586,6 +596,9 @@ def generate_video(
         - Generation time varies based on steps and duration (see get_duration function)
     """
     
+    if REQUIRE_SECRET and secret_key != SECRET_KEY:
+        raise gr.Error("Invalid secret key.")
+
     if input_image is None:
         raise gr.Error("Please upload an input image.")
 
@@ -631,6 +644,9 @@ CSS = """
     position: absolute;
     pointer-events: none;
 }
+#secret-box {
+    max-width: 560px;
+}
 """
 
 
@@ -640,6 +656,12 @@ with gr.Blocks(delete_cache=(3600, 10800)) as demo:
 
     with gr.Row():
         with gr.Column():
+            secret_key_input = gr.Textbox(
+                label="Secret Key",
+                type="password",
+                visible=REQUIRE_SECRET,
+                elem_id="secret-box",
+            )
             input_image_component = gr.Image(type="pil", label="Input Image", sources=["upload", "clipboard"])
             prompt_input = gr.Textbox(label="Prompt", value=default_prompt_i2v)
             duration_seconds_input = gr.Slider(minimum=MIN_DURATION, maximum=MAX_DURATION, step=0.1, value=3.5, label="Duration (seconds)", info=f"Clamped to model's {MIN_FRAMES_MODEL}-{MAX_FRAMES_MODEL} frames at {FIXED_FPS}fps.")
@@ -651,7 +673,7 @@ with gr.Blocks(delete_cache=(3600, 10800)) as demo:
             )
             safe_mode_checkbox = gr.Checkbox(
                 label="🛠️ Safe Mode",
-                value=True,
+                value=False,
                 info="Requests 30% extra processing time to try to prevent unfinished tasks when the server is busy."
             )
             with gr.Accordion("Advanced Settings", open=False):
@@ -660,7 +682,7 @@ with gr.Blocks(delete_cache=(3600, 10800)) as demo:
                 quality_slider = gr.Slider(minimum=1, maximum=10, step=1, value=6, label="Video Quality", info="If set to 10, the generated video may be too large and won't play in the Gradio preview.")
                 seed_input = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=42, interactive=True)
                 randomize_seed_checkbox = gr.Checkbox(label="Randomize seed", value=True, interactive=True)
-                steps_slider = gr.Slider(minimum=1, maximum=30, step=1, value=6, label="Inference Steps")
+                steps_slider = gr.Slider(minimum=1, maximum=8, step=1, value=6, label="Inference Steps")
                 guidance_scale_input = gr.Slider(minimum=0.0, maximum=10.0, step=0.5, value=1, label="Guidance Scale - high noise stage", info="Values above 1 increase GPU usage and may take longer to process.")
                 guidance_scale_2_input = gr.Slider(minimum=0.0, maximum=10.0, step=0.5, value=1, label="Guidance Scale 2 - low noise stage")
                 scheduler_dropdown = gr.Dropdown(
@@ -692,6 +714,7 @@ with gr.Blocks(delete_cache=(3600, 10800)) as demo:
             file_output = gr.File(label="Download Video")
 
     ui_inputs = [
+        secret_key_input,
         input_image_component, last_image_component, prompt_input, steps_slider,
         negative_prompt_input, duration_seconds_input,
         guidance_scale_input, guidance_scale_2_input, seed_input, randomize_seed_checkbox,