Upload 5 files

uis/choices.py

@@ -0,0 +1,23 @@
+from typing import Dict, List
+
+from facefusion.types import Color, WebcamMode
+from facefusion.uis.types import JobManagerAction, JobRunnerAction, PreviewMode
+
+job_manager_actions : List[JobManagerAction] = [ 'job-create', 'job-submit', 'job-delete', 'job-add-step', 'job-remix-step', 'job-insert-step', 'job-remove-step' ]
+job_runner_actions : List[JobRunnerAction] = [ 'job-run', 'job-run-all', 'job-retry', 'job-retry-all' ]
+
+preview_modes : List[PreviewMode] = [ 'default', 'frame-by-frame', 'face-by-face' ]
+preview_resolutions : List[str] = [ '512x512', '768x768', '1024x1024' ]
+
+webcam_modes : List[WebcamMode] = [ 'inline', 'udp', 'v4l2' ]
+webcam_resolutions : List[str] = [ '320x240', '640x480', '800x600', '1024x768', '1280x720', '1280x960', '1920x1080' ]
+
+background_remover_fill_colors : Dict[str, Color] =\
+{
+	'red' : (255, 0, 0, 255),
+	'green' : (0, 255, 0, 255),
+	'blue' : (0, 0, 255, 255),
+	'black' : (0, 0, 0, 255),
+	'white' : (255, 255, 255, 255),
+	'alpha' : (0, 0, 0, 0)
+}

uis/core.py

