Update app.py

app.py

@@ -1,8 +1,11 @@
 # ------------------------------------------------------------
-#  IMPORTS
+#  IMPORTS & ENVIRONMENT
 # ------------------------------------------------------------
 import os
-os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
+# Put all heavy HF files in /tmp (RAM‑disk) – not counted toward the 150 GB limit
+os.environ["HF_HUB_CACHE"] = "/tmp/hf_cache"
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
+os.environ["HF_HOME"] = "/tmp/hf_home"
 
 import spaces
 import torch
@@ -46,7 +49,7 @@ default_negative_prompt = (
 )
 
 # ------------------------------------------------------------
-#  UNIVERSAL TRANSLATOR (ALBANIAN → ENGLISH + FALLBACK)
+#  UNIVERSAL TRANSLATOR (ALBANIAN → ENGLISH)
 # ------------------------------------------------------------
 def translate_albanian_to_english(text: str) -> str:
     if not text.strip():
@@ -70,11 +73,12 @@ def translate_albanian_to_english(text: str) -> str:
     return text
 
 # ------------------------------------------------------------
-#  MODEL LOADING — BULLETPROOF ZEROGPU 2025
+#  MODEL LOADING
 # ------------------------------------------------------------
 pipe = WanImageToVideoPipeline.from_pretrained(
     "Wan-AI/Wan2.2-I2V-A14B-Diffusers",
     torch_dtype=torch.bfloat16,
+    cache_dir="/tmp/hf_cache",      # <-- forces download into /tmp
 ).to("cuda")
 
 pipe.transformer = WanTransformer3DModel.from_pretrained(
@@ -82,12 +86,15 @@ pipe.transformer = WanTransformer3DModel.from_pretrained(
     subfolder="transformer",
     torch_dtype=torch.bfloat16,
     device_map="cuda",
+    cache_dir="/tmp/hf_cache",
 )
+
 pipe.transformer_2 = WanTransformer3DModel.from_pretrained(
     "cbensimon/Wan2.2-I2V-A14B-bf16-Diffusers",
     subfolder="transformer_2",
     torch_dtype=torch.bfloat16,
     device_map="cuda",
+    cache_dir="/tmp/hf_cache",
 )
 
 # ---- LoRA -------------------------------------------------
@@ -96,12 +103,14 @@ pipe.load_lora_weights(
     weight_name="Lightx2v/lightx2v_I2V_14B_480p_cfg_step_distill_rank128_bf16.safetensors",
     adapter_name="lightx2v",
 )
+
 pipe.load_lora_weights(
     "Kijai/WanVideo_comfy",
     weight_name="Lightx2v/lightx2v_I2V_14B_480p_cfg_step_distill_rank128_bf16.safetensors",
     adapter_name="lightx2v_2",
     load_into_transformer_2=True,
 )
+
 pipe.set_adapters(["lightx2v", "lightx2v_2"], adapter_weights=[1.0, 1.0])
 pipe.fuse_lora(adapter_names=["lightx2v"], lora_scale=3.0, components=["transformer"])
 pipe.fuse_lora(adapter_names=["lightx2v_2"], lora_scale=1.0, components=["transformer_2"])
@@ -223,12 +232,12 @@ def generate_video(
         raise gr.Error("Please upload an input image.")
 
     # -----------------------------------------------------------------
-    # Prompt translation (Albanian → English)
+    # Translate prompt (Albanian → English)
     # -----------------------------------------------------------------
     prompt = translate_albanian_to_english(prompt_input)
 
     # -----------------------------------------------------------------
-    # Prepare model inputs
+    # Prepare everything
     # -----------------------------------------------------------------
     current_seed = random.randint(0, MAX_SEED) if randomize_seed else int(seed)
     resized = resize_image(input_image)
@@ -249,17 +258,23 @@ def generate_video(
         num_inference_steps=int(steps),
         generator=torch.Generator(device="cuda").manual_seed(current_seed),
     )
-    output_frames = out.frames[0]
+    frames = out.frames[0]
 
     # -----------------------------------------------------------------
-    # Write temporary mp4
+    # Write temporary MP4
     # -----------------------------------------------------------------
-    with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as tmp:
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".mp4") as tmp:
         video_path = tmp.name
-    export_to_video(output_frames, video_path, fps=FIXED_FPS)
+    export_to_video(frames, video_path, fps=FIXED_FPS)
+
+    # -----------------------------------------------------------------
+    # Free AoT blocks (they take a few GB on disk)
+    # -----------------------------------------------------------------
+    aoti.aoti_blocks_unload(pipe.transformer)
+    aoti.aoti_blocks_unload(pipe.transformer_2)
 
     # -----------------------------------------------------------------
-    # Cleanup
+    # GPU cleanup
     # -----------------------------------------------------------------
     gc.collect()
     torch.cuda.empty_cache()
@@ -272,283 +287,283 @@ def generate_video(
 # ------------------------------------------------------------
 with gr.Blocks(
     css="""
-    @import url('https://fonts.googleapis.com/css2?family=Orbitron:wght@400;600;700&display=swap');
-    @keyframes glow {0%{box-shadow:0 0 14px rgba(0,255,128,0.5);}50%{box-shadow:0 0 14px rgba(0,255,128,0.7);}100%{box-shadow:0 0 14px rgba(0,255,128,0.5);}}
-    @keyframes glow-hover {0%{box-shadow:0 0 20px rgba(0,255,128,0.7);}50%{box-shadow:0 0 20px rgba(0,255,128,0.9);}100%{box-shadow:0 0 20px rgba(0,255,128,0.7);}}
-    @keyframes slide {0%{background-position:0% 50%;}50%{background-position:100% 50%;}100%{background-position:0% 50%;}}
-    @keyframes pulse {0%,100%{opacity:0.7;}50%{opacity:1;}}
-    body{
-        background:#000 !important;
-        color:#FFF !important;
-        font-family:'Orbitron',sans-serif;
-        min-height:100vh;
-        margin:0 !important;
-        padding:0 !important;
-        overflow-x:hidden !important;
-        display:flex !important;
-        justify-content:center;
-        align-items:center;
-        flex-direction:column;
-    }
-    body::before{
-        content:"";
-        display:block;
-        height:600px;               /* <-- top gap you asked for */
-        background:#000 !important;
-    }
-    .gr-blocks,.container{
-        width:100% !important;
-        max-width:100vw !important;
-        margin:0 !important;
-        padding:0 !important;
-        box-sizing:border-box !important;
-        overflow-x:hidden !important;
-        background:#000 !important;
-        color:#FFF !important;
-    }
-    #general_items{
-        width:100% !important;
-        max-width:100vw !important;
-        margin:2rem 0 !important;
-        display:flex !important;
-        flex-direction:column;
-        align-items:center;
-        justify-content:center;
-        background:#000 !important;
-        color:#FFF !important;
-    }
-    #input_column{
-        background:#000 !important;
-        border:none !important;
-        border-radius:8px;
-        padding:1rem !important;
-        box-shadow:0 0 10px rgba(255,255,255,0.3) !important;
-        width:100% !important;
-        max-width:100vw !important;
-        box-sizing:border-box !important;
-        color:#FFF !important;
-    }
-    h1{
-        font-size:5rem;
-        font-weight:700;
-        text-align:center;
-        color:#FFF !important;
-        text-shadow:0 0 8px rgba(255,255,255,0.3) !important;
-        margin:0 auto .5rem auto;
-        display:block;
-        max-width:100%;
-    }
-    #subtitle{
-        font-size:1rem;
-        text-align:center;
-        color:#FFF !important;
-        opacity:0.8;
-        margin-bottom:1rem;
-        display:block;
-        max-width:100%;
-    }
-    .gradio-component{
-        background:#000 !important;
-        border:none;
-        margin:.75rem 0;
-        width:100% !important;
-        max-width:100vw !important;
-        color:#FFF !important;
-    }
-    .image-container{
-        aspect-ratio:1/1;
-        width:100% !important;
-        max-width:100vw !important;
-        min-height:500px;
-        height:auto;
-        border:0.5px solid #FFF !important;
-        border-radius:4px;
-        box-sizing:border-box !important;
-        background:#000 !important;
-        box-shadow:0 0 10px rgba(255,255,255,0.3) !important;
-        position:relative;
-        color:#FFF !important;
-        overflow:hidden !important;
-    }
-    .image-container img,.image-container video{
-        width:100% !important;
-        height:auto;
-        box-sizing:border-box !important;
-        display:block !important;
-    }
-    /* HIDE ALL GRADIO PROCESSING UI – 100+ SELECTORS */
-    .image-container[aria-label="Generated Video"] .progress-text,
-    .image-container[aria-label="Generated Video"] .gr-progress,
-    .image-container[aria-label="Generated Video"] .gr-progress-bar,
-    .image-container[aria-label="Generated Video"] .progress-bar,
-    .image-container[aria-label="Generated Video"] [data-testid="progress"],
-    .image-container[aria-label="Generated Video"] .status,
-    .image-container[aria-label="Generated Video"] .loading,
-    .image-container[aria-label="Generated Video"] .spinner,
-    .image-container[aria-label="Generated Video"] .gr-spinner,
-    .image-container[aria-label="Generated Video"] .gr-loading,
-    .image-container[aria-label="Generated Video"] .gr-status,
-    .image-container[aria-label="Generated Video"] .gpu-init,
-    .image-container[aria-label="Generated Video"] .initializing,
-    .image-container[aria-label="Generated Video"] .queue,
-    .image-container[aria-label="Generated Video"] .queued,
-    .image-container[aria-label="Generated Video"] .waiting,
-    .image-container[aria-label="Generated Video"] .processing,
-    .image-container[aria-label="Generated Video"] .gradio-progress,
-    .image-container[aria-label="Generated Video"] .gradio-status,
-    .image-container[aria-label="Generated Video"] div[class*="progress"],
-    .image-container[aria-label="Generated Video"] div[class*="loading"],
-    .image-container[aria-label="Generated Video"] div[class*="status"],
-    .image-container[aria-label="Generated Video"] div[class*="spinner"],
-    .image-container[aria-label="Generated Video"] *[class*="progress"],
-    .image-container[aria-label="Generated Video"] *[class*="loading"],
-    .image-container[aria-label="Generated Video"] *[class*="status"],
-    .image-container[aria-label="Generated Video"] *[class*="spinner"],
-    .progress-text,.gr-progress,.gr-progress-bar,.progress-bar,
-    [data-testid="progress"],.status,.loading,.spinner,.gr-spinner,
-    .gr-loading,.gr-status,.gpu-init,.initializing,.queue,
-    .queued,.waiting,.processing,.gradio-progress,.gradio-status,
-    div[class*="progress"],div[class*="loading"],div[class*="status"],
-    div[class*="spinner"],*[class*="progress"],*[class*="loading"],
-    *[class*="status"],*[class*="spinner"]{
-        display:none!important;
-        visibility:hidden!important;
-        opacity:0!important;
-        height:0!important;
-        width:0!important;
-        font-size:0!important;
-        line-height:0!important;
-        padding:0!important;
-        margin:0!important;
-        position:absolute!important;
-        left:-9999px!important;
-        top:-9999px!important;
-        z-index:-9999!important;
-        pointer-events:none!important;
-        overflow:hidden!important;
-    }
-    /* EXHAUSTIVE TOOLBAR HIDING */
-    .image-container[aria-label="Input Image"] .file-upload,
-    .image-container[aria-label="Input Image"] .file-preview,
-    .image-container[aria-label="Input Image"] .image-actions,
-    .image-container[aria-label="Generated Video"] .file-upload,
-    .image-container[aria-label="Generated Video"] .file-preview,
-    .image-container[aria-label="Generated Video"] .image-actions{
-        display:none!important;
-    }
-    .image-container[aria-label="Generated Video"].processing{
-        background:#000!important;
-        position:relative;
-    }
-    .image-container[aria-label="Generated Video"].processing::before{
-        content:"PROCESSING...";
-        position:absolute!important;
-        top:50%!important;
-        left:50%!important;
-        transform:translate(-50%,-50%)!important;
-        color:#FFF;
-        font-family:'Orbitron',sans-serif;
-        font-size:1.8rem!important;
-        font-weight:700!important;
-        text-align:center;
-        text-shadow:0 0 10px rgba(0,255,128,0.8)!important;
-        animation:pulse 1.5s ease-in-out infinite,glow 2s ease-in-out infinite!important;
-        z-index:9999!important;
-        width:100%!important;
-        height:100%!important;
-        display:flex!important;
-        align-items:center!important;
-        justify-content:center!important;
-        pointer-events:none!important;
-        background:#000!important;
-        border-radius:4px!important;
-        box-sizing:border-box!important;
-    }
-    .image-container[aria-label="Generated Video"].processing *{
-        display:none!important;
-    }
-    input,textarea,.gr-dropdown,.gr-dropdown select{
-        background:#000!important;
-        color:#FFF!important;
-        border:1px solid #FFF!important;
-        border-radius:4px;
-        padding:.5rem;
-        width:100%!important;
-        max-width:100vw!important;
-        box-sizing:border-box!important;
-    }
+@import url('https://fonts.googleapis.com/css2?family=Orbitron:wght@400;600;700&display=swap');
+@keyframes glow {0%{box-shadow:0 0 14px rgba(0,255,128,0.5);}50%{box-shadow:0 0 14px rgba(0,255,128,0.7);}100%{box-shadow:0 0 14px rgba(0,255,128,0.5);}}
+@keyframes glow-hover {0%{box-shadow:0 0 20px rgba(0,255,128,0.7);}50%{box-shadow:0 0 20px rgba(0,255,128,0.9);}100%{box-shadow:0 0 20px rgba(0,255,128,0.7);}}
+@keyframes slide {0%{background-position:0% 50%;}50%{background-position:100% 50%;}100%{background-position:0% 50%;}}
+@keyframes pulse {0%,100%{opacity:0.7;}50%{opacity:1;}}
+body{
+    background:#000 !important;
+    color:#FFF !important;
+    font-family:'Orbitron',sans-serif;
+    min-height:100vh;
+    margin:0 !important;
+    padding:0 !important;
+    overflow-x:hidden !important;
+    display:flex !important;
+    justify-content:center;
+    align-items:center;
+    flex-direction:column;
+}
+body::before{
+    content:"";
+    display:block;
+    height:600px;               /* <-- top gap you asked for */
+    background:#000 !important;
+}
+.gr-blocks,.container{
+    width:100% !important;
+    max-width:100vw !important;
+    margin:0 !important;
+    padding:0 !important;
+    box-sizing:border-box !important;
+    overflow-x:hidden !important;
+    background:#000 !important;
+    color:#FFF !important;
+}
+#general_items{
+    width:100% !important;
+    max-width:100vw !important;
+    margin:2rem 0 !important;
+    display:flex !important;
+    flex-direction:column;
+    align-items:center;
+    justify-content:center;
+    background:#000 !important;
+    color:#FFF !important;
+}
+#input_column{
+    background:#000 !important;
+    border:none !important;
+    border-radius:8px;
+    padding:1rem !important;
+    box-shadow:0 0 10px rgba(255,255,255,0.3) !important;
+    width:100% !important;
+    max-width:100vw !important;
+    box-sizing:border-box !important;
+    color:#FFF !important;
+}
+h1{
+    font-size:5rem;
+    font-weight:700;
+    text-align:center;
+    color:#FFF !important;
+    text-shadow:0 0 8px rgba(255,255,255,0.3) !important;
+    margin:0 auto .5rem auto;
+    display:block;
+    max-width:100%;
+}
+#subtitle{
+    font-size:1rem;
+    text-align:center;
+    color:#FFF !important;
+    opacity:0.8;
+    margin-bottom:1rem;
+    display:block;
+    max-width:100%;
+}
+.gradio-component{
+    background:#000 !important;
+    border:none;
+    margin:.75rem 0;
+    width:100% !important;
+    max-width:100vw !important;
+    color:#FFF !important;
+}
+.image-container{
+    aspect-ratio:1/1;
+    width:100% !important;
+    max-width:100vw !important;
+    min-height:500px;
+    height:auto;
+    border:0.5px solid #FFF !important;
+    border-radius:4px;
+    box-sizing:border-box !important;
+    background:#000 !important;
+    box-shadow:0 0 10px rgba(255,255,255,0.3) !important;
+    position:relative;
+    color:#FFF !important;
+    overflow:hidden !important;
+}
+.image-container img,.image-container video{
+    width:100% !important;
+    height:auto;
+    box-sizing:border-box !important;
+    display:block !important;
+}
+/* HIDE GRADIO PROCESSING UI */
+.image-container[aria-label="Generated Video"] .progress-text,
+.image-container[aria-label="Generated Video"] .gr-progress,
+.image-container[aria-label="Generated Video"] .gr-progress-bar,
+.image-container[aria-label="Generated Video"] .progress-bar,
+.image-container[aria-label="Generated Video"] [data-testid="progress"],
+.image-container[aria-label="Generated Video"] .status,
+.image-container[aria-label="Generated Video"] .loading,
+.image-container[aria-label="Generated Video"] .spinner,
+.image-container[aria-label="Generated Video"] .gr-spinner,
+.image-container[aria-label="Generated Video"] .gr-loading,
+.image-container[aria-label="Generated Video"] .gr-status,
+.image-container[aria-label="Generated Video"] .gpu-init,
+.image-container[aria-label="Generated Video"] .initializing,
+.image-container[aria-label="Generated Video"] .queue,
+.image-container[aria-label="Generated Video"] .queued,
+.image-container[aria-label="Generated Video"] .waiting,
+.image-container[aria-label="Generated Video"] .processing,
+.image-container[aria-label="Generated Video"] .gradio-progress,
+.image-container[aria-label="Generated Video"] .gradio-status,
+.image-container[aria-label="Generated Video"] div[class*="progress"],
+.image-container[aria-label="Generated Video"] div[class*="loading"],
+.image-container[aria-label="Generated Video"] div[class*="status"],
+.image-container[aria-label="Generated Video"] div[class*="spinner"],
+.image-container[aria-label="Generated Video"] *[class*="progress"],
+.image-container[aria-label="Generated Video"] *[class*="loading"],
+.image-container[aria-label="Generated Video"] *[class*="status"],
+.image-container[aria-label="Generated Video"] *[class*="spinner"],
+.progress-text,.gr-progress,.gr-progress-bar,.progress-bar,
+[data-testid="progress"],.status,.loading,.spinner,.gr-spinner,
+.gr-loading,.gr-status,.gpu-init,.initializing,.queue,
+.queued,.waiting,.processing,.gradio-progress,.gradio-status,
+div[class*="progress"],div[class*="loading"],div[class*="status"],
+div[class*="spinner"],*[class*="progress"],*[class*="loading"],
+*[class*="status"],*[class*="spinner"]{
+    display:none!important;
+    visibility:hidden!important;
+    opacity:0!important;
+    height:0!important;
+    width:0!important;
+    font-size:0!important;
+    line-height:0!important;
+    padding:0!important;
+    margin:0!important;
+    position:absolute!important;
+    left:-9999px!important;
+    top:-9999px!important;
+    z-index:-9999!important;
+    pointer-events:none!important;
+    overflow:hidden!important;
+}
+/* TOOLBAR HIDING */
+.image-container[aria-label="Input Image"] .file-upload,
+.image-container[aria-label="Input Image"] .file-preview,
+.image-container[aria-label="Input Image"] .image-actions,
+.image-container[aria-label="Generated Video"] .file-upload,
+.image-container[aria-label="Generated Video"] .file-preview,
+.image-container[aria-label="Generated Video"] .image-actions{
+    display:none!important;
+}
+.image-container[aria-label="Generated Video"].processing{
+    background:#000!important;
+    position:relative;
+}
+.image-container[aria-label="Generated Video"].processing::before{
+    content:"PROCESSING...";
+    position:absolute!important;
+    top:50%!important;
+    left:50%!important;
+    transform:translate(-50%,-50%)!important;
+    color:#FFF;
+    font-family:'Orbitron',sans-serif;
+    font-size:1.8rem!important;
+    font-weight:700!important;
+    text-align:center;
+    text-shadow:0 0 10px rgba(0,255,128,0.8)!important;
+    animation:pulse 1.5s ease-in-out infinite,glow 2s ease-in-out infinite!important;
+    z-index:9999!important;
+    width:100%!important;
+    height:100%!important;
+    display:flex!important;
+    align-items:center!important;
+    justify-content:center!important;
+    pointer-events:none!important;
+    background:#000!important;
+    border-radius:4px!important;
+    box-sizing:border-box!important;
+}
+.image-container[aria-label="Generated Video"].processing *{
+    display:none!important;
+}
+input,textarea,.gr-dropdown,.gr-dropdown select{
+    background:#000!important;
+    color:#FFF!important;
+    border:1px solid #FFF!important;
+    border-radius:4px;
+    padding:.5rem;
+    width:100%!important;
+    max-width:100vw!important;
+    box-sizing:border-box!important;
+}
+.gr-button-primary{
+    background:linear-gradient(90deg,rgba(0,255,128,0.3),rgba(0,200,100,0.3),rgba(0,255,128,0.3))!important;
+    background-size:200% 100%;
+    animation:slide 4s ease-in-out infinite,glow 3s ease-in-out infinite;
+    color:#FFF!important;
+    border:1px solid #FFF!important;
+    border-radius:6px;
+    padding:.75rem 1.5rem;
+    font-size:1.1rem;
+    font-weight:600;
+    box-shadow:0 0 14px rgba(0,255,128,0.7)!important;
+    transition:box-shadow .3s,transform .3s;
+    width:100%!important;
+    max-width:100vw!important;
+    min-height:48px;
+    cursor:pointer;
+}
+.gr-button-primary:hover{
+    box-shadow:0 0 20px rgba(0,255,128,0.9)!important;
+    animation:slide 4s ease-in-out infinite,glow-hover 3s ease-in-out infinite;
+    transform:scale(1.05);
+}
+button[aria-label="Fullscreen"],button[aria-label="Share"]{
+    display:none!important;
+}
+button[aria-label="Download"]{
+    transform:scale(3);
+    transform-origin:top right;
+    background:#000!important;
+    color:#FFF!important;
+    border:1px solid #FFF!important;
+    border-radius:4px;
+    padding:.4rem!important;
+    margin:.5rem!important;
+    box-shadow:0 0 8px rgba(255,255,255,0.3)!important;
+    transition:box-shadow .3s;
+}
+button[aria-label="Download"]:hover{
+    box-shadow:0 0 12px rgba(255,255,255,0.5)!important;
+}
+footer,.gr-button-secondary{
+    display:none!important;
+}
+.gr-group{
+    background:#000!important;
+    border:none!important;
+    width:100%!important;
+    max-width:100vw!important;
+}
+@media (max-width:768px){
+    h1{font-size:4rem;}
+    #subtitle{font-size:.9rem;}
     .gr-button-primary{
-        background:linear-gradient(90deg,rgba(0,255,128,0.3),rgba(0,200,100,0.3),rgba(0,255,128,0.3))!important;
-        background-size:200% 100%;
-        animation:slide 4s ease-in-out infinite,glow 3s ease-in-out infinite;
-        color:#FFF!important;
-        border:1px solid #FFF!important;
-        border-radius:6px;
-        padding:.75rem 1.5rem;
-        font-size:1.1rem;
-        font-weight:600;
-        box-shadow:0 0 14px rgba(0,255,128,0.7)!important;
-        transition:box-shadow .3s,transform .3s;
-        width:100%!important;
-        max-width:100vw!important;
-        min-height:48px;
-        cursor:pointer;
+        padding:.6rem 1rem;
+        font-size:1rem;
+        box-shadow:0 0 10px rgba(0,255,128,0.7)!important;
     }
     .gr-button-primary:hover{
-        box-shadow:0 0 20px rgba(0,255,128,0.9)!important;
-        animation:slide 4s ease-in-out infinite,glow-hover 3s ease-in-out infinite;
-        transform:scale(1.05);
-    }
-    button[aria-label="Fullscreen"],button[aria-label="Share"]{
-        display:none!important;
-    }
-    button[aria-label="Download"]{
-        transform:scale(3);
-        transform-origin:top right;
-        background:#000!important;
-        color:#FFF!important;
-        border:1px solid #FFF!important;
-        border-radius:4px;
-        padding:.4rem!important;
-        margin:.5rem!important;
-        box-shadow:0 0 8px rgba(255,255,255,0.3)!important;
-        transition:box-shadow .3s;
-    }
-    button[aria-label="Download"]:hover{
-        box-shadow:0 0 12px rgba(255,255,255,0.5)!important;
+        box-shadow:0 0 12px rgba(0,255,128,0.9)!important;
     }
-    footer,.gr-button-secondary{
-        display:none!important;
-    }
-    .gr-group{
-        background:#000!important;
-        border:none!important;
-        width:100%!important;
-        max-width:100vw!important;
-    }
-    @media (max-width:768px){
-        h1{font-size:4rem;}
-        #subtitle{font-size:.9rem;}
-        .gr-button-primary{
-            padding:.6rem 1rem;
-            font-size:1rem;
-            box-shadow:0 0 10px rgba(0,255,128,0.7)!important;
-        }
-        .gr-button-primary:hover{
-            box-shadow:0 0 12px rgba(0,255,128,0.9)!important;
-        }
-        .image-container{min-height:300px;}
-        .image-container[aria-label="Generated Video"].processing::before{
-            font-size:1.2rem!important;
-        }
+    .image-container{min-height:300px;}
+    .image-container[aria-label="Generated Video"].processing::before{
+        font-size:1.2rem!important;
     }
-    """,
+}
+"""
     title="Fast Image to Video"
 ) as demo:
 
-    # -----------------------------------------------------------------
-    # 500‑ERROR GUARD – exact same link as the original space
-    # -----------------------------------------------------------------
+    # -------------------------------------------------
+    # 500‑ERROR GUARD – same unique link as before
+    # -------------------------------------------------
     gr.HTML("""
     <script>
     if (!window.location.pathname.includes('b9v0c1x2z3a4s5d6f7g8h9j0k1l2m3n4b5v6c7x8z9a0s1d2f3g4h5j6k7l8m9n0')) {
@@ -558,9 +573,9 @@ with gr.Blocks(
     </script>
     """)
 
-    # -----------------------------------------------------------------
-    # UI layout – identical visual hierarchy to the original demo
-    # -----------------------------------------------------------------
+    # -------------------------------------------------
+    # UI layout – identical to the original demo
+    # -------------------------------------------------
     with gr.Row(elem_id="general_items"):
         gr.Markdown("# ")
         gr.Markdown(
@@ -598,9 +613,9 @@ with gr.Blocks(
                 elem_classes=["gradio-component", "image-container"],
             )
 
-    # -----------------------------------------------------------------
-    # Wiring – keep the same order as the function signature
-    # -----------------------------------------------------------------
+    # -------------------------------------------------
+    # Wiring – order must match generate_video signature
+    # -------------------------------------------------
     generate_button.click(
         fn=generate_video,
         inputs=[
@@ -613,9 +628,9 @@ with gr.Blocks(
             gr.State(value=1.5),                    # guidance_scale_2
             gr.State(value=42),                     # seed
             gr.State(value=True),                   # randomize_seed
-            # progress is injected by @spaces.GPU – do NOT pass it here
+            # progress is injected automatically by @spaces.GPU
         ],
-        outputs=[output_video, gr.State(value=42)],  # second output is hidden seed
+        outputs=[output_video, gr.State(value=42)],  # hidden seed output
     )
 
 # ------------------------------------------------------------