deepfuze

2024-06-13 07:56:13 +05:30
commit 47d3520c19
184 changed files with 10075 additions and 0 deletions
@@ -0,0 +1,40 @@
+from typing import List, Optional
+import gradio
+
+import deepfuze.globals
+from deepfuze import wording
+from deepfuze.processors.frame.core import load_frame_processor_module, clear_frame_processors_modules
+from deepfuze.filesystem import list_directory
+from deepfuze.uis.core import register_ui_component
+
+FRAME_PROCESSORS_CHECKBOX_GROUP : Optional[gradio.CheckboxGroup] = None
+
+
+def render() -> None:
+	global FRAME_PROCESSORS_CHECKBOX_GROUP
+
+	FRAME_PROCESSORS_CHECKBOX_GROUP = gradio.CheckboxGroup(
+		label = wording.get('uis.frame_processors_checkbox_group'),
+		choices = sort_frame_processors(deepfuze.globals.frame_processors),
+		value = deepfuze.globals.frame_processors
+	)
+	register_ui_component('frame_processors_checkbox_group', FRAME_PROCESSORS_CHECKBOX_GROUP)
+
+
+def listen() -> None:
+	FRAME_PROCESSORS_CHECKBOX_GROUP.change(update_frame_processors, inputs = FRAME_PROCESSORS_CHECKBOX_GROUP, outputs = FRAME_PROCESSORS_CHECKBOX_GROUP)
+
+
+def update_frame_processors(frame_processors : List[str]) -> gradio.CheckboxGroup:
+	deepfuze.globals.frame_processors = frame_processors
+	clear_frame_processors_modules()
+	for frame_processor in frame_processors:
+		frame_processor_module = load_frame_processor_module(frame_processor)
+		if not frame_processor_module.pre_check():
+			return gradio.CheckboxGroup()
+	return gradio.CheckboxGroup(value = deepfuze.globals.frame_processors, choices = sort_frame_processors(deepfuze.globals.frame_processors))
+
+
+def sort_frame_processors(frame_processors : List[str]) -> list[str]:
+	available_frame_processors = list_directory('facefusion/processors/frame/modules')
+	return sorted(available_frame_processors, key = lambda frame_processor : frame_processors.index(frame_processor) if frame_processor in frame_processors else len(frame_processors))