Update app.py

app.py

@@ -1,25 +1,24 @@
 import gradio as gr
 from selenium import webdriver
-from selenium.common.exceptions import WebDriverException
 from selenium.webdriver.common.by import By
 from selenium.webdriver.common.keys import Keys
-from selenium.webdriver.chrome.service import Service as ChromeService
-from webdriver_manager.chrome import ChromeDriverManager
+from selenium.webdriver.firefox.service import Service as FirefoxService
+from webdriver_manager.firefox import GeckoDriverManager
 from PIL import Image
 from io import BytesIO
 import time
 
 def take_screenshot_and_login(url, email, password):
-    options = webdriver.ChromeOptions()
+    options = webdriver.FirefoxOptions()
     options.add_argument('--headless')
     options.add_argument('--no-sandbox')
     options.add_argument('--disable-dev-shm-usage')
     
     wd = None
     try:
-        # Use ChromeDriverManager to automatically manage the ChromeDriver executable
-        service = ChromeService(ChromeDriverManager().install())
-        wd = webdriver.Chrome(service=service, options=options)
+        # Use GeckoDriverManager to automatically manage the Geckodriver executable
+        service = FirefoxService(GeckoDriverManager().install())
+        wd = webdriver.Firefox(service=service, options=options)
         
         wd.set_window_size(1080, 720)
         wd.get(url)
@@ -37,8 +36,8 @@ def take_screenshot_and_login(url, email, password):
         
         # Take a screenshot
         screenshot = wd.get_screenshot_as_png()
-    except WebDriverException as e:
-        print(f"WebDriverException occurred: {e}")
+    except Exception as e:
+        print(f"Exception occurred: {e}")
         return Image.new('RGB', (1080, 720), color='white')
     finally:
         if wd: