deepfuze

2024-06-13 07:56:13 +05:30
commit 47d3520c19
184 changed files with 10075 additions and 0 deletions
@@ -0,0 +1,83 @@
+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)