Update app.py

app.py

@@ -10,6 +10,7 @@ from selenium.webdriver.chrome.service import Service
 from selenium.webdriver.chrome.options import Options
 from webdriver_manager.chrome import ChromeDriverManager
 from huggingface_hub import InferenceClient
+from selenium.common.exceptions import WebDriverException
 
 # Check for Hugging Face API key
 api_key = os.getenv('access')
@@ -63,36 +64,38 @@ def monitor_urls(storage_location, url1, url2, scrape_interval, content_type):
         options.add_argument("--no-sandbox")
         options.add_argument("--disable-dev-shm-usage")
         
-        with webdriver.Chrome(service=Service(ChromeDriverManager().install()), options=options) as driver:
-            try:
-                while True:
-                    for i, url in enumerate(urls):
-                        try:
-                            driver.get(url)
-                            time.sleep(2)  # Wait for the page to load
-                            if content_type == "text":
-                                current_content = driver.page_source
-                            elif content_type == "media":
-                                current_content = driver.find_elements_by_tag_name("img")
-                            else:
-                                current_content = driver.page_source
-                            
-                            current_hash = hashlib.md5(str(current_content).encode('utf-8')).hexdigest()
-                            
-                            if current_hash != previous_hashes[i]:
-                                previous_hashes[i] = current_hash
-                                date_time_str = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-                                history.append(f"Change detected at {url} on {date_time_str}")
-                                csv_toolkit.writerow({"date": date_time_str.split()[0], "time": date_time_str.split()[1], "url": url, "change": "Content changed"})
-                                logging.info(f"Change detected at {url} on {date_time_str}")
-                        except Exception as e:
-                            logging.error(f"Error accessing {url}: {e}")
-                    
-                    time.sleep(scrape_interval * 60)  # Check every scrape_interval minutes
-            except KeyboardInterrupt:
-                logging.info("Monitoring stopped by user.")
-            finally:
-                driver.quit()
+        try:
+            with webdriver.Chrome(service=Service(ChromeDriverManager().install()), options=options) as driver:
+                try:
+                    while True:
+                        for i, url in enumerate(urls):
+                            try:
+                                driver.get(url)
+                                time.sleep(2)  # Wait for the page to load
+                                if content_type == "text":
+                                    current_content = driver.page_source
+                                elif content_type == "media":
+                                    current_content = driver.find_elements_by_tag_name("img")
+                                else:
+                                    current_content = driver.page_source
+                                
+                                current_hash = hashlib.md5(str(current_content).encode('utf-8')).hexdigest()
+                                
+                                if current_hash != previous_hashes[i]:
+                                    previous_hashes[i] = current_hash
+                                    date_time_str = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+                                    history.append(f"Change detected at {url} on {date_time_str}")
+                                    csv_toolkit.writerow({"date": date_time_str.split()[0], "time": date_time_str.split()[1], "url": url, "change": "Content changed"})
+                                    logging.info(f"Change detected at {url} on {date_time_str}")
+                            except Exception as e:
+                                logging.error(f"Error accessing {url}: {e}")
+                        
+                        time.sleep(scrape_interval * 60)  # Check every scrape_interval minutes
+                except KeyboardInterrupt:
+                    logging.info("Monitoring stopped by user.")
+        except WebDriverException as e:
+            logging.error(f"WebDriverException: {e}")
+            raise
 
 # Define main function to handle user input
 def handle_input(task, storage_location, url1, url2, scrape_interval, content_type):