From 0b9bc946d36f5acb32fcdeef7ab6ecc3289317c1 Mon Sep 17 00:00:00 2001
From: KRSHH <136873090+KRSHH@users.noreply.github.com>
Date: Wed, 18 Sep 2024 17:23:03 +0530
Subject: [PATCH] Use Icon

---
 modules/ui.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/modules/ui.py b/modules/ui.py
index dc497df..c0651ef 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -3,7 +3,7 @@ import webbrowser
 import customtkinter as ctk
 from typing import Callable, Tuple
 import cv2
-from PIL import Image, ImageOps
+from PIL import Image, ImageOps, ImageTk
 import tkinterdnd2 as tkdnd
 
 import modules.globals
@@ -220,6 +220,10 @@ def create_root(
     root.protocol("WM_DELETE_WINDOW", lambda: destroy())
     root.resizable(True, True)
 
+    # Add icon (replace 'icon.ico' with your actual icon file)
+    icon_path = os.path.join(os.path.dirname(__file__), "icon.ico")
+    root.iconbitmap(icon_path)
+
     main_frame = ctk.CTkFrame(root, fg_color="#1a1a1a")
     main_frame.pack(fill="both", expand=True, padx=20, pady=20)