Spaceai / app.py
Artificial-superintelligence's picture
Create app.py
a4ea8a9 verified
raw
history blame
1.07 kB
import streamlit as st
from selenium import webdriver
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.common.by import By
from webdriver_manager.chrome import ChromeDriverManager
from selenium.webdriver.chrome.options import Options
def setup_chrome():
chrome_options = Options()
chrome_options.add_argument("--headless") # Run in headless mode
chrome_options.add_argument("--no-sandbox") # Required for cloud environments
chrome_options.add_argument("--disable-dev-shm-usage") # Avoid shared memory issues
chrome_options.add_argument("--disable-gpu") # Disable GPU acceleration
# Initialize WebDriver
driver = webdriver.Chrome(service=Service(ChromeDriverManager().install()), options=chrome_options)
return driver
def main():
st.title("Selenium in Streamlit Cloud")
if st.button("Run Chrome"):
driver = setup_chrome()
driver.get("https://www.google.com")
st.write("Page title:", driver.title)
driver.quit()
if __name__ == "__main__":
main()