Update app.py

app.py

@@ -12,9 +12,10 @@ from selenium.webdriver.support.ui import WebDriverWait
 from selenium.webdriver.support import expected_conditions as EC
 
 def install_chrome():
-    if not os.path.exists("/usr/bin/google-chrome"):
-        subprocess.run(["apt-get", "update"])
-        subprocess.run(["apt-get", "install", "-y", "chromium-browser"])
+    if not os.path.exists("/usr/bin/chromium-browser"):
+        subprocess.run(["apt-get", "update"], check=True)
+        subprocess.run(["apt-get", "install", "-y", "chromium-browser"], check=True)
+    os.environ["PATH"] += os.pathsep + "/usr/bin/"
 
 def scrape_redfin(zipcode):
     install_chrome()  # Ensure Chrome/Chromium is installed
@@ -30,7 +31,8 @@ def scrape_redfin(zipcode):
     options.add_argument("user-agent=Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36")
     
     options.binary_location = "/usr/bin/chromium-browser"  # Use Chromium
-    driver = webdriver.Chrome(options=options)
+    service = Service(chromedriver_autoinstaller.install())
+    driver = webdriver.Chrome(service=service, options=options)
     url = f"https://www.redfin.com/zipcode/{zipcode}"
     driver.get(url)
     
@@ -111,6 +113,7 @@ if st.button("Scrape Data"):
 
 
 
+
 ## working best code ever
 
 # import streamlit as st