From 32955415ffd5b043d2483cb712b8d60c56f2c853 Mon Sep 17 00:00:00 2001
From: SoClose <33631880+SoClosee@users.noreply.github.com>
Date: Tue, 3 Mar 2026 14:48:06 +0100
Subject: [PATCH] fix: add error handling for webdriver initialization

---
 main.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/main.py b/main.py
index 886b7dc..9e69e69 100644
--- a/main.py
+++ b/main.py
@@ -153,7 +153,7 @@ def discover_images(folder_path):
 
 
 def create_driver(headless=False):
-    """Create and return a configured Chrome WebDriver."""
+    """Create and return a configured Chrome WebDriver with error handling."""
     options = webdriver.ChromeOptions()
     options.add_experimental_option("excludeSwitches", ["enable-logging"])
     options.add_argument("--disable-notifications")
@@ -166,9 +166,12 @@ def create_driver(headless=False):
     try:
         service = Service(ChromeDriverManager().install())
         driver = webdriver.Chrome(service=service, options=options)
-    except Exception as exc:
+    except WebDriverException as exc:
         logger.error("Failed to start Chrome: %s", exc)
-        logger.info("Make sure Google Chrome is installed on your system.")
+        logger.info("Make sure Google Chrome is installed on your system and the correct version of chromedriver is available.")
+        sys.exit(1)
+    except Exception as exc:
+        logger.error("An unexpected error occurred while starting Chrome: %s", exc)
         sys.exit(1)
 
     return driver