@@ -0,0 +1,200 @@
+import importlib
+import logging
+import os
+import warnings
+from types import ModuleType
+from typing import Any, Dict, List, Optional
+
+import gradio
+from gradio.themes import Size
+
+import facefusion.uis.overrides as uis_overrides
+from facefusion import logger, metadata, state_manager, translator
+from facefusion.exit_helper import hard_exit
+from facefusion.filesystem import resolve_relative_path
+from facefusion.uis.types import Component, ComponentName
+
+UI_COMPONENTS: Dict[ComponentName, Component] = {}
+UI_LAYOUT_MODULES : List[ModuleType] = []
+UI_LAYOUT_METHODS =\
+[
+	'pre_check',
+	'render',
+	'listen',
+	'run'
+]
+
+
+def load_ui_layout_module(ui_layout : str) -> Any:
+	try:
+		ui_layout_module = importlib.import_module('facefusion.uis.layouts.' + ui_layout)
+		for method_name in UI_LAYOUT_METHODS:
+			if not hasattr(ui_layout_module, method_name):
+				raise NotImplementedError
+	except ModuleNotFoundError as exception:
+		logger.error(translator.get('ui_layout_not_loaded').format(ui_layout = ui_layout), __name__)
+		logger.debug(exception.msg, __name__)
+		hard_exit(1)
+	except NotImplementedError:
+		logger.error(translator.get('ui_layout_not_implemented').format(ui_layout = ui_layout), __name__)
+		hard_exit(1)
+	return ui_layout_module
+
+
+def get_ui_layouts_modules(ui_layouts : List[str]) -> List[ModuleType]:
+	if not UI_LAYOUT_MODULES:
+		for ui_layout in ui_layouts:
+			ui_layout_module = load_ui_layout_module(ui_layout)
+			UI_LAYOUT_MODULES.append(ui_layout_module)
+	return UI_LAYOUT_MODULES
+
+
+def get_ui_component(component_name : ComponentName) -> Optional[Component]:
+	if component_name in UI_COMPONENTS:
+		return UI_COMPONENTS[component_name]
+	return None
+
+
+def get_ui_components(component_names : List[ComponentName]) -> Optional[List[Component]]:
+	ui_components = []
+
+	for component_name in component_names:
+		component = get_ui_component(component_name)
+		if component:
+			ui_components.append(component)
+	return ui_components
+
+
+def register_ui_component(component_name : ComponentName, component: Component) -> None:
+	UI_COMPONENTS[component_name] = component
+
+
+def init() -> None:
+	os.environ['GRADIO_ANALYTICS_ENABLED'] = '0'
+	os.environ['GRADIO_TEMP_DIR'] = os.path.join(state_manager.get_item('temp_path'), 'gradio')
+
+	logging.getLogger('asyncio').setLevel(logging.CRITICAL)
+	warnings.filterwarnings('ignore', category = UserWarning, module = 'gradio')
+	gradio.processing_utils._check_allowed = uis_overrides.mock
+	gradio.processing_utils.convert_video_to_playable_mp4 = uis_overrides.convert_video_to_playable_mp4
+	gradio.components.Number.raise_if_out_of_bounds = uis_overrides.mock
+
+
+def launch() -> None:
+	ui_layouts_total = len(state_manager.get_item('ui_layouts'))
+	with gradio.Blocks(theme = get_theme(), css = get_css(), title = metadata.get('name') + ' ' + metadata.get('version'), fill_width = True) as ui:
+		for ui_layout in state_manager.get_item('ui_layouts'):
+			ui_layout_module = load_ui_layout_module(ui_layout)
+
+			if ui_layouts_total > 1:
+				with gradio.Tab(ui_layout):
+					ui_layout_module.render()
+					ui_layout_module.listen()
+			else:
+				ui_layout_module.render()
+				ui_layout_module.listen()
+
+	for ui_layout in state_manager.get_item('ui_layouts'):
+		ui_layout_module = load_ui_layout_module(ui_layout)
+		ui_layout_module.run(ui)
+
+
+def get_theme() -> gradio.Theme:
+	return gradio.themes.Base(
+		primary_hue = gradio.themes.colors.red,
+		secondary_hue = gradio.themes.Color(
+			name = 'neutral',
+			c50 = '#fafafa',
+			c100 = '#f5f5f5',
+			c200 = '#e5e5e5',
+			c300 = '#d4d4d4',
+			c400 = '#a3a3a3',
+			c500 = '#737373',
+			c600 = '#525252',
+			c700 = '#404040',
+			c800 = '#262626',
+			c900 = '#212121',
+			c950 = '#171717',
+		),
+		radius_size = Size(
+			xxs = '0.375rem',
+			xs = '0.375rem',
+			sm = '0.375rem',
+			md = '0.375rem',
+			lg = '0.375rem',
+			xl = '0.375rem',
+			xxl = '0.375rem',
+		),
+		font = gradio.themes.GoogleFont('Open Sans')
+	).set(
+		color_accent = 'transparent',
+		color_accent_soft = 'transparent',
+		color_accent_soft_dark = 'transparent',
+		background_fill_primary = '*neutral_100',
+		background_fill_primary_dark = '*neutral_950',
+		background_fill_secondary = '*neutral_50',
+		background_fill_secondary_dark = '*neutral_800',
+		block_background_fill = 'white',
+		block_background_fill_dark = '*neutral_900',
+		block_border_width = '0',
+		block_label_background_fill = '*neutral_100',
+		block_label_background_fill_dark = '*neutral_800',
+		block_label_border_width = 'none',
+		block_label_margin = '0.5rem',
+		block_label_radius = '*radius_md',
+		block_label_text_color = '*neutral_700',
+		block_label_text_size = '*text_sm',
+		block_label_text_color_dark = 'white',
+		block_label_text_weight = '600',
+		block_title_background_fill = '*neutral_100',
+		block_title_background_fill_dark = '*neutral_800',
+		block_title_padding = '*block_label_padding',
+		block_title_radius = '*block_label_radius',
+		block_title_text_color = '*neutral_700',
+		block_title_text_size = '*text_sm',
+		block_title_text_weight = '600',
+		block_padding = '0.5rem',
+		border_color_accent = 'transparent',
+		border_color_accent_dark = 'transparent',
+		border_color_accent_subdued = 'transparent',
+		border_color_accent_subdued_dark = 'transparent',
+		border_color_primary = 'transparent',
+		border_color_primary_dark = 'transparent',
+		button_large_padding = '2rem 0.5rem',
+		button_large_text_weight = 'normal',
+		button_primary_background_fill = '*primary_500',
+		button_primary_background_fill_dark = '*primary_600',
+		button_primary_text_color = 'white',
+		button_secondary_background_fill = 'white',
+		button_secondary_background_fill_dark = '*neutral_800',
+		button_secondary_background_fill_hover = 'white',
+		button_secondary_background_fill_hover_dark = '*neutral_800',
+		button_secondary_text_color = '*neutral_800',
+		button_small_padding = '0.75rem',
+		button_small_text_size = '0.875rem',
+		checkbox_background_color = '*neutral_200',
+		checkbox_background_color_dark = '*neutral_900',
+		checkbox_background_color_selected = '*primary_600',
+		checkbox_background_color_selected_dark = '*primary_700',
+		checkbox_label_background_fill = '*neutral_50',
+		checkbox_label_background_fill_dark = '*neutral_800',
+		checkbox_label_background_fill_hover = '*neutral_50',
+		checkbox_label_background_fill_hover_dark = '*neutral_800',
+		checkbox_label_background_fill_selected = '*primary_500',
+		checkbox_label_background_fill_selected_dark = '*primary_600',
+		checkbox_label_text_color_selected = 'white',
+		error_background_fill = 'white',
+		error_background_fill_dark = '*neutral_900',
+		error_text_color = '*primary_500',
+		error_text_color_dark = '*primary_600',
+		input_background_fill = '*neutral_50',
+		input_background_fill_dark = '*neutral_800',
+		shadow_drop = 'none',
+		slider_color = '*primary_500',
+		slider_color_dark = '*primary_600'
+	)
+
+
+def get_css() -> str:
+	overrides_css_path = resolve_relative_path('uis/assets/overrides.css')
+	return open(overrides_css_path).read()

