vnyash
84 lines
2.8 KiB
Python
84 lines
2.8 KiB
Python
from typing import Tuple, Optional
|
|
import gradio
|
|
|
|
import deepfuze.globals
|
|
from deepfuze import wording
|
|
from deepfuze.face_store import clear_static_faces, clear_reference_faces
|
|
from deepfuze.uis.typing import File
|
|
from deepfuze.filesystem import get_file_size, is_image, is_video
|
|
from deepfuze.uis.core import register_ui_component
|
|
from deepfuze.vision import get_video_frame, normalize_frame_color
|
|
|
|
FILE_SIZE_LIMIT = 512 * 1024 * 1024
|
|
|
|
TARGET_FILE : Optional[gradio.File] = None
|
|
TARGET_IMAGE : Optional[gradio.Image] = None
|
|
TARGET_VIDEO : Optional[gradio.Video] = None
|
|
|
|
|
|
def render() -> None:
|
|
global TARGET_FILE
|
|
global TARGET_IMAGE
|
|
global TARGET_VIDEO
|
|
|
|
is_target_image = is_image(deepfuze.globals.target_path)
|
|
is_target_video = is_video(deepfuze.globals.target_path)
|
|
TARGET_FILE = gradio.File(
|
|
label = wording.get('uis.target_file'),
|
|
file_count = 'single',
|
|
file_types =
|
|
[
|
|
'.png',
|
|
'.jpg',
|
|
'.webp',
|
|
'.webm',
|
|
'.mp4'
|
|
],
|
|
value = deepfuze.globals.target_path if is_target_image or is_target_video else None
|
|
)
|
|
target_image_args =\
|
|
{
|
|
'show_label': False,
|
|
'visible': False
|
|
}
|
|
target_video_args =\
|
|
{
|
|
'show_label': False,
|
|
'visible': False
|
|
}
|
|
if is_target_image:
|
|
target_image_args['value'] = TARGET_FILE.value['name']
|
|
target_image_args['visible'] = True
|
|
if is_target_video:
|
|
if get_file_size(deepfuze.globals.target_path) > FILE_SIZE_LIMIT:
|
|
preview_vision_frame = normalize_frame_color(get_video_frame(deepfuze.globals.target_path))
|
|
target_image_args['value'] = preview_vision_frame
|
|
target_image_args['visible'] = True
|
|
else:
|
|
target_video_args['value'] = TARGET_FILE.value['name']
|
|
target_video_args['visible'] = True
|
|
TARGET_IMAGE = gradio.Image(**target_image_args)
|
|
TARGET_VIDEO = gradio.Video(**target_video_args)
|
|
register_ui_component('target_image', TARGET_IMAGE)
|
|
register_ui_component('target_video', TARGET_VIDEO)
|
|
|
|
|
|
def listen() -> None:
|
|
TARGET_FILE.change(update, inputs = TARGET_FILE, outputs = [ TARGET_IMAGE, TARGET_VIDEO ])
|
|
|
|
|
|
def update(file : File) -> Tuple[gradio.Image, gradio.Video]:
|
|
clear_reference_faces()
|
|
clear_static_faces()
|
|
if file and is_image(file.name):
|
|
deepfuze.globals.target_path = file.name
|
|
return gradio.Image(value = file.name, visible = True), gradio.Video(value = None, visible = False)
|
|
if file and is_video(file.name):
|
|
deepfuze.globals.target_path = file.name
|
|
if get_file_size(file.name) > FILE_SIZE_LIMIT:
|
|
preview_vision_frame = normalize_frame_color(get_video_frame(file.name))
|
|
return gradio.Image(value = preview_vision_frame, visible = True), gradio.Video(value = None, visible = False)
|
|
return gradio.Image(value = None, visible = False), gradio.Video(value = file.name, visible = True)
|
|
deepfuze.globals.target_path = None
|
|
return gradio.Image(value = None, visible = False), gradio.Video(value = None, visible = False)
|