Add lip-syncer support to source component

This commit is contained in:
henryruhs
2024-02-01 12:59:23 +01:00
parent 3c8ad63d25
commit cc915171ea
+5 -4
View File
@@ -39,8 +39,7 @@ def render() -> None:
SOURCE_AUDIO = gradio.Audio( SOURCE_AUDIO = gradio.Audio(
value = source_audio_path if are_source_audios else None, value = source_audio_path if are_source_audios else None,
visible = are_source_audios, visible = are_source_audios,
show_label = False, show_label = False
show_download_button = False
) )
SOURCE_IMAGE = gradio.Image( SOURCE_IMAGE = gradio.Image(
value = source_image_path if are_source_images else None, value = source_image_path if are_source_images else None,
@@ -56,10 +55,12 @@ def listen() -> None:
def update(files : List[File]) -> Tuple[gradio.Audio, gradio.Image]: def update(files : List[File]) -> Tuple[gradio.Audio, gradio.Image]:
file_names = [ file.name for file in files ] if files else None file_names = [ file.name for file in files ] if files else None
if are_audios(file_names) or are_images(file_names): 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:
facefusion.globals.source_paths = file_names facefusion.globals.source_paths = file_names
audio_path = get_first_item(filter_audio_paths(facefusion.globals.source_paths)) audio_path = get_first_item(filter_audio_paths(facefusion.globals.source_paths))
image_path = get_first_item(filter_image_paths(facefusion.globals.source_paths)) image_path = get_first_item(filter_image_paths(facefusion.globals.source_paths))
return gradio.Audio(value = audio_path, visible = True), gradio.Image(value = image_path, visible = True) return gradio.Audio(value = audio_path, visible = are_source_audios), gradio.Image(value = image_path, visible = are_source_images)
facefusion.globals.source_paths = None facefusion.globals.source_paths = None
return gradio.Audio(value = None, visible = False), gradio.Image(value = None, visible = False) return gradio.Audio(value = None, visible = False), gradio.Image(value = None, visible = False)