From f0c66732e77a31127911a9fea9042a31d0031311 Mon Sep 17 00:00:00 2001
From: KRSHH <136873090+KRSHH@users.noreply.github.com>
Date: Sun, 26 Jan 2025 23:20:08 +0530
Subject: [PATCH] Use Logo

---
 modules/ui.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/modules/ui.py b/modules/ui.py
index 7a14f68..cd8f03e 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -156,6 +156,11 @@ def create_root(start: Callable[[], None], destroy: Callable[[], None]) -> ctk.C
     )
     root.configure()
     root.protocol("WM_DELETE_WINDOW", lambda: destroy())
+    
+    # Add icon to the main window
+    icon_path = resolve_relative_path("deeplivecam.ico")
+    if os.path.exists(icon_path):
+        root.iconbitmap(icon_path)
 
     # Image Selection Area (Top)
     source_label = ctk.CTkLabel(root, text=None)
@@ -519,6 +524,11 @@ def create_preview(parent: ctk.CTkToplevel) -> ctk.CTkToplevel:
     preview.configure()
     preview.protocol("WM_DELETE_WINDOW", lambda: toggle_preview())
     preview.resizable(width=True, height=True)
+    
+    # Add icon to the preview window
+    icon_path = resolve_relative_path("deeplivecam.ico")
+    if os.path.exists(icon_path):
+        preview.iconbitmap(icon_path)
 
     preview_label = ctk.CTkLabel(preview, text=None)
     preview_label.pack(fill="both", expand=True)