mirror of
https://github.com/deepfakes/faceswap
synced 2025-06-07 19:05:02 -04:00
1c0d112588
- Fix spacing between buttons on Control Panels - Configurable background on Control Panels - Fix background color of TreeView menu in settings pop up - Change TreeView selected item highlight color - Change console background color to match control panel - Fix unfilled color in control panel background
217 lines
8.3 KiB
Python
217 lines
8.3 KiB
Python
#!/usr/bin python3
|
|
""" The optional GUI for faceswap """
|
|
|
|
import logging
|
|
import sys
|
|
import tkinter as tk
|
|
from tkinter import messagebox, ttk
|
|
|
|
from lib.gui import (TaskBar, CliOptions, CommandNotebook, ConsoleOut, DisplayNotebook,
|
|
get_images, initialize_images, initialize_config, LastSession,
|
|
MainMenuBar, preview_trigger, ProcessWrapper, StatusBar)
|
|
|
|
logger = logging.getLogger(__name__) # pylint: disable=invalid-name
|
|
|
|
|
|
class FaceswapGui(tk.Tk):
|
|
""" The Graphical User Interface """
|
|
|
|
def __init__(self, debug):
|
|
logger.debug("Initializing %s", self.__class__.__name__)
|
|
super().__init__()
|
|
|
|
self._init_args = dict(debug=debug)
|
|
self._config = self.initialize_globals()
|
|
self.set_fonts()
|
|
self.set_styles()
|
|
self._config.set_geometry(1200, 640, self._config.user_config_dict["fullscreen"])
|
|
|
|
self.wrapper = ProcessWrapper()
|
|
self.objects = dict()
|
|
|
|
get_images().delete_preview()
|
|
preview_trigger().clear()
|
|
self.protocol("WM_DELETE_WINDOW", self.close_app)
|
|
self.build_gui()
|
|
self._last_session = LastSession(self._config)
|
|
logger.debug("Initialized %s", self.__class__.__name__)
|
|
|
|
def initialize_globals(self):
|
|
""" Initialize config and images global constants """
|
|
cliopts = CliOptions()
|
|
statusbar = StatusBar(self)
|
|
config = initialize_config(self, cliopts, statusbar)
|
|
initialize_images()
|
|
return config
|
|
|
|
def set_fonts(self):
|
|
""" Set global default font """
|
|
tk.font.nametofont("TkFixedFont").configure(size=self._config.default_font[1])
|
|
for font in ("TkDefaultFont", "TkHeadingFont", "TkMenuFont"):
|
|
tk.font.nametofont(font).configure(family=self._config.default_font[0],
|
|
size=self._config.default_font[1])
|
|
|
|
def set_styles(self):
|
|
""" Set global custom styles """
|
|
gui_style = ttk.Style()
|
|
font = self._config.default_font
|
|
gui_style.configure('H1.TLabel', font=(font[0], font[1] + 4, "bold"))
|
|
gui_style.configure('H2.TLabel', font=(font[0], font[1] + 2, "bold"))
|
|
|
|
# Control and settings panel styles
|
|
for _type in ("CPanel", "SPanel"):
|
|
# Common control panel items
|
|
for lbl in ["TLabel", "TFrame", "TLabelframe", "TCheckbutton", "TRadiobutton"]:
|
|
gui_style.configure(f"{_type}.{lbl}", background="#FFFFFF")
|
|
|
|
# Background colors
|
|
color = "#CDD3D5" if _type == "CPanel" else "#DAD2D8"
|
|
gui_style.configure(f"Holder.{_type}.TFrame", background=color)
|
|
|
|
# Highlight Colors
|
|
color = "#176087" if _type == "CPanel" else "#9B1D20"
|
|
gui_style.configure(f"{_type}.TLabelframe.Label",
|
|
background="#FFFFFF",
|
|
foreground=color)
|
|
gui_style.configure(f"{_type}.Groupheader.TLabel",
|
|
background=color,
|
|
foreground="#FFFFFF",
|
|
font=(font[0], font[1], "bold"))
|
|
|
|
# Control Panel Info Box
|
|
gui_style.configure(f"InfoHeader.{_type}.TLabel",
|
|
background='#FFFFFF',
|
|
font=(font[0], font[1], "bold"))
|
|
gui_style.configure(f"InfoBody.{_type}.TLabel", background="#FFFFFF")
|
|
|
|
def build_gui(self, rebuild=False):
|
|
""" Build the GUI """
|
|
logger.debug("Building GUI")
|
|
if not rebuild:
|
|
self.tk.call('wm', 'iconphoto', self._w, get_images().icons["favicon"])
|
|
self.configure(menu=MainMenuBar(self))
|
|
|
|
if rebuild:
|
|
objects = list(self.objects.keys())
|
|
for obj in objects:
|
|
self.objects[obj].destroy()
|
|
del self.objects[obj]
|
|
|
|
self.objects["taskbar"] = TaskBar(self)
|
|
self.add_containers()
|
|
|
|
self.objects["command"] = CommandNotebook(self.objects["container_top"])
|
|
self.objects["display"] = DisplayNotebook(self.objects["container_top"])
|
|
self.objects["console"] = ConsoleOut(self.objects["container_bottom"],
|
|
self._init_args["debug"])
|
|
self.set_initial_focus()
|
|
self.set_layout()
|
|
self._config.set_default_options()
|
|
logger.debug("Built GUI")
|
|
|
|
def add_containers(self):
|
|
""" Add the paned window containers that
|
|
hold each main area of the gui """
|
|
logger.debug("Adding containers")
|
|
maincontainer = ttk.PanedWindow(self,
|
|
orient=tk.VERTICAL,
|
|
name="pw_main")
|
|
maincontainer.pack(fill=tk.BOTH, expand=True)
|
|
|
|
topcontainer = ttk.PanedWindow(maincontainer,
|
|
orient=tk.HORIZONTAL,
|
|
name="pw_top")
|
|
maincontainer.add(topcontainer)
|
|
|
|
bottomcontainer = ttk.Frame(maincontainer, name="frame_bottom")
|
|
maincontainer.add(bottomcontainer)
|
|
self.objects["container_main"] = maincontainer
|
|
self.objects["container_top"] = topcontainer
|
|
self.objects["container_bottom"] = bottomcontainer
|
|
|
|
logger.debug("Added containers")
|
|
|
|
def set_initial_focus(self):
|
|
""" Set the tab focus from settings """
|
|
tab = self._config.user_config_dict["tab"]
|
|
logger.debug("Setting focus for tab: %s", tab)
|
|
self._config.set_active_tab_by_name(tab)
|
|
logger.debug("Focus set to: %s", tab)
|
|
|
|
def set_layout(self):
|
|
""" Set initial layout """
|
|
self.update_idletasks()
|
|
config_opts = self._config.user_config_dict
|
|
r_width = self.winfo_width()
|
|
r_height = self.winfo_height()
|
|
w_ratio = config_opts["options_panel_width"] / 100.0
|
|
h_ratio = 1 - (config_opts["console_panel_height"] / 100.0)
|
|
width = round(r_width * w_ratio)
|
|
height = round(r_height * h_ratio)
|
|
logger.debug("Setting Initial Layout: (root_width: %s, root_height: %s, width_ratio: %s, "
|
|
"height_ratio: %s, width: %s, height: %s", r_width, r_height, w_ratio,
|
|
h_ratio, width, height)
|
|
self.objects["container_top"].sashpos(0, width)
|
|
self.objects["container_main"].sashpos(0, height)
|
|
self.update_idletasks()
|
|
|
|
def rebuild(self):
|
|
""" Rebuild the GUI on config change """
|
|
logger.debug("Redrawing GUI")
|
|
session_state = self._last_session.to_dict()
|
|
self._config.refresh_config()
|
|
get_images().__init__()
|
|
self.set_fonts()
|
|
self.build_gui(rebuild=True)
|
|
if session_state is not None:
|
|
self._last_session.from_dict(session_state)
|
|
logger.debug("GUI Redrawn")
|
|
|
|
def close_app(self, *args): # pylint: disable=unused-argument
|
|
""" Close Python. This is here because the graph
|
|
animation function continues to run even when
|
|
tkinter has gone away """
|
|
logger.debug("Close Requested")
|
|
|
|
if not self._confirm_close_on_running_task():
|
|
return
|
|
if not self._config.project.confirm_close():
|
|
return
|
|
|
|
if self._config.tk_vars["runningtask"].get():
|
|
self.wrapper.task.terminate()
|
|
|
|
self._last_session.save()
|
|
get_images().delete_preview()
|
|
preview_trigger().clear()
|
|
self.quit()
|
|
logger.debug("Closed GUI")
|
|
sys.exit(0)
|
|
|
|
def _confirm_close_on_running_task(self):
|
|
""" Pop a confirmation box to close the GUI if a task is running
|
|
|
|
Returns
|
|
-------
|
|
bool: ``True`` if user confirms close, ``False`` if user cancels close
|
|
"""
|
|
if not self._config.tk_vars["runningtask"].get():
|
|
logger.debug("No tasks currently running")
|
|
return True
|
|
|
|
confirmtxt = "Processes are still running.\n\nAre you sure you want to exit?"
|
|
if not messagebox.askokcancel("Close", confirmtxt, default="cancel", icon="warning"):
|
|
logger.debug("Close Cancelled")
|
|
return True
|
|
logger.debug("Close confirmed")
|
|
return False
|
|
|
|
|
|
class Gui(): # pylint: disable=too-few-public-methods
|
|
""" The GUI process. """
|
|
def __init__(self, arguments):
|
|
self.root = FaceswapGui(arguments.debug)
|
|
|
|
def process(self):
|
|
""" Builds the GUI """
|
|
self.root.mainloop()
|