Spaces:
Runtime error
Runtime error
File size: 3,469 Bytes
6cf4883 |
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 |
import gradio as gr
from .video_audio_utilities import extract_number, get_quick_vid_info
def change_visibility_from_skip_video(choice):
return gr.update(visible=False) if choice else gr.update(visible=True)
def update_r_upscale_factor(choice):
return gr.update(value='x4', choices = ['x4']) if choice != 'realesr-animevideov3' else gr.update(value='x2', choices = ['x2', 'x3', 'x4'])
def change_perlin_visibility(choice):
return gr.update(visible=choice=="perlin")
def change_color_coherence_video_every_N_frames_visibility(choice):
return gr.update(visible=choice=="Video Input")
def change_seed_iter_visibility(choice):
return gr.update(visible=choice=="iter")
def change_seed_schedule_visibility(choice):
return gr.update(visible=choice=="schedule")
def disable_pers_flip_accord(choice):
return gr.update(visible=True) if choice in ['2D','3D'] else gr.update(visible=False)
def change_max_frames_visibility(choice):
return gr.update(visible=choice != "Video Input")
def change_diffusion_cadence_visibility(choice):
return gr.update(visible=choice not in ['Video Input', 'Interpolation'])
def disble_3d_related_stuff(choice):
return gr.update(visible=False) if choice != '3D' else gr.update(visible=True)
def enable_2d_related_stuff(choice):
return gr.update(visible=True) if choice == '2D' else gr.update(visible=False)
def disable_by_interpolation(choice):
return gr.update(visible=False) if choice in ['Interpolation'] else gr.update(visible=True)
def disable_by_video_input(choice):
return gr.update(visible=False) if choice in ['Video Input'] else gr.update(visible=True)
def change_comp_mask_x_visibility(choice):
return gr.update(visible=choice != "None")
def change_gif_button_visibility(choice):
return gr.update(visible=False, value=False) if int(choice) > 30 else gr.update(visible=True)
def disable_by_hybrid_composite(choice):
return gr.update(visible=True) if choice else gr.update(visible=False)
def disable_by_hybrid_composite_dynamic(choice, comp_mask_type):
if choice == True:
if comp_mask_type != 'None':
return gr.update(visible=True)
return gr.update(visible=False)
def disable_by_comp_mask(choice):
return gr.update(visible=False) if choice == 'None' else gr.update(visible=True)
def disable_by_non_optical_flow(choice):
return gr.update(visible=False) if choice != 'Optical Flow' else gr.update(visible=True)
# Upscaling Gradio UI related funcs
def vid_upscale_gradio_update_stats(vid_path, upscale_factor):
if not vid_path:
return '---', '---', '---', '---'
factor = extract_number(upscale_factor)
fps, fcount, resolution = get_quick_vid_info(vid_path.name)
in_res_str = f"{resolution[0]}*{resolution[1]}"
out_res_str = f"{resolution[0] * factor}*{resolution[1] * factor}"
return fps, fcount, in_res_str, out_res_str
def update_upscale_out_res(in_res, upscale_factor):
if not in_res:
return '---'
factor = extract_number(upscale_factor)
w, h = [int(x) * factor for x in in_res.split('*')]
return f"{w}*{h}"
def update_upscale_out_res_by_model_name(in_res, upscale_model_name):
if not upscale_model_name or in_res == '---':
return '---'
factor = 2 if upscale_model_name == 'realesr-animevideov3' else 4
return f"{int(in_res.split('*')[0]) * factor}*{int(in_res.split('*')[1]) * factor}" |