diaojiaolou
/
Deep-Live-Cam
mirror of https://github.com/hacksider/Deep-Live-Cam.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

drag-n-drop source / target in main ui

pull/933/head
Vladislav Naumov 2025-02-11 19:39:27 +06:00
parent 9746f78d45
commit df14b65c1b
1 changed files with 26 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
modules/ui.py
View File

@ -156,12 +156,12 @@ def create_root(start: Callable[[], None], destroy: Callable[[], None]) -> ctk.C
source_label = ctk.CTkLabel(root, text=None) source_label = ctk.CTkLabel(root, text=None)
source_label.place(relx=0.1, rely=0.1, relwidth=0.3, relheight=0.25) source_label.place(relx=0.1, rely=0.1, relwidth=0.3, relheight=0.25)
source_label.drop_target_register(DND_FILES) source_label.drop_target_register(DND_FILES)
source_label.dnd_bind('<<Drop>>', lambda e: print("SOURCE: ", e.data)) source_label.dnd_bind('<<Drop>>', lambda e: select_source_path(e.data))
target_label = ctk.CTkLabel(root, text=None) target_label = ctk.CTkLabel(root, text=None)
target_label.place(relx=0.6, rely=0.1, relwidth=0.3, relheight=0.25) target_label.place(relx=0.6, rely=0.1, relwidth=0.3, relheight=0.25)
target_label.drop_target_register(DND_FILES) target_label.drop_target_register(DND_FILES)
target_label.dnd_bind('<<Drop>>', lambda e: print("TARGET: ", e.data)) target_label.dnd_bind('<<Drop>>', lambda e: select_target_path(e.data))
select_face_button = ctk.CTkButton( select_face_button = ctk.CTkButton(
root, text=_("Select a face"), cursor="hand2", command=lambda: select_source_path() root, text=_("Select a face"), cursor="hand2", command=lambda: select_source_path()
@ -585,15 +585,21 @@ def update_tumbler(var: str, value: bool) -> None:
) )
def select_source_path() -> None: def select_source_path(filepath: str = None) -> None:
global RECENT_DIRECTORY_SOURCE, img_ft, vid_ft global RECENT_DIRECTORY_SOURCE, img_ft, vid_ft
PREVIEW.withdraw() PREVIEW.withdraw()
source_path = ctk.filedialog.askopenfilename(
title=_("select an source image"), # drag-n-drop target, no need to open file dialog
initialdir=RECENT_DIRECTORY_SOURCE, if filepath is None:
filetypes=[img_ft], source_path = ctk.filedialog.askopenfilename(
) title=_("select an source image"),
initialdir=RECENT_DIRECTORY_SOURCE,
filetypes=[img_ft],
)
else:
source_path = filepath
if is_image(source_path): if is_image(source_path):
modules.globals.source_path = source_path modules.globals.source_path = source_path
RECENT_DIRECTORY_SOURCE = os.path.dirname(modules.globals.source_path) RECENT_DIRECTORY_SOURCE = os.path.dirname(modules.globals.source_path)
@ -628,15 +634,21 @@ def swap_faces_paths() -> None:
target_label.configure(image=target_image) target_label.configure(image=target_image)
def select_target_path() -> None: def select_target_path(filepath: str = None) -> None:
global RECENT_DIRECTORY_TARGET, img_ft, vid_ft global RECENT_DIRECTORY_TARGET, img_ft, vid_ft
PREVIEW.withdraw() PREVIEW.withdraw()
target_path = ctk.filedialog.askopenfilename(
title=_("select an target image or video"), # drag-n-drop target, no need to open file dialog
initialdir=RECENT_DIRECTORY_TARGET, if filepath is None:
filetypes=[img_ft, vid_ft], target_path = ctk.filedialog.askopenfilename(
) title=_("select an target image or video"),
initialdir=RECENT_DIRECTORY_TARGET,
filetypes=[img_ft, vid_ft],
)
else:
target_path = filepath
if is_image(target_path): if is_image(target_path):
modules.globals.target_path = target_path modules.globals.target_path = target_path
RECENT_DIRECTORY_TARGET = os.path.dirname(modules.globals.target_path) RECENT_DIRECTORY_TARGET = os.path.dirname(modules.globals.target_path)