6f0675030e
* Purge filetype dependency, Rename file_extension to file_format, Introduce custom format detections * Changed a lot * Purge filetype dependency, Rename file_extension to file_format, Introduce custom format detections * Fix stuff * Fix stuff * Simplify all the is_ and has_ methods * Simplify all the is_ and has_ methods * Use the new helper on more places * Introduce are_ next to is_ and has_ * Get rid of the type-ignores * Add more video types
60 lines
2.3 KiB
Python
60 lines
2.3 KiB
Python
from typing import List, Optional, Tuple
|
|
|
|
import gradio
|
|
|
|
from facefusion import state_manager, wording
|
|
from facefusion.common_helper import get_first
|
|
from facefusion.filesystem import filter_audio_paths, filter_image_paths, has_audio, has_image
|
|
from facefusion.uis.core import register_ui_component
|
|
from facefusion.uis.typing import File
|
|
|
|
SOURCE_FILE : Optional[gradio.File] = None
|
|
SOURCE_AUDIO : Optional[gradio.Audio] = None
|
|
SOURCE_IMAGE : Optional[gradio.Image] = None
|
|
|
|
|
|
def render() -> None:
|
|
global SOURCE_FILE
|
|
global SOURCE_AUDIO
|
|
global SOURCE_IMAGE
|
|
|
|
has_source_audio = has_audio(state_manager.get_item('source_paths'))
|
|
has_source_image = has_image(state_manager.get_item('source_paths'))
|
|
SOURCE_FILE = gradio.File(
|
|
label = wording.get('uis.source_file'),
|
|
file_count = 'multiple',
|
|
value = state_manager.get_item('source_paths') if has_source_audio or has_source_image else None
|
|
)
|
|
source_file_names = [ source_file_value.get('path') for source_file_value in SOURCE_FILE.value ] if SOURCE_FILE.value else None
|
|
source_audio_path = get_first(filter_audio_paths(source_file_names))
|
|
source_image_path = get_first(filter_image_paths(source_file_names))
|
|
SOURCE_AUDIO = gradio.Audio(
|
|
value = source_audio_path if has_source_audio else None,
|
|
visible = has_source_audio,
|
|
show_label = False
|
|
)
|
|
SOURCE_IMAGE = gradio.Image(
|
|
value = source_image_path if has_source_image else None,
|
|
visible = has_source_image,
|
|
show_label = False
|
|
)
|
|
register_ui_component('source_audio', SOURCE_AUDIO)
|
|
register_ui_component('source_image', SOURCE_IMAGE)
|
|
|
|
|
|
def listen() -> None:
|
|
SOURCE_FILE.change(update, inputs = SOURCE_FILE, outputs = [ SOURCE_AUDIO, SOURCE_IMAGE ])
|
|
|
|
|
|
def update(files : List[File]) -> Tuple[gradio.Audio, gradio.Image]:
|
|
file_names = [ file.name for file in files ] if files else None
|
|
has_source_audio = has_audio(file_names)
|
|
has_source_image = has_image(file_names)
|
|
if has_source_audio or has_source_image:
|
|
source_audio_path = get_first(filter_audio_paths(file_names))
|
|
source_image_path = get_first(filter_image_paths(file_names))
|
|
state_manager.set_item('source_paths', file_names)
|
|
return gradio.Audio(value = source_audio_path, visible = has_source_audio), gradio.Image(value = source_image_path, visible = has_source_image)
|
|
state_manager.clear_item('source_paths')
|
|
return gradio.Audio(value = None, visible = False), gradio.Image(value = None, visible = False)
|