Update app.py

app.py

@@ -38,9 +38,6 @@ import zipfile
 import os
 import subprocess
 
-import os
-import subprocess
-
 import streamlit as st
 import numpy as np
 import pandas as pd
@@ -77,23 +74,23 @@ 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
+from selenium import webdriver
+from selenium.webdriver.firefox.service import Service as FirefoxService
+from webdriver_manager.firefox import GeckoDriverManager
+
 @st.cache_resource
 def get_driver():
-    options = Options()
+    options = webdriver.FirefoxOptions()
     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
-    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
-    
+    options.add_argument("--no-sandbox")
+    options.add_argument("--disable-dev-shm-usage")
+
     try:
-        driver = webdriver.Chrome(service=Service(ChromeDriverManager().install()), options=options)
+        driver = webdriver.Firefox(service=FirefoxService(GeckoDriverManager().install()), options=options)
         return driver
     except Exception as e:
         st.error(f"Error initializing WebDriver: {e}")
-        return None  # Return None if driver initialization fails
+        return None
 
 import streamlit as st
 from streamlit_option_menu import option_menu