From 81a1986ef8ab4e5805922436a7db30f5dd9efb43 Mon Sep 17 00:00:00 2001 From: Kenneth Estanislao Date: Fri, 15 May 2026 16:33:27 +0800 Subject: [PATCH] Changed to pyqtUI Standardizing the UI from quickstart to github version --- modules/metadata.py | 2 +- modules/ui.py | 2784 +++++++++++++++++++++---------------------- requirements.txt | 3 +- run.py | 3 - 4 files changed, 1376 insertions(+), 1416 deletions(-) diff --git a/modules/metadata.py b/modules/metadata.py index bc6598d..9bc9ba9 100644 --- a/modules/metadata.py +++ b/modules/metadata.py @@ -1,3 +1,3 @@ name = 'Deep-Live-Cam' -version = '2.1.2' +version = '2.1.5' edition = 'GitHub Edition' \ No newline at end of file diff --git a/modules/ui.py b/modules/ui.py index c3c672b..35b0c13 100644 --- a/modules/ui.py +++ b/modules/ui.py @@ -1,77 +1,90 @@ +"""PySide6 UI for Deep-Live-Cam. + +Public API kept stable for the rest of the codebase: + init(start, destroy, lang) -> _Window + Returned object has .mainloop() that core.py calls. + update_status(text) + Thread-safe; routed through Qt signal when called off-UI. + check_and_ignore_nsfw(target, destroy=None) -> bool +""" + +from __future__ import annotations + import os -import webbrowser -import customtkinter as ctk -from typing import Callable, Tuple -import cv2 -from modules.gpu_processing import gpu_cvt_color, gpu_resize, gpu_flip -from PIL import Image, ImageOps -import time -import json +import platform import queue +import sys +import tempfile import threading +import time +import webbrowser +from typing import Callable, List, Optional, Tuple + +import cv2 import numpy as np import requests -import tempfile +from PIL import Image, ImageOps +from PySide6.QtCore import ( + QObject, + QThread, + QTimer, + Qt, + Signal, +) +from PySide6.QtGui import QImage, QPixmap +from PySide6.QtWidgets import ( + QApplication, + QCheckBox, + QComboBox, + QDialog, + QFileDialog, + QGridLayout, + QGroupBox, + QHBoxLayout, + QLabel, + QMainWindow, + QPushButton, + QScrollArea, + QSizePolicy, + QSlider, + QVBoxLayout, + QWidget, +) + import modules.globals import modules.metadata +from modules.capturer import get_video_frame, get_video_frame_total from modules.face_analyser import ( - get_one_face, - get_many_faces, - detect_one_face_fast, + add_blank_map, detect_many_faces_fast, + detect_one_face_fast, + get_one_face, get_unique_faces_from_target_image, get_unique_faces_from_target_video, - add_blank_map, has_valid_map, simplify_maps, ) -from modules.capturer import get_video_frame, get_video_frame_total +from modules.gettext import LanguageManager +from modules.gpu_processing import gpu_cvt_color, gpu_flip, gpu_resize from modules.processors.frame.core import get_frame_processors_modules from modules.utilities import ( + has_image_extension, is_image, is_video, - resolve_relative_path, - has_image_extension, ) from modules.video_capture import VideoCapturer -from modules.gettext import LanguageManager -from modules.ui_tooltip import ToolTip -from modules import globals -import platform if platform.system() == "Windows": from pygrabber.dshow_graph import FilterGraph -# --- Tk 9.0 compatibility patch --- -# In Tk 9.0, Menu.index("end") returns "" instead of raising TclError -# when the menu is empty. CustomTkinter's CTkOptionMenu doesn't handle -# this, causing crashes. This patch adds the missing guard. -try: - from customtkinter.windows.widgets.core_widget_classes import DropdownMenu as _DropdownMenu +import json - _original_add_menu_commands = _DropdownMenu._add_menu_commands - def _patched_add_menu_commands(self, *args, **kwargs): - try: - end_index = self._menu.index("end") - if end_index == "" or end_index is None: - return - except Exception: - pass - _original_add_menu_commands(self, *args, **kwargs) +# ─── constants ──────────────────────────────────────────────────────────── - _DropdownMenu._add_menu_commands = _patched_add_menu_commands -except (ImportError, AttributeError): - pass # CustomTkinter version doesn't have this class path -# --- End Tk 9.0 patch --- +ROOT_HEIGHT = 820 +ROOT_WIDTH = 640 -ROOT = None -POPUP = None -POPUP_LIVE = None -ROOT_HEIGHT = 800 -ROOT_WIDTH = 600 - -PREVIEW = None PREVIEW_MAX_HEIGHT = 700 PREVIEW_MAX_WIDTH = 1200 PREVIEW_DEFAULT_WIDTH = 640 @@ -79,52 +92,211 @@ PREVIEW_DEFAULT_HEIGHT = 360 POPUP_WIDTH = 750 POPUP_HEIGHT = 810 -POPUP_SCROLL_WIDTH = (740,) +POPUP_SCROLL_WIDTH = 720 POPUP_SCROLL_HEIGHT = 700 POPUP_LIVE_WIDTH = 900 POPUP_LIVE_HEIGHT = 820 -POPUP_LIVE_SCROLL_WIDTH = (890,) +POPUP_LIVE_SCROLL_WIDTH = 870 POPUP_LIVE_SCROLL_HEIGHT = 700 -MAPPER_PREVIEW_MAX_HEIGHT = 100 -MAPPER_PREVIEW_MAX_WIDTH = 100 - -DEFAULT_BUTTON_WIDTH = 200 -DEFAULT_BUTTON_HEIGHT = 40 - -RECENT_DIRECTORY_SOURCE = None -RECENT_DIRECTORY_TARGET = None -RECENT_DIRECTORY_OUTPUT = None - -_ = None -preview_label = None -preview_slider = None -source_label = None -target_label = None -status_label = None -popup_status_label = None -popup_status_label_live = None -source_label_dict = {} -source_label_dict_live = {} -target_label_dict_live = {} - -img_ft, vid_ft = modules.globals.file_types +MAPPER_PREVIEW_SIZE = 100 +SOURCE_TARGET_PREVIEW_SIZE = 200 -def init(start: Callable[[], None], destroy: Callable[[], None], lang: str) -> ctk.CTk: - global ROOT, PREVIEW, _ +# ─── modern dark stylesheet ─────────────────────────────────────────────── - lang_manager = LanguageManager(lang) - _ = lang_manager._ - ROOT = create_root(start, destroy) - PREVIEW = create_preview(ROOT) +QSS = """ +QMainWindow, QDialog { background-color: #1e1e1e; color: #e6e6e6; } +QWidget { color: #e6e6e6; font-family: "Segoe UI", "SF Pro Display", "Helvetica Neue", Arial, sans-serif; font-size: 11pt; } - return ROOT +QGroupBox { + background-color: #262626; + border: 1px solid #333333; + border-radius: 10px; + margin-top: 14px; + padding-top: 18px; + font-weight: 600; +} +QGroupBox::title { + subcontrol-origin: margin; + subcontrol-position: top left; + padding: 0 8px; + color: #9ec5ff; +} + +QPushButton { + background-color: #2d6cdf; + color: white; + border: none; + border-radius: 8px; + padding: 8px 16px; + font-weight: 600; +} +QPushButton:hover { background-color: #3a7af0; } +QPushButton:pressed{ background-color: #1d57c2; } +QPushButton:disabled { background-color: #444; color: #888; } +QPushButton#secondary { + background-color: #3a3a3a; +} +QPushButton#secondary:hover { background-color: #4a4a4a; } +QPushButton#danger { background-color: #c2412d; } +QPushButton#danger:hover { background-color: #d8523c; } + +QComboBox { + background-color: #2a2a2a; + border: 1px solid #404040; + border-radius: 6px; + padding: 6px 10px; + min-height: 24px; +} +QComboBox:hover { border-color: #2d6cdf; } +QComboBox QAbstractItemView { + background-color: #2a2a2a; + selection-background-color: #2d6cdf; + border: 1px solid #404040; +} + +QCheckBox { + spacing: 8px; + padding: 4px 0; +} +QCheckBox::indicator { + width: 36px; height: 18px; + border-radius: 9px; + background-color: #3a3a3a; +} +QCheckBox::indicator:checked { + background-color: #2d6cdf; +} + +QSlider::groove:horizontal { + height: 6px; + background: #3a3a3a; + border-radius: 3px; +} +QSlider::handle:horizontal { + background: #ffffff; + width: 16px; height: 16px; + margin: -5px 0; + border-radius: 8px; + border: 1px solid #cccccc; +} +QSlider::sub-page:horizontal { + background: #2d6cdf; + border-radius: 3px; +} + +QLabel#imageDrop { + background-color: #2a2a2a; + border: 2px dashed #444; + border-radius: 8px; +} +QLabel#statusLabel { + color: #b9b9b9; + font-size: 10pt; + font-style: italic; +} +QLabel#linkLabel { + color: #6ea8ff; + text-decoration: underline; +} + +QScrollArea { border: none; background: transparent; } + +QFrame#card { + background-color: #262626; + border-radius: 10px; +} +""" + + +# ─── module-level state ─────────────────────────────────────────────────── + +_APP: Optional[QApplication] = None +_MAIN: Optional["MainWindow"] = None +_PREVIEW: Optional["PreviewWindow"] = None +_WEBCAM_PREVIEW: Optional["WebcamPreviewWindow"] = None +_MAPPER: Optional["MapperDialog"] = None +_LIVE_MAPPER: Optional["LiveMapperDialog"] = None +_LANG: Optional[LanguageManager] = None +_BRIDGE: Optional["_UIBridge"] = None + + +def _(text: str) -> str: + """Translate via LanguageManager; falls back to identity.""" + if _LANG is None: + return text + return _LANG._(text) + + +# Preserve original cwd state for file dialogs. +_RECENT_SOURCE_DIR: Optional[str] = None +_RECENT_TARGET_DIR: Optional[str] = None +_RECENT_OUTPUT_DIR: Optional[str] = None + + +# ─── image utilities ───────────────────────────────────────────────────── + + +def fit_image_to_size(image, width: int, height: int): + """BGR ndarray → BGR ndarray scaled to fit within (width, height).""" + if width is None and height is None or width <= 0 or height <= 0: + return image + h, w = image.shape[:2] + ratio_w = width / w + ratio_h = height / h + ratio = min(ratio_w, ratio_h) + new_size = (max(1, int(w * ratio)), max(1, int(h * ratio))) + return gpu_resize(image, dsize=new_size) + + +def _bgr_to_qpixmap(bgr: np.ndarray) -> QPixmap: + """Zero-copy BGR ndarray → QPixmap.""" + h, w = bgr.shape[:2] + rgb = cv2.cvtColor(bgr, cv2.COLOR_BGR2RGB) + qimg = QImage(rgb.data, w, h, w * 3, QImage.Format.Format_RGB888).copy() + return QPixmap.fromImage(qimg) + + +def _pil_to_qpixmap(image: Image.Image) -> QPixmap: + """PIL.Image → QPixmap.""" + image = image.convert("RGBA") + data = image.tobytes("raw", "RGBA") + qimg = QImage(data, image.width, image.height, QImage.Format.Format_RGBA8888) + return QPixmap.fromImage(qimg.copy()) + + +def render_image_preview(image_path: str, size: Tuple[int, int]) -> QPixmap: + image = Image.open(image_path) + if size: + image = ImageOps.fit(image, size, Image.LANCZOS) + return _pil_to_qpixmap(image) + + +def render_video_preview( + video_path: str, size: Tuple[int, int], frame_number: int = 0 +) -> Optional[QPixmap]: + capture = cv2.VideoCapture(video_path) + try: + if frame_number: + capture.set(cv2.CAP_PROP_POS_FRAMES, frame_number) + has_frame, frame = capture.read() + if not has_frame: + return None + image = Image.fromarray(gpu_cvt_color(frame, cv2.COLOR_BGR2RGB)) + if size: + image = ImageOps.fit(image, size, Image.LANCZOS) + return _pil_to_qpixmap(image) + finally: + capture.release() + + +# ─── persistence ───────────────────────────────────────────────────────── def save_switch_states(): - switch_states = { + state = { "keep_fps": modules.globals.keep_fps, "keep_audio": modules.globals.keep_audio, "keep_frames": modules.globals.keep_frames, @@ -141,1424 +313,1216 @@ def save_switch_states(): "show_mouth_mask_box": modules.globals.show_mouth_mask_box, "mouth_mask_size": modules.globals.mouth_mask_size, } - with open("switch_states.json", "w") as f: - json.dump(switch_states, f) + try: + with open("switch_states.json", "w") as f: + json.dump(state, f) + except OSError: + pass def load_switch_states(): try: with open("switch_states.json", "r") as f: - switch_states = json.load(f) - modules.globals.keep_fps = switch_states.get("keep_fps", True) - modules.globals.keep_audio = switch_states.get("keep_audio", True) - modules.globals.keep_frames = switch_states.get("keep_frames", False) - modules.globals.many_faces = switch_states.get("many_faces", False) - modules.globals.map_faces = switch_states.get("map_faces", False) - modules.globals.poisson_blend = switch_states.get("poisson_blend", False) - modules.globals.color_correction = switch_states.get("color_correction", False) - modules.globals.nsfw_filter = switch_states.get("nsfw_filter", False) - modules.globals.live_mirror = switch_states.get("live_mirror", False) - modules.globals.live_resizable = switch_states.get("live_resizable", False) - modules.globals.fp_ui = switch_states.get("fp_ui", {"face_enhancer": False}) - modules.globals.show_fps = switch_states.get("show_fps", False) - modules.globals.mouth_mask_size = switch_states.get("mouth_mask_size", 0.0) - # mouth_mask is driven by the slider: on if size > 0, off if 0 + state = json.load(f) + modules.globals.keep_fps = state.get("keep_fps", True) + modules.globals.keep_audio = state.get("keep_audio", True) + modules.globals.keep_frames = state.get("keep_frames", False) + modules.globals.many_faces = state.get("many_faces", False) + modules.globals.map_faces = state.get("map_faces", False) + modules.globals.poisson_blend = state.get("poisson_blend", False) + modules.globals.color_correction = state.get("color_correction", False) + modules.globals.nsfw_filter = state.get("nsfw_filter", False) + modules.globals.live_mirror = state.get("live_mirror", False) + modules.globals.live_resizable = state.get("live_resizable", False) + modules.globals.fp_ui = state.get("fp_ui", {"face_enhancer": False}) + modules.globals.show_fps = state.get("show_fps", False) + modules.globals.mouth_mask_size = state.get("mouth_mask_size", 0.0) modules.globals.mouth_mask = modules.globals.mouth_mask_size > 0 - modules.globals.show_mouth_mask_box = False # always start hidden + modules.globals.show_mouth_mask_box = False except FileNotFoundError: - # If the file doesn't exist, use default values + pass + except (OSError, json.JSONDecodeError): pass -def create_root(start: Callable[[], None], destroy: Callable[[], None]) -> ctk.CTk: - global source_label, target_label, status_label, show_fps_switch - - load_switch_states() - - ctk.deactivate_automatic_dpi_awareness() - ctk.set_appearance_mode("system") - ctk.set_default_color_theme(resolve_relative_path("ui.json")) - - root = ctk.CTk() - root.minsize(ROOT_WIDTH, ROOT_HEIGHT) - root.title( - f"{modules.metadata.name} {modules.metadata.version} {modules.metadata.edition}" - ) - root.configure() - root.protocol("WM_DELETE_WINDOW", lambda: destroy()) - - source_label = ctk.CTkLabel(root, text=None) - source_label.place(relx=0.1, rely=0.05, relwidth=0.275, relheight=0.225) - - target_label = ctk.CTkLabel(root, text=None) - target_label.place(relx=0.6, rely=0.05, relwidth=0.275, relheight=0.225) - - select_face_button = ctk.CTkButton( - root, text=_("Select a face"), cursor="hand2", command=lambda: select_source_path() - ) - select_face_button.place(relx=0.1, rely=0.30, relwidth=0.24, relheight=0.1) - ToolTip(select_face_button, _("Choose the source face image to swap onto the target")) - - random_face_button = ctk.CTkButton( - root, text="🔄", cursor="hand2", width=30, command=lambda: fetch_random_face() - ) - random_face_button.place(relx=0.35, rely=0.30, relwidth=0.05, relheight=0.1) - ToolTip(random_face_button, _("Get a random face from thispersondoesnotexist.com")) - - swap_faces_button = ctk.CTkButton( - root, text="↔", cursor="hand2", command=lambda: swap_faces_paths() - ) - swap_faces_button.place(relx=0.45, rely=0.30, relwidth=0.1, relheight=0.1) - ToolTip(swap_faces_button, _("Swap source and target images")) - - select_target_button = ctk.CTkButton( - root, - text=_("Select a target"), - cursor="hand2", - command=lambda: select_target_path(), - ) - select_target_button.place(relx=0.6, rely=0.30, relwidth=0.3, relheight=0.1) - ToolTip(select_target_button, _("Choose the target image or video to apply face swap to")) - - keep_fps_value = ctk.BooleanVar(value=modules.globals.keep_fps) - keep_fps_checkbox = ctk.CTkSwitch( - root, - text=_("Keep fps"), - variable=keep_fps_value, - cursor="hand2", - command=lambda: ( - setattr(modules.globals, "keep_fps", keep_fps_value.get()), - save_switch_states(), - ), - ) - keep_fps_checkbox.place(relx=0.1, rely=0.42) - ToolTip(keep_fps_checkbox, _("Output video keeps the original frame rate")) - - keep_frames_value = ctk.BooleanVar(value=modules.globals.keep_frames) - keep_frames_switch = ctk.CTkSwitch( - root, - text=_("Keep frames"), - variable=keep_frames_value, - cursor="hand2", - command=lambda: ( - setattr(modules.globals, "keep_frames", keep_frames_value.get()), - save_switch_states(), - ), - ) - keep_frames_switch.place(relx=0.1, rely=0.47) - ToolTip(keep_frames_switch, _("Keep extracted frames on disk after processing")) - - keep_audio_value = ctk.BooleanVar(value=modules.globals.keep_audio) - keep_audio_switch = ctk.CTkSwitch( - root, - text=_("Keep audio"), - variable=keep_audio_value, - cursor="hand2", - command=lambda: ( - setattr(modules.globals, "keep_audio", keep_audio_value.get()), - save_switch_states(), - ), - ) - keep_audio_switch.place(relx=0.6, rely=0.42) - ToolTip(keep_audio_switch, _("Copy audio track from the source video to output")) - - many_faces_value = ctk.BooleanVar(value=modules.globals.many_faces) - many_faces_switch = ctk.CTkSwitch( - root, - text=_("Many faces"), - variable=many_faces_value, - cursor="hand2", - command=lambda: ( - setattr(modules.globals, "many_faces", many_faces_value.get()), - save_switch_states(), - ), - ) - many_faces_switch.place(relx=0.6, rely=0.47) - ToolTip(many_faces_switch, _("Swap every detected face, not just the primary one")) - - color_correction_value = ctk.BooleanVar(value=modules.globals.color_correction) - color_correction_switch = ctk.CTkSwitch( - root, - text=_("Fix Blueish Cam"), - variable=color_correction_value, - cursor="hand2", - command=lambda: ( - setattr(modules.globals, "color_correction", color_correction_value.get()), - save_switch_states(), - ), - ) - color_correction_switch.place(relx=0.6, rely=0.57) - ToolTip(color_correction_switch, _("Fix blue/green color cast from some webcams")) - - # nsfw_value = ctk.BooleanVar(value=modules.globals.nsfw_filter) - # nsfw_switch = ctk.CTkSwitch(root, text='NSFW filter', variable=nsfw_value, cursor='hand2', command=lambda: setattr(modules.globals, 'nsfw_filter', nsfw_value.get())) - # nsfw_switch.place(relx=0.6, rely=0.7) - - map_faces = ctk.BooleanVar(value=modules.globals.map_faces) - map_faces_switch = ctk.CTkSwitch( - root, - text=_("Map faces"), - variable=map_faces, - cursor="hand2", - command=lambda: ( - setattr(modules.globals, "map_faces", map_faces.get()), - save_switch_states(), - close_mapper_window() if not map_faces.get() else None - ), - ) - map_faces_switch.place(relx=0.1, rely=0.52) - ToolTip(map_faces_switch, _("Manually assign which source face maps to which target face")) - - poisson_blend_value = ctk.BooleanVar(value=modules.globals.poisson_blend) - poisson_blend_switch = ctk.CTkSwitch( - root, - text=_("Poisson Blend"), - variable=poisson_blend_value, - cursor="hand2", - command=lambda: ( - setattr(modules.globals, "poisson_blend", poisson_blend_value.get()), - save_switch_states(), - ), - ) - poisson_blend_switch.place(relx=0.1, rely=0.57) - ToolTip(poisson_blend_switch, _("Blend face edges smoothly using Poisson blending")) - - show_fps_value = ctk.BooleanVar(value=modules.globals.show_fps) - show_fps_switch = ctk.CTkSwitch( - root, - text=_("Show FPS"), - variable=show_fps_value, - cursor="hand2", - command=lambda: ( - setattr(modules.globals, "show_fps", show_fps_value.get()), - save_switch_states(), - ), - ) - show_fps_switch.place(relx=0.6, rely=0.52) - ToolTip(show_fps_switch, _("Display frames-per-second counter on the live preview")) - - # mouth_mask and show_mouth_mask_box are auto-controlled by the Mouth Mask slider - mouth_mask_var = ctk.BooleanVar(value=modules.globals.mouth_mask) - show_mouth_mask_box_var = ctk.BooleanVar(value=modules.globals.show_mouth_mask_box) - - start_button = ctk.CTkButton( - root, text=_("Start"), cursor="hand2", command=lambda: analyze_target(start, root) - ) - start_button.place(relx=0.15, rely=0.78, relwidth=0.2, relheight=0.04) - ToolTip(start_button, _("Begin processing the target image/video with selected face")) - - stop_button = ctk.CTkButton( - root, text=_("Destroy"), cursor="hand2", command=lambda: destroy() - ) - stop_button.place(relx=0.4, rely=0.78, relwidth=0.2, relheight=0.04) - ToolTip(stop_button, _("Stop processing and close the application")) - - preview_button = ctk.CTkButton( - root, text=_("Preview"), cursor="hand2", command=lambda: toggle_preview() - ) - preview_button.place(relx=0.65, rely=0.78, relwidth=0.2, relheight=0.04) - ToolTip(preview_button, _("Show/hide a preview of the processed output")) - - # --- Camera Selection --- - camera_label = ctk.CTkLabel(root, text=_("Select Camera:")) - camera_label.place(relx=0.1, rely=0.83, relwidth=0.2, relheight=0.03) - - available_cameras = get_available_cameras() - camera_indices, camera_names = available_cameras - - if not camera_names or camera_names[0] == "No cameras found": - camera_variable = ctk.StringVar(value="No cameras found") - camera_optionmenu = ctk.CTkOptionMenu( - root, - variable=camera_variable, - values=["No cameras found"], - state="disabled", - ) - else: - camera_variable = ctk.StringVar(value=camera_names[0]) - camera_optionmenu = ctk.CTkOptionMenu( - root, variable=camera_variable, values=camera_names - ) - - camera_optionmenu.place(relx=0.35, rely=0.83, relwidth=0.25, relheight=0.03) - ToolTip(camera_optionmenu, _("Select which camera to use for live mode")) - - live_button = ctk.CTkButton( - root, - text=_("Live"), - cursor="hand2", - command=lambda: webcam_preview( - root, - ( - camera_indices[camera_names.index(camera_variable.get())] - if camera_names and camera_names[0] != "No cameras found" - else None - ), - ), - state=( - "normal" - if camera_names and camera_names[0] != "No cameras found" - else "disabled" - ), - ) - live_button.place(relx=0.65, rely=0.83, relwidth=0.2, relheight=0.03) - ToolTip(live_button, _("Start real-time face swap using webcam")) - # --- End Camera Selection --- - - # --- Face Enhancer Dropdown --- - enhancer_options = ["None", "GFPGAN", "GPEN-512", "GPEN-256"] - enhancer_key_map = { - "None": None, - "GFPGAN": "face_enhancer", - "GPEN-512": "face_enhancer_gpen512", - "GPEN-256": "face_enhancer_gpen256", - } - - # Determine initial value from current fp_ui state - initial_enhancer = "None" - if modules.globals.fp_ui.get("face_enhancer", False): - initial_enhancer = "GFPGAN" - elif modules.globals.fp_ui.get("face_enhancer_gpen512", False): - initial_enhancer = "GPEN-512" - elif modules.globals.fp_ui.get("face_enhancer_gpen256", False): - initial_enhancer = "GPEN-256" - - enhancer_variable = ctk.StringVar(value=initial_enhancer) - - def on_enhancer_change(choice: str): - # Disable all enhancers first - for key in ["face_enhancer", "face_enhancer_gpen256", "face_enhancer_gpen512"]: - update_tumbler(key, False) - # Enable the selected one - selected_key = enhancer_key_map.get(choice) - if selected_key: - update_tumbler(selected_key, True) - save_switch_states() - - enhancer_label = ctk.CTkLabel(root, text="Face Enhancer:") - enhancer_label.place(relx=0.1, rely=0.62, relwidth=0.2, relheight=0.03) - - enhancer_dropdown = ctk.CTkOptionMenu( - root, - variable=enhancer_variable, - values=enhancer_options, - command=on_enhancer_change, - ) - enhancer_dropdown.place(relx=0.35, rely=0.62, relwidth=0.3, relheight=0.03) - ToolTip(enhancer_dropdown, _("Select a face enhancement model (None = no enhancement)")) - - # 1) Define a DoubleVar for transparency (0 = fully transparent, 1 = fully opaque) - transparency_var = ctk.DoubleVar(value=1.0) - - def on_transparency_change(value: float): - # Convert slider value to float - val = float(value) - modules.globals.opacity = val # Set global opacity - percentage = int(val * 100) - - if percentage == 0: - modules.globals.fp_ui["face_enhancer"] = False - update_status("Transparency set to 0% - Face swapping disabled.") - elif percentage == 100: - modules.globals.face_swapper_enabled = True - update_status("Transparency set to 100%.") - else: - modules.globals.face_swapper_enabled = True - update_status(f"Transparency set to {percentage}%") - - # 2) Transparency label and slider - transparency_label = ctk.CTkLabel(root, text="Transparency:") - transparency_label.place(relx=0.15, rely=0.66, relwidth=0.2, relheight=0.03) - - transparency_slider = ctk.CTkSlider( - root, - from_=0.0, - to=1.0, - variable=transparency_var, - command=on_transparency_change, - fg_color="#E0E0E0", - progress_color="#007BFF", - button_color="#FFFFFF", - button_hover_color="#CCCCCC", - height=5, - border_width=1, - corner_radius=3, - ) - transparency_slider.place(relx=0.35, rely=0.67, relwidth=0.5, relheight=0.02) - ToolTip(transparency_slider, _("Blend between original and swapped face (0% = original, 100% = fully swapped)")) - - # 3) Sharpness label & slider - sharpness_var = ctk.DoubleVar(value=0.0) # start at 0.0 - def on_sharpness_change(value: float): - modules.globals.sharpness = float(value) - update_status(f"Sharpness set to {value:.1f}") - - sharpness_label = ctk.CTkLabel(root, text="Sharpness:") - sharpness_label.place(relx=0.15, rely=0.69, relwidth=0.2, relheight=0.03) - - sharpness_slider = ctk.CTkSlider( - root, - from_=0.0, - to=5.0, - variable=sharpness_var, - command=on_sharpness_change, - fg_color="#E0E0E0", - progress_color="#007BFF", - button_color="#FFFFFF", - button_hover_color="#CCCCCC", - height=5, - border_width=1, - corner_radius=3, - ) - sharpness_slider.place(relx=0.35, rely=0.70, relwidth=0.5, relheight=0.02) - ToolTip(sharpness_slider, _("Sharpen the enhanced face output")) - - # 4) Mouth Mask Size slider - mouth_mask_size_var = ctk.DoubleVar(value=modules.globals.mouth_mask_size) - - def on_mouth_mask_size_change(value: float): - val = float(value) - modules.globals.mouth_mask_size = val - # Auto-enable/disable mouth mask based on slider position - if val > 0: - modules.globals.mouth_mask = True - mouth_mask_var.set(True) - else: - modules.globals.mouth_mask = False - mouth_mask_var.set(False) - modules.globals.show_mouth_mask_box = False - - def on_mouth_mask_slider_release(event): - # Hide bounding box when user releases the slider - modules.globals.show_mouth_mask_box = False - - def on_mouth_mask_slider_press(event): - # Show bounding box while dragging - if modules.globals.mouth_mask_size > 0: - modules.globals.show_mouth_mask_box = True - - mouth_mask_size_label = ctk.CTkLabel(root, text="Mouth Mask:") - mouth_mask_size_label.place(relx=0.15, rely=0.72, relwidth=0.2, relheight=0.03) - - mouth_mask_size_slider = ctk.CTkSlider( - root, - from_=0.0, - to=100.0, - variable=mouth_mask_size_var, - command=on_mouth_mask_size_change, - fg_color="#E0E0E0", - progress_color="#007BFF", - button_color="#FFFFFF", - button_hover_color="#CCCCCC", - height=5, - border_width=1, - corner_radius=3, - ) - mouth_mask_size_slider.place(relx=0.35, rely=0.73, relwidth=0.5, relheight=0.02) - mouth_mask_size_slider.bind("", on_mouth_mask_slider_press) - mouth_mask_size_slider.bind("", on_mouth_mask_slider_release) - ToolTip(mouth_mask_size_slider, _("0 = use swapped mouth, 100 = expose original mouth to chin area")) - - # Status and link at the bottom - global status_label - status_label = ctk.CTkLabel(root, text=None, justify="center") - status_label.place(relx=0.1, rely=0.75, relwidth=0.8) - - donate_label = ctk.CTkLabel( - root, text="Deep Live Cam", justify="center", cursor="hand2" - ) - donate_label.place(relx=0.1, rely=0.87, relwidth=0.8) - donate_label.configure( - text_color=ctk.ThemeManager.theme.get("URL").get("text_color") - ) - donate_label.bind( - "