mirror of
https://github.com/deepfakes/faceswap
synced 2025-06-08 03:26:47 -04:00
7f53911453
* Convert prints to logger. Further logging improvements. Tidy up * Fix system verbosity. Allow SystemExit * Fix reload extract bug * Child Traceback handling * Safer shutdown procedure * Add shutdown event to queue manager * landmarks_as_xy > property. GUI notes + linting. Aligner bugfix * fix FaceFilter. Enable nFilter when no Filter is supplied * Fix blurry face filter * Continue on IO error. Better error handling * Explicitly print stack trace tocrash log * Windows Multiprocessing bugfix * Add git info and conda version to crash log * Windows/Anaconda mp bugfix * Logging fixes for training
141 lines
4.9 KiB
Python
141 lines
4.9 KiB
Python
#!/usr/bin python3
|
|
""" The optional GUI for faceswap """
|
|
|
|
# NB: The GUI can't currently log as it is a wrapper for the python scripts, so don't
|
|
# implement logging unless you can handle the conflicts
|
|
|
|
import os
|
|
import sys
|
|
import tkinter as tk
|
|
|
|
from tkinter import messagebox, ttk
|
|
|
|
from lib.gui import (CliOptions, CurrentSession, CommandNotebook, Config,
|
|
ConsoleOut, DisplayNotebook, Images, ProcessWrapper,
|
|
StatusBar)
|
|
|
|
|
|
class FaceswapGui(tk.Tk):
|
|
""" The Graphical User Interface """
|
|
|
|
def __init__(self, pathscript):
|
|
tk.Tk.__init__(self)
|
|
self.scaling_factor = self.get_scaling()
|
|
self.set_geometry()
|
|
|
|
pathcache = os.path.join(pathscript, "lib", "gui", ".cache")
|
|
self.images = Images(pathcache)
|
|
self.cliopts = CliOptions()
|
|
self.session = CurrentSession()
|
|
statusbar = StatusBar(self)
|
|
self.wrapper = ProcessWrapper(statusbar,
|
|
self.session,
|
|
pathscript,
|
|
self.cliopts)
|
|
|
|
self.images.delete_preview()
|
|
self.protocol("WM_DELETE_WINDOW", self.close_app)
|
|
|
|
def get_scaling(self):
|
|
""" Get the display DPI """
|
|
dpi = self.winfo_fpixels("1i")
|
|
return dpi / 72.0
|
|
|
|
def set_geometry(self):
|
|
""" Set GUI geometry """
|
|
self.tk.call("tk", "scaling", self.scaling_factor)
|
|
width = int(1200 * self.scaling_factor)
|
|
height = int(640 * self.scaling_factor)
|
|
self.geometry("{}x{}+80+80".format(str(width), str(height)))
|
|
|
|
def build_gui(self, debug_console):
|
|
""" Build the GUI """
|
|
self.title("Faceswap.py")
|
|
self.menu()
|
|
|
|
topcontainer, bottomcontainer = self.add_containers()
|
|
|
|
CommandNotebook(topcontainer,
|
|
self.cliopts,
|
|
self.wrapper.tk_vars,
|
|
self.scaling_factor)
|
|
DisplayNotebook(topcontainer,
|
|
self.session,
|
|
self.wrapper.tk_vars,
|
|
self.scaling_factor)
|
|
ConsoleOut(bottomcontainer, debug_console, self.wrapper.tk_vars)
|
|
|
|
def menu(self):
|
|
""" Menu bar for loading and saving configs """
|
|
menubar = tk.Menu(self)
|
|
filemenu = tk.Menu(menubar, tearoff=0)
|
|
|
|
config = Config(self.cliopts, self.wrapper.tk_vars)
|
|
|
|
filemenu.add_command(label="Load full config...",
|
|
underline=0,
|
|
command=config.load)
|
|
filemenu.add_command(label="Save full config...",
|
|
underline=0,
|
|
command=config.save)
|
|
filemenu.add_separator()
|
|
filemenu.add_command(label="Reset all to default",
|
|
underline=0,
|
|
command=self.cliopts.reset)
|
|
filemenu.add_command(label="Clear all",
|
|
underline=0,
|
|
command=self.cliopts.clear)
|
|
filemenu.add_separator()
|
|
filemenu.add_command(label="Quit",
|
|
underline=0,
|
|
command=self.close_app)
|
|
|
|
menubar.add_cascade(label="File", menu=filemenu, underline=0)
|
|
self.config(menu=menubar)
|
|
|
|
def add_containers(self):
|
|
""" Add the paned window containers that
|
|
hold each main area of the gui """
|
|
maincontainer = tk.PanedWindow(self,
|
|
sashrelief=tk.RAISED,
|
|
orient=tk.VERTICAL)
|
|
maincontainer.pack(fill=tk.BOTH, expand=True)
|
|
|
|
topcontainer = tk.PanedWindow(maincontainer,
|
|
sashrelief=tk.RAISED,
|
|
orient=tk.HORIZONTAL)
|
|
maincontainer.add(topcontainer)
|
|
|
|
bottomcontainer = ttk.Frame(maincontainer, height=150)
|
|
maincontainer.add(bottomcontainer)
|
|
|
|
return topcontainer, bottomcontainer
|
|
|
|
def close_app(self):
|
|
""" Close Python. This is here because the graph
|
|
animation function continues to run even when
|
|
tkinter has gone away """
|
|
confirm = messagebox.askokcancel
|
|
confirmtxt = "Processes are still running. Are you sure...?"
|
|
if (self.wrapper.tk_vars["runningtask"].get()
|
|
and not confirm("Close", confirmtxt)):
|
|
return
|
|
if self.wrapper.tk_vars["runningtask"].get():
|
|
self.wrapper.task.terminate()
|
|
self.images.delete_preview()
|
|
self.quit()
|
|
exit()
|
|
|
|
|
|
class Gui(): # pylint: disable=too-few-public-methods
|
|
""" The GUI process. """
|
|
def __init__(self, arguments):
|
|
cmd = sys.argv[0]
|
|
pathscript = os.path.realpath(os.path.dirname(cmd))
|
|
self.args = arguments
|
|
self.root = FaceswapGui(pathscript)
|
|
|
|
def process(self):
|
|
""" Builds the GUI """
|
|
self.root.build_gui(self.args.debug)
|
|
self.root.mainloop()
|