Spaces:
Running on Zero
Running on Zero
File size: 7,181 Bytes
31e9635 ca59823 31e9635 347cf89 31e9635 5d38bdd 31e9635 5d38bdd 31e9635 5d38bdd 31e9635 5d38bdd 31e9635 5550086 31e9635 347cf89 5d38bdd 31e9635 01b6505 31e9635 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 | import gradio as gr
import config
import lora_loader
import prompt_enhancer
from image_utils import load_image_from_url
def render_column_inputs():
with gr.Column(scale=4):
input_image_component = gr.Image(
type="pil",
label="🖼️ Input Image",
sources=["upload", "clipboard"],
height=240
)
with gr.Row():
image_url_input = gr.Textbox(
label="🌐 Load Input Image from URL",
placeholder="e.g. https://example.com/image.jpg",
value="",
scale=3
)
load_image_btn = gr.Button("📥 Load Image", variant="secondary", scale=1)
load_image_btn.click(
fn=load_image_from_url,
inputs=[image_url_input],
outputs=[input_image_component]
)
with gr.Accordion("✨ Base Prompt & Style Modifiers", open=True):
prompt_input = gr.Textbox(
label="✨ Base Prompt",
value=config.default_prompt_i2v,
lines=3,
max_lines=6,
placeholder="Describe global style, lighting, or overall video mood...",
info="💡 Note: If Prompt Relay is enabled, Base Prompt is ignored (Prompt Relay takes full control)."
)
with gr.Row():
describe_image_btn = gr.Button("🔍 Describe Image (CPU ~1s)", variant="secondary")
enhance_prompt_btn = gr.Button("✨ Auto-Enhance Prompt (CPU)", variant="secondary")
describe_image_btn.click(
fn=prompt_enhancer.describe_image,
inputs=[input_image_component],
outputs=[prompt_input]
)
enhance_prompt_btn.click(
fn=prompt_enhancer.enhance_prompt,
inputs=[prompt_input],
outputs=[prompt_input]
)
duration_seconds_input = gr.Slider(
minimum=config.MIN_DURATION,
maximum=config.MAX_DURATION,
step=0.1,
value=4.0,
label="⏱️ Duration (seconds)",
info=f"Set to 4.0s for direct 65-frame model generation."
)
motion_extension_dropdown = gr.Dropdown(
choices=[
"⚡ Real-Time RIFE Interpolation (32/64 FPS Ultra-Smooth)",
"🔂 Ending-Only Boomerang Loop (Real Speed, Tail 1.5s Loop)",
"🔂 Classic Full Boomerang Loop (100% Real-Speed Forward+Reverse)",
"🌊 Adaptive Motion Speed Ramping (Ease-In/Out Curve)",
"🐢 Cinematic Slow-Motion (16 FPS RIFE Time-Stretch)"
],
value="⚡ Real-Time RIFE Interpolation (32/64 FPS Ultra-Smooth)",
label="🎬 Motion Extension & Loop Technique",
info="Choose Real-Time RIFE (Default), Ending-Only Boomerang, Classic Boomerang, or Speed Ramping."
)
frame_multi = gr.Dropdown(
choices=[
(f"16 FPS (Original Duration)", config.FIXED_FPS),
(f"32 FPS (2x RIFE -> Converts 2s GPU to 4s Video)", config.FIXED_FPS*2),
(f"64 FPS (4x RIFE -> Converts 2s GPU to 8s Video)", config.FIXED_FPS*4),
],
value=config.FIXED_FPS,
label="🎬 Video Fluidity & RIFE Extension (FPS)",
info="Select 32 FPS (2x RIFE) to extend 2s GPU output into a smooth 4s video on CPU (0 GPU quota)."
)
noise_temperature_slider = gr.Slider(
minimum=0.1,
maximum=2.0,
step=0.05,
value=1.0,
label="🌡️ Noise Temperature",
info="Scales initial noise variance (0.5 = focused & smooth, 1.0 = standard, 1.5 = high visual randomness)."
)
with gr.Accordion("🔗 Custom Civitai / Direct LoRA URL & CPU Cache", open=False):
with gr.Row():
cached_lora_dropdown = gr.Dropdown(
label="📦 Pre-Downloaded / Cached LoRAs (CPU Storage)",
choices=lora_loader.list_cached_loras(),
value="(None / Disable)",
scale=3,
info="Select any previously downloaded LoRA from CPU cache"
)
refresh_cached_lora_btn = gr.Button("🔄 Refresh List", variant="secondary", scale=1)
custom_lora_url_input = gr.Textbox(
label="Civitai / Direct LoRA Download URL (New Download)",
placeholder="e.g. https://civitai.red/api/download/models/2098405?fileId=1994044",
value="",
info="Paste any Civitai or direct HTTP/HTTPS .safetensors link to download to CPU cache"
)
custom_lora_scale_input = gr.Slider(
label="Custom LoRA Weight Scale",
minimum=0.0,
maximum=2.0,
step=0.05,
value=1.0,
info="Strength of the custom LoRA effect"
)
with gr.Row():
download_lora_btn = gr.Button("📥 Pre-Download LoRA to Cache (CPU)", variant="secondary")
download_status_box = gr.HTML()
download_lora_btn.click(
fn=lora_loader.download_custom_lora_ui_action,
inputs=[custom_lora_url_input],
outputs=[download_status_box, cached_lora_dropdown]
)
refresh_cached_lora_btn.click(
fn=lambda: gr.update(choices=lora_loader.list_cached_loras()),
inputs=None,
outputs=[cached_lora_dropdown]
)
with gr.Row():
safe_mode_checkbox = gr.Checkbox(
label="🛠️ Safe Mode",
value=False,
info="Requests extra processing buffer time to prevent timeout."
)
generate_button = gr.Button("🚀 Generate Video", variant="primary", elem_id="generate-btn")
extra_gpu_buffer_slider = gr.Slider(
label="⚡ Extra GPU Reservation Buffer (Seconds)",
minimum=0,
maximum=10,
step=1,
value=3,
info="Manually add 0-10 extra seconds to ZeroGPU time reservation to prevent timeout during heavy processing"
)
return {
"input_image_component": input_image_component,
"image_url_input": image_url_input,
"load_image_btn": load_image_btn,
"prompt_input": prompt_input,
"describe_image_btn": describe_image_btn,
"enhance_prompt_btn": enhance_prompt_btn,
"duration_seconds_input": duration_seconds_input,
"motion_extension_dropdown": motion_extension_dropdown,
"frame_multi": frame_multi,
"noise_temperature_slider": noise_temperature_slider,
"cached_lora_dropdown": cached_lora_dropdown,
"custom_lora_url_input": custom_lora_url_input,
"custom_lora_scale_input": custom_lora_scale_input,
"extra_gpu_buffer_slider": extra_gpu_buffer_slider,
"safe_mode_checkbox": safe_mode_checkbox,
"generate_button": generate_button
}
|