mirror of
https://github.com/facefusion/facefusion.git
synced 2026-08-24 09:42:29 +02:00
Henry RuhsandSumit
82eaf76da8
* Cosmetic changes * Cosmetic changes * Run single warm up for the benchmark suite * Use latest version of Gradio * More testing * Introduce basic installer * Fix typo * Move more to installer file * Fix the installer with the uninstall all trick * Adjust wording * Fix coreml in installer * Allow Pyhton 3.9 * Add VENV to installer * Just some cosmetics * Just some cosmetics * Dedicated headless mode, Refine API of UI layouts * Use --headless for pytest * Fix testing for Windows * Normalize output path that lacks extension * Fix CI for Windows * Fix CI for Windows * UI to change output path * Add conda support for the installer * Improve installer quite a bit * Drop conda support * Install community wheels for coreml silicon * Improve output video component * Fix silicon wheel downloading * Remove venv from installer as we cannot activate via subprocess * Use join to create wheel name * Refine the output path normalization * Refine the output path normalization * Introduce ProcessMode and rename some methods * Introduce ProcessMode and rename some methods * Basic webcam integration and open_ffmpeg() * Basic webcam integration part2 * Benchmark resolutions now selectable * Rename benchmark resolution back to benchmark runs * Fix repeating output path in UI * Keep output_path untouched if not resolvable * Add more cases to normalize output path * None for those tests that don't take source path into account * Finish basic webcam integration, UI layout now with custom run() * Fix CI and hide link in webcam UI * Cosmetics on webcam UI * Move get_device to utilities * Fix CI * Introduce output-image-quality, Show and hide UI according to target media type * Benchmark with partial result updates * fix: trim frame sliders not appearing after draggin video * fix: output and temp frame setting inputs not appearing * Fix: set increased update delay to 250ms to let Gradio update conditional inputs properly * Reverted .gitignore * Adjust timings * Remove timeout hacks and get fully event driven * Update dependencies * Update dependencies * Revert NSFW library, Conditional unset trim args * Face selector works better on preview slider release * Add limit resources to UI * Introduce vision.py for all CV2 operations, Rename some methods * Add restoring audio failed * Decouple updates for preview image and preview frame slider, Move reduce_preview_frame to vision * Refactor detect_fps based on JSON output * Only webcam when open * More conditions to vision.py * Add udp and v4l2 streaming to webcam UI * Detect v4l2 device to be used * Refactor code a bit * Use static max memory for UI * Fix CI * Looks stable to me * Update preview * Update preview --------- Co-authored-by: Sumit <vizsumit@gmail.com>
64 lines
2.0 KiB
Python
64 lines
2.0 KiB
Python
from typing import Any, IO, Tuple, Optional
|
|
import gradio
|
|
|
|
import facefusion.globals
|
|
from facefusion import wording
|
|
from facefusion.face_reference import clear_face_reference
|
|
from facefusion.uis import core as ui
|
|
from facefusion.uis.typing import Update
|
|
from facefusion.utilities import is_image, is_video
|
|
|
|
TARGET_FILE : Optional[gradio.File] = None
|
|
TARGET_IMAGE : Optional[gradio.Image] = None
|
|
TARGET_VIDEO : Optional[gradio.Video] = None
|
|
|
|
|
|
def render() -> None:
|
|
global TARGET_FILE
|
|
global TARGET_IMAGE
|
|
global TARGET_VIDEO
|
|
|
|
with gradio.Box():
|
|
is_target_image = is_image(facefusion.globals.target_path)
|
|
is_target_video = is_video(facefusion.globals.target_path)
|
|
TARGET_FILE = gradio.File(
|
|
label = wording.get('target_file_label'),
|
|
file_count = 'single',
|
|
file_types =
|
|
[
|
|
'.png',
|
|
'.jpg',
|
|
'.webp',
|
|
'.mp4'
|
|
],
|
|
value = facefusion.globals.target_path if is_target_image or is_target_video else None
|
|
)
|
|
TARGET_IMAGE = gradio.Image(
|
|
value = TARGET_FILE.value['name'] if is_target_image else None,
|
|
visible = is_target_image,
|
|
show_label = False
|
|
)
|
|
TARGET_VIDEO = gradio.Video(
|
|
value = TARGET_FILE.value['name'] if is_target_video else None,
|
|
visible = is_target_video,
|
|
show_label = False
|
|
)
|
|
ui.register_component('target_image', TARGET_IMAGE)
|
|
ui.register_component('target_video', TARGET_VIDEO)
|
|
|
|
|
|
def listen() -> None:
|
|
TARGET_FILE.change(update, inputs = TARGET_FILE, outputs = [ TARGET_IMAGE, TARGET_VIDEO ])
|
|
|
|
|
|
def update(file : IO[Any]) -> Tuple[Update, Update]:
|
|
clear_face_reference()
|
|
if file and is_image(file.name):
|
|
facefusion.globals.target_path = file.name
|
|
return gradio.update(value = file.name, visible = True), gradio.update(value = None, visible = False)
|
|
if file and is_video(file.name):
|
|
facefusion.globals.target_path = file.name
|
|
return gradio.update(value = None, visible = False), gradio.update(value = file.name, visible = True)
|
|
facefusion.globals.target_path = None
|
|
return gradio.update(value = None, visible = False), gradio.update(value = None, visible = False)
|