uis/overrides.py

@@ -0,0 +1,31 @@
+from facefusion import ffmpeg_builder
+from facefusion.ffmpeg import run_ffmpeg
+from facefusion.filesystem import get_file_size
+from facefusion.temp_helper import create_temp_directory, get_temp_file_path
+from facefusion.uis.types import MockArgs
+
+
+def convert_video_to_playable_mp4(video_path : str) -> str:
+	video_file_size = get_file_size(video_path)
+	max_file_size = 512 * 1024 * 1024
+
+	create_temp_directory(video_path)
+	temp_video_path = get_temp_file_path(video_path)
+	commands = ffmpeg_builder.set_input(video_path)
+
+	if video_file_size > max_file_size:
+		commands.extend(ffmpeg_builder.set_video_duration(10))
+
+	commands.extend(ffmpeg_builder.force_output(temp_video_path))
+
+	process = run_ffmpeg(commands)
+	process.communicate()
+
+	if process.returncode == 0:
+		return temp_video_path
+
+	return video_path
+
+
+def mock(*args : MockArgs, **kwargs : MockArgs) -> None:
+	pass

uis/types.py

@@ -0,0 +1,104 @@
+from typing import Any, Dict, IO, Literal, TypeAlias
+
+File : TypeAlias = IO[Any]
+ComponentName = Literal\
+[
+	'age_modifier_direction_slider',
+	'age_modifier_model_dropdown',
+	'background_remover_model_dropdown',
+	'background_remover_fill_color_red_number',
+	'background_remover_fill_color_green_number',
+	'background_remover_fill_color_blue_number',
+	'background_remover_fill_color_alpha_number',
+	'background_remover_despill_color_red_number',
+	'background_remover_despill_color_green_number',
+	'background_remover_despill_color_blue_number',
+	'background_remover_despill_color_alpha_number',
+	'deep_swapper_model_dropdown',
+	'deep_swapper_morph_slider',
+	'expression_restorer_factor_slider',
+	'expression_restorer_model_dropdown',
+	'expression_restorer_areas_checkbox_group',
+	'face_debugger_items_checkbox_group',
+	'face_detector_angles_checkbox_group',
+	'face_detector_model_dropdown',
+	'face_detector_margin_slider',
+	'face_detector_score_slider',
+	'face_detector_size_dropdown',
+	'face_editor_eyebrow_direction_slider',
+	'face_editor_eye_gaze_horizontal_slider',
+	'face_editor_eye_gaze_vertical_slider',
+	'face_editor_eye_open_ratio_slider',
+	'face_editor_head_pitch_slider',
+	'face_editor_head_roll_slider',
+	'face_editor_head_yaw_slider',
+	'face_editor_lip_open_ratio_slider',
+	'face_editor_model_dropdown',
+	'face_editor_mouth_grim_slider',
+	'face_editor_mouth_position_horizontal_slider',
+	'face_editor_mouth_position_vertical_slider',
+	'face_editor_mouth_pout_slider',
+	'face_editor_mouth_purse_slider',
+	'face_editor_mouth_smile_slider',
+	'face_enhancer_blend_slider',
+	'face_enhancer_model_dropdown',
+	'face_enhancer_weight_slider',
+	'face_landmarker_model_dropdown',
+	'face_landmarker_score_slider',
+	'face_mask_types_checkbox_group',
+	'face_mask_areas_checkbox_group',
+	'face_mask_regions_checkbox_group',
+	'face_mask_blur_slider',
+	'face_mask_padding_bottom_slider',
+	'face_mask_padding_left_slider',
+	'face_mask_padding_right_slider',
+	'face_mask_padding_top_slider',
+	'face_selector_age_range_slider',
+	'face_selector_gender_dropdown',
+	'face_selector_mode_dropdown',
+	'face_selector_order_dropdown',
+	'face_selector_race_dropdown',
+	'face_tracker_score_slider',
+	'face_swapper_model_dropdown',
+	'face_swapper_pixel_boost_dropdown',
+	'face_swapper_weight_slider',
+	'face_occluder_model_dropdown',
+	'face_parser_model_dropdown',
+	'voice_extractor_model_dropdown',
+	'frame_colorizer_blend_slider',
+	'frame_colorizer_model_dropdown',
+	'frame_colorizer_size_dropdown',
+	'frame_enhancer_blend_slider',
+	'frame_enhancer_model_dropdown',
+	'job_list_job_status_checkbox_group',
+	'lip_syncer_model_dropdown',
+	'lip_syncer_weight_slider',
+	'output_image',
+	'output_video',
+	'output_video_fps_slider',
+	'preview_image',
+	'preview_frame_slider',
+	'preview_mode_dropdown',
+	'preview_resolution_dropdown',
+	'processors_checkbox_group',
+	'reference_face_distance_slider',
+	'reference_face_position_gallery',
+	'source_audio',
+	'source_image',
+	'target_image',
+	'target_video',
+	'ui_workflow_dropdown',
+	'webcam_device_id_dropdown',
+	'webcam_fps_slider',
+	'webcam_mode_radio',
+	'webcam_resolution_dropdown'
+]
+Component : TypeAlias = Any
+ComponentOptions : TypeAlias = Dict[str, Any]
+
+JobManagerAction = Literal['job-create', 'job-submit', 'job-delete', 'job-add-step', 'job-remix-step', 'job-insert-step', 'job-remove-step']
+JobRunnerAction = Literal['job-run', 'job-run-all', 'job-retry', 'job-retry-all']
+
+PreviewMode = Literal[ 'default', 'frame-by-frame', 'face-by-face' ]
+
+MockArgs : TypeAlias = Any

uis/ui_helper.py

@@ -0,0 +1,26 @@
+import hashlib
+import os
+from typing import Optional
+
+from facefusion import state_manager
+from facefusion.filesystem import get_file_extension, is_image, is_video
+
+
+def convert_int_none(value : int) -> Optional[int]:
+	if value == 'none':
+		return None
+	return value
+
+
+def convert_str_none(value : str) -> Optional[str]:
+	if value == 'none':
+		return None
+	return value
+
+
+def suggest_output_path(output_directory_path : str, target_path : str) -> Optional[str]:
+	if is_image(target_path) or is_video(target_path):
+		output_file_name = hashlib.sha1(str(state_manager.get_state()).encode()).hexdigest()[:8]
+		target_file_extension = get_file_extension(target_path)
+		return os.path.join(output_directory_path, output_file_name + target_file_extension)
+	return None