feat: add support for Firefox browser compatibility

main.py

@@ -6,10 +6,12 @@
 import click
 from selenium import webdriver
 from selenium.webdriver.chrome.service import Service as ChromeService
+from selenium.webdriver.firefox.service import Service as FirefoxService
 from webdriver_manager.chrome import ChromeDriverManager
+from webdriver_manager.firefox import GeckoDriverManager
 from selenium.common.exceptions import WebDriverException
 from lib_resume_builder_AIHawk import Resume, FacadeManager, ResumeGenerator, StyleManager
-from src.utils import chrome_browser_options
+from src.utils import chrome_browser_options, firefox_browser_options
 from src.llm.llm_manager import GPTAnswerer
 from src.aihawk_authenticator import AIHawkAuthenticator
 from src.aihawk_bot_facade import AIHawkBotFacade
@@ -149,11 +151,16 @@ def file_paths_to_dict(resume_file: Path | None, plain_text_resume_file: Path) -
 
         return result
 
-def init_browser() -> webdriver.Chrome:
+def init_browser(browser: str = 'chrome') -> webdriver.Chrome | webdriver.Firefox:
     try:
-        options = chrome_browser_options()
-        service = ChromeService(ChromeDriverManager().install())
-        return webdriver.Chrome(service=service, options=options)
+        if browser == 'chrome':
+            options = chrome_browser_options()
+            service = ChromeService(ChromeDriverManager().install())
+            return webdriver.Chrome(service=service, options=options)
+        elif browser == 'firefox':
+            options = firefox_browser_options()
+            service = FirefoxService(GeckoDriverManager().install())
+            return webdriver.Firefox(service=service, options=options)
     except Exception as e:
         raise RuntimeError(f"Failed to initialize browser: {str(e)}")
 
@@ -171,7 +178,7 @@ def create_and_run_bot(parameters, llm_api_key):
 
         job_application_profile_object = JobApplicationProfile(plain_text_resume)
 
-        browser = init_browser()
+        browser = init_browser(parameters['browser'])
         login_component = AIHawkAuthenticator(browser)
         apply_component = AIHawkJobManager(browser)
         gpt_answerer_component = GPTAnswerer(parameters, llm_api_key)
@@ -194,8 +201,9 @@ def create_and_run_bot(parameters, llm_api_key):
 
 @click.command()
 @click.option('--resume', type=click.Path(exists=True, file_okay=True, dir_okay=False, path_type=Path), help="Path to the resume PDF file")
+@click.option('--browser', type=click.Choice(['chrome', 'firefox']), default=None, help='Browser to use for the bot')
 @click.option('--collect', is_flag=True, help="Only collects data job information into data.json file")
-def main(collect: False, resume: Path = None):
+def main(collect: False, browser: str = 'chrome', resume: Path = None):
     try:
         data_folder = Path("data_folder")
         secrets_file, config_file, plain_text_resume_file, output_folder = FileManager.validate_data_folder(data_folder)
@@ -206,6 +214,7 @@ def main(collect: False, resume: Path = None):
         parameters['uploads'] = FileManager.file_paths_to_dict(resume, plain_text_resume_file)
         parameters['outputFileDirectory'] = output_folder
         parameters['collectMode'] = collect
+        parameters['browser'] = browser
 
         create_and_run_bot(parameters, llm_api_key)
     except ConfigError as ce:

src/utils.py

@@ -21,6 +21,7 @@
     logger.add(sys.stderr, level="DEBUG")
 
 chromeProfilePath = os.path.join(os.getcwd(), "chrome_profile", "linkedin_profile")
+firefoxProfilePath = os.path.join(os.getcwd(), "firefox_profile", "linkedin_profile")
 
 def ensure_chrome_profile():
     logger.debug(f"Ensuring Chrome profile exists at path: {chromeProfilePath}")
@@ -33,6 +34,16 @@ def ensure_chrome_profile():
         logger.debug(f"Created Chrome profile directory: {chromeProfilePath}")
     return chromeProfilePath
 
+def ensure_firefox_profile():
+    logger.debug(f"Ensuring Firefox profile exists at path: {firefoxProfilePath}")
+    profile_dir = os.path.dirname(firefoxProfilePath)
+    if not os.path.exists(profile_dir):
+        os.makedirs(profile_dir)
+        logger.debug(f"Created directory for Firefox profile: {profile_dir}")
+    if not os.path.exists(firefoxProfilePath):
+        os.makedirs(firefoxProfilePath)
+        logger.debug(f"Created Firefox profile directory: {firefoxProfilePath}")
+    return firefoxProfilePath
 
 def is_scrollable(element):
     scroll_height = element.get_attribute("scrollHeight")
@@ -153,6 +164,44 @@ def chrome_browser_options():
 
     return options
 
+def firefox_browser_options():
+    logger.debug("Setting Firefox browser options")
+    ensure_firefox_profile()
+    options = webdriver.FirefoxOptions()
+    options.add_argument("--start-maximized")
+    options.add_argument("--no-sandbox")
+    options.add_argument("--disable-dev-shm-usage")
+    options.add_argument("--ignore-certificate-errors")
+    options.add_argument("--disable-extensions")
+    options.add_argument("--disable-gpu")
+    options.add_argument("--disable-background-timer-throttling")
+    options.add_argument("--disable-backgrounding-occluded-windows")
+    options.add_argument("--disable-translate")
+    options.add_argument("--disable-popup-blocking")
+    options.add_argument("--no-first-run")
+    options.add_argument("--no-default-browser-check")
+    options.add_argument("--disable-logging")
+    options.add_argument("--disable-autofill")
+    options.add_argument("--disable-plugins")
+    options.add_argument("--disable-animations")
+    options.add_argument("--disable-cache")
+
+    prefs = {
+        "permissions.default.image": 2,
+        "permissions.default.stylesheet": 2,
+    }
+    for key, value in prefs.items():
+        options.set_preference(key, value)
+
+    if len(firefoxProfilePath) > 0:  # You'll need to define firefoxProfilePath similar to chromeProfilePath
+        profile = webdriver.FirefoxProfile(firefoxProfilePath)
+        options.profile = profile
+        logger.debug(f"Using Firefox profile directory: {firefoxProfilePath}")
+    else:
+        options.set_preference("browser.privatebrowsing.autostart", True)
+        logger.debug("Using Firefox in private browsing mode")
+
+    return options
 
 def printred(text):
     red = "\033[91m"