Update app.py

app.py

@@ -1,7 +1,8 @@
 import streamlit as st
 import pandas as pd
 import time
-import shutil
+import os
+import subprocess
 import chromedriver_autoinstaller
 from selenium import webdriver
 from selenium.webdriver.common.by import By
@@ -10,7 +11,13 @@ from selenium.webdriver.chrome.options import Options
 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"])
+
 def scrape_redfin(zipcode):
+    install_chrome()  # Ensure Chrome/Chromium is installed
     chromedriver_autoinstaller.install()  # Ensure the correct chromedriver version is installed
     
     options = Options()
@@ -22,11 +29,7 @@ def scrape_redfin(zipcode):
     options.add_argument("start-maximized")
     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")
     
-    # Find the installed Chrome binary path
-    chrome_path = shutil.which("google-chrome") or shutil.which("chrome")
-    if chrome_path:
-        options.binary_location = chrome_path
-    
+    options.binary_location = "/usr/bin/chromium-browser"  # Use Chromium
     driver = webdriver.Chrome(options=options)
     url = f"https://www.redfin.com/zipcode/{zipcode}"
     driver.get(url)
@@ -107,6 +110,7 @@ if st.button("Scrape Data"):
 
 
 
+
 ## working best code ever
 
 # import streamlit as st