CalvinBackup/facefusion
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-04-26 11:26:41 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix the check in the source component

Browse Source
This commit is contained in:
henryruhs
2024-02-01 14:08:18 +01:00
parent cc915171ea
commit 697266fda2
4 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
facefusion/uis/components/output_options.py
-1
View File
@@ -89,7 +89,6 @@ def listen() -> None:
OUTPUT_VIDEO_FPS_SLIDER.change(update_output_video_fps, inputs = OUTPUT_VIDEO_FPS_SLIDER)
multi_component_names : List[ComponentName] =\
[
'source_image',
'target_image',
'target_video'
]
facefusion/uis/components/preview.py
+1
View File
@@ -62,6 +62,7 @@ def listen() -> None:
PREVIEW_FRAME_SLIDER.release(update_preview_image, inputs = PREVIEW_FRAME_SLIDER, outputs = PREVIEW_IMAGE)
multi_one_component_names : List[ComponentName] =\
[
'source_audio',
'source_image',
'target_image',
'target_video'
facefusion/uis/components/source.py
+7 -6
View File
@@ -46,6 +46,7 @@ def render() -> None:
visible = are_source_images,
show_label = False
)
register_ui_component('source_audio', SOURCE_AUDIO)
register_ui_component('source_image', SOURCE_IMAGE)
@@ -55,12 +56,12 @@ def listen() -> None:
def update(files : List[File]) -> Tuple[gradio.Audio, gradio.Image]:
file_names = [ file.name for file in files ] if files else None
are_source_audios = are_audios(facefusion.globals.source_paths)
are_source_images = are_images(facefusion.globals.source_paths)
if are_source_audios or are_source_images:
audio_path = get_first_item(filter_audio_paths(file_names))
image_path = get_first_item(filter_image_paths(file_names))
has_audio = bool(audio_path)
has_image = bool(image_path)
if audio_path or image_path:
facefusion.globals.source_paths = file_names
audio_path = get_first_item(filter_audio_paths(facefusion.globals.source_paths))
image_path = get_first_item(filter_image_paths(facefusion.globals.source_paths))
return gradio.Audio(value = audio_path, visible = are_source_audios), gradio.Image(value = image_path, visible = are_source_images)
return gradio.Audio(value = audio_path, visible = has_audio), gradio.Image(value = image_path, visible = has_image)
facefusion.globals.source_paths = None
return gradio.Audio(value = None, visible = False), gradio.Image(value = None, visible = False)
facefusion/uis/typing.py
+1
View File
@@ -5,6 +5,7 @@ File = IO[Any]
Component = gradio.File or gradio.Image or gradio.Video or gradio.Slider
ComponentName = Literal\
[
'source_audio',
'source_image',
'target_image',
'target_video',