Update app.py

app.py

@@ -76,6 +76,7 @@ site = st.sidebar.selectbox("Select one website", webs)
 job = st.sidebar.selectbox("Select one job", jobs)
 num_jobs = st.sidebar.selectbox("Select number of jobs you want to scrap", nums)
 
+# Function to get Selenium driver
 # Function to get Selenium driver
 @st.cache_resource
 def get_driver():
@@ -83,8 +84,16 @@ def get_driver():
     options.add_argument("--headless")  # Run in headless mode
     options.add_argument("--no-sandbox")  # Bypass OS security model
     options.add_argument("--disable-dev-shm-usage")  # Overcome limited resource problems
-    return webdriver.Chrome(service=Service(ChromeDriverManager().install()), options=options)
-
+    options.add_argument("--disable-gpu")  # Disable GPU acceleration if necessary
+    options.add_argument("--window-size=1920x1080")  # Set window size for headless mode
+    options.add_argument("--remote-debugging-port=9222")  # Enable remote debugging
+    
+    try:
+        driver = webdriver.Chrome(service=Service(ChromeDriverManager().install()), options=options)
+        return driver
+    except Exception as e:
+        st.error(f"Error initializing WebDriver: {e}")
+        return None  # Return None if driver initialization fails
 
 import streamlit as st
 from streamlit_option_menu import option_menu