mirror of
https://github.com/facefusion/facefusion.git
synced 2026-04-30 05:17:49 +02:00
henryruhs
49 lines
1.2 KiB
Python
49 lines
1.2 KiB
Python
from typing import Any, IO, Optional
|
|
import gradio
|
|
|
|
import facefusion.globals
|
|
from facefusion import wording
|
|
from facefusion.uis import core as ui
|
|
from facefusion.uis.typing import Update
|
|
from facefusion.utilities import is_image
|
|
|
|
SOURCE_FILE : Optional[gradio.File] = None
|
|
SOURCE_IMAGE : Optional[gradio.Image] = None
|
|
|
|
|
|
def render() -> None:
|
|
global SOURCE_FILE
|
|
global SOURCE_IMAGE
|
|
|
|
with gradio.Box():
|
|
is_source_image = is_image(facefusion.globals.source_path)
|
|
SOURCE_FILE = gradio.File(
|
|
file_count = 'single',
|
|
file_types=
|
|
[
|
|
'.png',
|
|
'.jpg',
|
|
'.webp'
|
|
],
|
|
label = wording.get('source_file_label'),
|
|
value = facefusion.globals.source_path if is_source_image else None
|
|
)
|
|
ui.register_component('source_file', SOURCE_FILE)
|
|
SOURCE_IMAGE = gradio.Image(
|
|
value = SOURCE_FILE.value['name'] if is_source_image else None,
|
|
visible = is_source_image,
|
|
show_label = False
|
|
)
|
|
|
|
|
|
def listen() -> None:
|
|
SOURCE_FILE.change(update, inputs = SOURCE_FILE, outputs = SOURCE_IMAGE)
|
|
|
|
|
|
def update(file: IO[Any]) -> Update:
|
|
if file and is_image(file.name):
|
|
facefusion.globals.source_path = file.name
|
|
return gradio.update(value = file.name, visible = True)
|
|
facefusion.globals.source_path = None
|
|
return gradio.update(value = None, visible = False)
|