jones/facefusion
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Remove Gradio preview interception for now

Browse Source
This commit is contained in:
henryruhs
2025-01-28 10:56:06 +01:00
parent cd6d2012f6
commit c363faa757
1 changed files with 3 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
facefusion/uis/components/target.py
View File

@@ -4,12 +4,9 @@ import gradio
from facefusion import state_manager, wording from facefusion import state_manager, wording
from facefusion.face_store import clear_reference_faces, clear_static_faces from facefusion.face_store import clear_reference_faces, clear_static_faces
from facefusion.filesystem import get_file_size, is_image, is_video from facefusion.filesystem import is_image, is_video
from facefusion.uis.core import register_ui_component from facefusion.uis.core import register_ui_component
from facefusion.uis.typing import ComponentOptions, File from facefusion.uis.typing import ComponentOptions, File
from facefusion.vision import normalize_frame_color, read_video_frame
FILE_SIZE_LIMIT = 512 * 1024 * 1024
TARGET_FILE : Optional[gradio.File] = None TARGET_FILE : Optional[gradio.File] = None
TARGET_IMAGE : Optional[gradio.Image] = None TARGET_IMAGE : Optional[gradio.Image] = None
@@ -42,13 +39,8 @@ def render() -> None:
target_image_options['value'] = TARGET_FILE.value.get('path') target_image_options['value'] = TARGET_FILE.value.get('path')
target_image_options['visible'] = True target_image_options['visible'] = True
if is_target_video: if is_target_video:
if get_file_size(state_manager.get_item('target_path')) > FILE_SIZE_LIMIT: target_video_options['value'] = TARGET_FILE.value.get('path')
preview_vision_frame = normalize_frame_color(read_video_frame(state_manager.get_item('target_path'))) target_video_options['visible'] = True
target_image_options['value'] = preview_vision_frame
target_image_options['visible'] = True
else:
target_video_options['value'] = TARGET_FILE.value.get('path')
target_video_options['visible'] = True
TARGET_IMAGE = gradio.Image(**target_image_options) TARGET_IMAGE = gradio.Image(**target_image_options)
TARGET_VIDEO = gradio.Video(**target_video_options) TARGET_VIDEO = gradio.Video(**target_video_options)
register_ui_component('target_image', TARGET_IMAGE) register_ui_component('target_image', TARGET_IMAGE)
@@ -69,9 +61,6 @@ def update(file : File) -> Tuple[gradio.Image, gradio.Video]:
if file and is_video(file.name): if file and is_video(file.name):
state_manager.set_item('target_path', file.name) state_manager.set_item('target_path', file.name)
if get_file_size(file.name) > FILE_SIZE_LIMIT:
preview_vision_frame = normalize_frame_color(read_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) return gradio.Image(value = None, visible = False), gradio.Video(value = file.name, visible = True)
state_manager.clear_item('target_path') state_manager.clear_item('target_path')