From 674f584895ad5955f9b3ccfd88eb732b63c14d0f Mon Sep 17 00:00:00 2001 From: K <136873090+KRSHH@users.noreply.github.com> Date: Mon, 16 Sep 2024 13:31:20 +0530 Subject: [PATCH] UI Change Changes in UI --- modules/ui.py | 401 ++++++++++++++++++++++++++++---------------------- 1 file changed, 221 insertions(+), 180 deletions(-) diff --git a/modules/ui.py b/modules/ui.py index e2470bc..1779859 100644 --- a/modules/ui.py +++ b/modules/ui.py @@ -28,27 +28,27 @@ from modules.utilities import ( ROOT = None POPUP = None POPUP_LIVE = None -ROOT_HEIGHT = 700 -ROOT_WIDTH = 600 +ROOT_HEIGHT = 800 +ROOT_WIDTH = 1000 PREVIEW = None -PREVIEW_MAX_HEIGHT = 700 -PREVIEW_MAX_WIDTH = 1200 -PREVIEW_DEFAULT_WIDTH = 960 -PREVIEW_DEFAULT_HEIGHT = 540 +PREVIEW_MAX_HEIGHT = 800 +PREVIEW_MAX_WIDTH = 1400 +PREVIEW_DEFAULT_WIDTH = 1280 +PREVIEW_DEFAULT_HEIGHT = 720 -POPUP_WIDTH = 750 -POPUP_HEIGHT = 810 -POPUP_SCROLL_WIDTH = 740 -POPUP_SCROLL_HEIGHT = 700 +POPUP_WIDTH = 700 +POPUP_HEIGHT = 800 +POPUP_SCROLL_WIDTH = 680 +POPUP_SCROLL_HEIGHT = 600 -POPUP_LIVE_WIDTH = 900 -POPUP_LIVE_HEIGHT = 820 -POPUP_LIVE_SCROLL_WIDTH = 890 -POPUP_LIVE_SCROLL_HEIGHT = 700 +POPUP_LIVE_WIDTH = 800 +POPUP_LIVE_HEIGHT = 700 +POPUP_LIVE_SCROLL_WIDTH = 780 +POPUP_LIVE_SCROLL_HEIGHT = 600 -MAPPER_PREVIEW_MAX_HEIGHT = 100 -MAPPER_PREVIEW_MAX_WIDTH = 100 +MAPPER_PREVIEW_MAX_HEIGHT = 120 +MAPPER_PREVIEW_MAX_WIDTH = 120 DEFAULT_BUTTON_WIDTH = 200 DEFAULT_BUTTON_HEIGHT = 40 @@ -71,12 +71,25 @@ target_label_dict_live = {} img_ft, vid_ft = modules.globals.file_types -class DragDropButton(ctk.CTkButton): +class ModernButton(ctk.CTkButton): + def __init__(self, master, **kwargs): + super().__init__(master, **kwargs) + self.configure( + font=("Roboto", 16, "bold"), + corner_radius=15, + border_width=2, + border_color="#3a7ebf", + hover_color="#2b5d8b", + fg_color="#3a7ebf", + text_color="white", + ) + + +class DragDropButton(ModernButton): def __init__(self, master, **kwargs): super().__init__(master, **kwargs) self.drop_target_register(tkdnd.DND_FILES) self.dnd_bind("<>", self.drop) - self.configure(font=("Arial", 14, "bold")) # Increased font size and made bold def drop(self, event): file_path = event.data @@ -94,7 +107,7 @@ class SourceButton(DragDropButton): modules.globals.source_path = file_path global RECENT_DIRECTORY_SOURCE RECENT_DIRECTORY_SOURCE = os.path.dirname(modules.globals.source_path) - image = render_image_preview(modules.globals.source_path, (200, 200)) + image = render_image_preview(modules.globals.source_path, (250, 250)) source_label.configure(image=image) source_label.configure(text="") @@ -107,7 +120,9 @@ class SourceMapperButton(DragDropButton): def handle_drop(self, file_path): if is_image(file_path): - update_popup_source(self.master, self.map, self.button_num, file_path) + update_popup_source( + self.master.master, self.map, self.button_num, file_path + ) class TargetButton(DragDropButton): @@ -117,21 +132,31 @@ class TargetButton(DragDropButton): modules.globals.target_path = file_path RECENT_DIRECTORY_TARGET = os.path.dirname(modules.globals.target_path) if is_image(file_path): - image = render_image_preview(modules.globals.target_path, (200, 200)) + image = render_image_preview(modules.globals.target_path, (250, 250)) target_label.configure(image=image) target_label.configure(text="") elif is_video(file_path): - video_frame = render_video_preview(file_path, (200, 200)) + video_frame = render_video_preview(file_path, (250, 250)) target_label.configure(image=video_frame) target_label.configure(text="") -class DragDropLabel(ctk.CTkLabel): +class ModernLabel(ctk.CTkLabel): + def __init__(self, master, **kwargs): + super().__init__(master, **kwargs) + self.configure( + font=("Roboto", 16), + corner_radius=10, + fg_color="#2a2d2e", + text_color="white", + ) + + +class DragDropLabel(ModernLabel): def __init__(self, master, **kwargs): super().__init__(master, **kwargs) self.drop_target_register(tkdnd.DND_FILES) self.dnd_bind("<>", self.drop) - self.configure(font=("Arial", 14, "bold")) # Increased font size and made bold def drop(self, event): file_path = event.data @@ -149,7 +174,7 @@ class SourceLabel(DragDropLabel): modules.globals.source_path = file_path global RECENT_DIRECTORY_SOURCE RECENT_DIRECTORY_SOURCE = os.path.dirname(modules.globals.source_path) - image = render_image_preview(modules.globals.source_path, (200, 200)) + image = render_image_preview(modules.globals.source_path, (250, 250)) source_label.configure(image=image) source_label.configure(text="") @@ -161,11 +186,11 @@ class TargetLabel(DragDropLabel): modules.globals.target_path = file_path RECENT_DIRECTORY_TARGET = os.path.dirname(modules.globals.target_path) if is_image(file_path): - image = render_image_preview(modules.globals.target_path, (200, 200)) + image = render_image_preview(modules.globals.target_path, (250, 250)) target_label.configure(image=image) target_label.configure(text="") elif is_video(file_path): - video_frame = render_video_preview(file_path, (200, 200)) + video_frame = render_video_preview(file_path, (250, 250)) target_label.configure(image=video_frame) target_label.configure(text="") @@ -185,234 +210,240 @@ def create_root( global source_label, target_label, status_label ctk.set_appearance_mode("dark") - ctk.set_default_color_theme("blue") # Use a built-in theme + ctk.set_default_color_theme("blue") root = tkdnd.TkinterDnD.Tk() root.minsize(ROOT_WIDTH, ROOT_HEIGHT) root.title( f"{modules.metadata.name} {modules.metadata.version} {modules.metadata.edition}" ) - root.configure(bg="gray12") + root.configure(bg="#1a1a1a") root.protocol("WM_DELETE_WINDOW", lambda: destroy()) + main_frame = ctk.CTkFrame(root, fg_color="#1a1a1a") + main_frame.pack(fill="both", expand=True, padx=20, pady=20) + + # Create two vertical frames for source and target + source_frame = ctk.CTkFrame(main_frame, fg_color="#2a2d2e", corner_radius=15) + source_frame.grid(row=0, column=0, padx=10, pady=10, sticky="nsew") + + target_frame = ctk.CTkFrame(main_frame, fg_color="#2a2d2e", corner_radius=15) + target_frame.grid(row=0, column=2, padx=10, pady=10, sticky="nsew") + + # Create a middle frame for swap button + middle_frame = ctk.CTkFrame(main_frame, fg_color="#1a1a1a") + middle_frame.grid(row=0, column=1, padx=5, pady=10, sticky="ns") + source_label = SourceLabel( - root, + source_frame, text="Drag & Drop\nSource Image Here", - text_color="gray", - font=("Arial", 18, "bold"), justify="center", + width=250, + height=250, ) - source_label.place(relx=0.1, rely=0.1, relwidth=0.3, relheight=0.25) + source_label.pack(pady=(20, 10)) target_label = TargetLabel( - root, + target_frame, text="Drag & Drop\nTarget Image/Video Here", - text_color="gray", - font=("Arial", 18, "bold"), justify="center", + width=250, + height=250, ) - target_label.place(relx=0.6, rely=0.1, relwidth=0.3, relheight=0.25) + target_label.pack(pady=(20, 10)) select_face_button = SourceButton( - root, + source_frame, text="Select a face", cursor="hand2", command=lambda: select_source_path(), - fg_color=("gray75", "gray25"), # Modern button color - hover_color=("gray85", "gray35"), - corner_radius=10, # Rounded corners ) - select_face_button.place(relx=0.1, rely=0.4, relwidth=0.3, relheight=0.1) - - swap_faces_button = ctk.CTkButton( - root, - text="↔", - cursor="hand2", - command=lambda: swap_faces_paths(), - fg_color=("gray75", "gray25"), - hover_color=("gray85", "gray35"), - corner_radius=10, - font=("Arial", 14, "bold"), - ) - swap_faces_button.place(relx=0.45, rely=0.4, relwidth=0.1, relheight=0.1) + select_face_button.pack(pady=10) select_target_button = TargetButton( - root, + target_frame, text="Select a target", cursor="hand2", command=lambda: select_target_path(), - fg_color=("gray75", "gray25"), - hover_color=("gray85", "gray35"), - corner_radius=10, ) - select_target_button.place(relx=0.6, rely=0.4, relwidth=0.3, relheight=0.1) + select_target_button.pack(pady=10) + swap_faces_button = ModernButton( + middle_frame, + text="↔", + cursor="hand2", + command=lambda: swap_faces_paths(), + width=50, + height=50, + ) + swap_faces_button.pack(expand=True) + + options_frame = ctk.CTkFrame(main_frame, fg_color="#2a2d2e", corner_radius=15) + options_frame.grid(row=1, column=0, columnspan=3, padx=10, pady=10, sticky="nsew") + + # Create two columns for options, centered + options_column_left = ctk.CTkFrame(options_frame, fg_color="#2a2d2e") + options_column_left.pack(side="left", expand=True) + + options_column_right = ctk.CTkFrame(options_frame, fg_color="#2a2d2e") + options_column_right.pack(side="right", expand=True) + + # Left column switches keep_fps_value = ctk.BooleanVar(value=modules.globals.keep_fps) keep_fps_checkbox = ctk.CTkSwitch( - root, + options_column_left, text="Keep fps", variable=keep_fps_value, cursor="hand2", command=lambda: setattr( modules.globals, "keep_fps", not modules.globals.keep_fps ), - fg_color=("gray75", "gray25"), # Modern switch color - progress_color=("DodgerBlue", "DodgerBlue"), - font=("Arial", 14, "bold"), + progress_color="#3a7ebf", + font=("Roboto", 14, "bold"), ) - keep_fps_checkbox.place(relx=0.1, rely=0.6) + keep_fps_checkbox.pack(pady=5) keep_frames_value = ctk.BooleanVar(value=modules.globals.keep_frames) keep_frames_switch = ctk.CTkSwitch( - root, + options_column_left, text="Keep frames", variable=keep_frames_value, cursor="hand2", command=lambda: setattr( modules.globals, "keep_frames", keep_frames_value.get() ), - fg_color=("gray75", "gray25"), - progress_color=("DodgerBlue", "DodgerBlue"), - font=("Arial", 14, "bold"), + progress_color="#3a7ebf", + font=("Roboto", 14, "bold"), ) - keep_frames_switch.place(relx=0.1, rely=0.65) + keep_frames_switch.pack(pady=5) enhancer_value = ctk.BooleanVar(value=modules.globals.fp_ui["face_enhancer"]) enhancer_switch = ctk.CTkSwitch( - root, + options_column_left, text="Face Enhancer", variable=enhancer_value, cursor="hand2", command=lambda: update_tumbler("face_enhancer", enhancer_value.get()), - fg_color=("gray75", "gray25"), - progress_color=("DodgerBlue", "DodgerBlue"), - font=("Arial", 14, "bold"), + progress_color="#3a7ebf", + font=("Roboto", 14, "bold"), ) - enhancer_switch.place(relx=0.1, rely=0.7) + enhancer_switch.pack(pady=5) keep_audio_value = ctk.BooleanVar(value=modules.globals.keep_audio) keep_audio_switch = ctk.CTkSwitch( - root, + options_column_left, text="Keep audio", variable=keep_audio_value, cursor="hand2", command=lambda: setattr(modules.globals, "keep_audio", keep_audio_value.get()), - fg_color=("gray75", "gray25"), - progress_color=("DodgerBlue", "DodgerBlue"), - font=("Arial", 14, "bold"), + progress_color="#3a7ebf", + font=("Roboto", 14, "bold"), ) - keep_audio_switch.place(relx=0.6, rely=0.6) + keep_audio_switch.pack(pady=5) + # Right column switches many_faces_value = ctk.BooleanVar(value=modules.globals.many_faces) many_faces_switch = ctk.CTkSwitch( - root, + options_column_right, text="Many faces", variable=many_faces_value, cursor="hand2", command=lambda: setattr(modules.globals, "many_faces", many_faces_value.get()), - fg_color=("gray75", "gray25"), - progress_color=("DodgerBlue", "DodgerBlue"), - font=("Arial", 14, "bold"), + progress_color="#3a7ebf", + font=("Roboto", 14, "bold"), ) - many_faces_switch.place(relx=0.6, rely=0.65) + many_faces_switch.pack(pady=5) color_correction_value = ctk.BooleanVar(value=modules.globals.color_correction) color_correction_switch = ctk.CTkSwitch( - root, - text="Fix Blueish Cam -\nforce cv2 to\nuse RGB instead of BGR", + options_column_right, + text="Fix Blueish Cam", variable=color_correction_value, cursor="hand2", command=lambda: setattr( modules.globals, "color_correction", color_correction_value.get() ), - fg_color=("gray75", "gray25"), - progress_color=("DodgerBlue", "DodgerBlue"), - font=("Arial", 14, "bold"), + progress_color="#3a7ebf", + font=("Roboto", 14, "bold"), ) - color_correction_switch.place(relx=0.6, rely=0.70) + color_correction_switch.pack(pady=5) map_faces = ctk.BooleanVar(value=modules.globals.map_faces) map_faces_switch = ctk.CTkSwitch( - root, + options_column_right, text="Map faces", variable=map_faces, cursor="hand2", command=lambda: setattr(modules.globals, "map_faces", map_faces.get()), - fg_color=("gray75", "gray25"), - progress_color=("DodgerBlue", "DodgerBlue"), - font=("Arial", 14, "bold"), + progress_color="#3a7ebf", + font=("Roboto", 14, "bold"), ) - map_faces_switch.place(relx=0.1, rely=0.75) + map_faces_switch.pack(pady=5) - start_button = ctk.CTkButton( - root, + button_frame = ctk.CTkFrame(main_frame, fg_color="#1a1a1a") + button_frame.grid(row=2, column=0, columnspan=3, padx=10, pady=10, sticky="nsew") + + start_button = ModernButton( + button_frame, text="Start", cursor="hand2", command=lambda: analyze_target(start, root), - fg_color=("DodgerBlue", "DodgerBlue"), # Modern button color - hover_color=("RoyalBlue", "RoyalBlue"), - corner_radius=10, - font=("Arial", 14, "bold"), + fg_color="#4CAF50", + hover_color="#45a049", ) - start_button.place(relx=0.15, rely=0.80, relwidth=0.2, relheight=0.05) + start_button.pack(side="left", padx=10, expand=True) - stop_button = ctk.CTkButton( - root, - text="Destroy", - cursor="hand2", - command=lambda: destroy(), - fg_color=("gray75", "gray25"), - hover_color=("gray85", "gray35"), - corner_radius=10, - font=("Arial", 14, "bold"), - ) - stop_button.place(relx=0.4, rely=0.80, relwidth=0.2, relheight=0.05) - - preview_button = ctk.CTkButton( - root, + preview_button = ModernButton( + button_frame, text="Preview", cursor="hand2", command=lambda: toggle_preview(), - fg_color=("gray75", "gray25"), - hover_color=("gray85", "gray35"), - corner_radius=10, - font=("Arial", 14, "bold"), ) - preview_button.place(relx=0.65, rely=0.80, relwidth=0.2, relheight=0.05) + preview_button.pack(side="left", padx=10, expand=True) - live_button = ctk.CTkButton( - root, + live_button = ModernButton( + button_frame, text="Live", cursor="hand2", command=lambda: webcam_preview(root), - fg_color=("gray75", "gray25"), - hover_color=("gray85", "gray35"), - corner_radius=10, - font=("Arial", 14, "bold"), ) - live_button.place(relx=0.40, rely=0.86, relwidth=0.2, relheight=0.05) + live_button.pack(side="left", padx=10, expand=True) - status_label = ctk.CTkLabel( - root, text=None, justify="center", font=("Arial", 14, "bold") + stop_button = ModernButton( + button_frame, + text="Destroy", + cursor="hand2", + command=lambda: destroy(), + fg_color="#f44336", + hover_color="#d32f2f", ) - status_label.place(relx=0.1, rely=0.9, relwidth=0.8) + stop_button.pack(side="left", padx=10, expand=True) - donate_label = ctk.CTkLabel( - root, - text="Deep Live Cam", + status_label = ModernLabel( + main_frame, text=None, justify="center", fg_color="#1a1a1a" + ) + status_label.grid(row=3, column=0, columnspan=3, pady=10, sticky="ew") + + donate_label = ModernLabel( + main_frame, + text="Donate", justify="center", cursor="hand2", - font=("Arial", 14, "bold"), + fg_color="#1870c4", + text_color="#1870c4", ) - donate_label.place(relx=0.1, rely=0.95, relwidth=0.8) - - # Access the URL text color directly - donate_label.configure(text_color="dodger blue") + donate_label.grid(row=4, column=0, columnspan=3, pady=5, sticky="ew") donate_label.bind( "