Update app.py

app.py

@@ -67,41 +67,26 @@ def scrape_klikindomaret(nama_barang, num_items):
 @st.cache_data
 def scrape_shopee(nama_barang, num_items):
     products = []
-
- 
-    #path = ''
-    
-    #Customize chrome display
-    #chrome_options = Options()
-    #chrome_options.add_argument('--no-sandbox')
-    #chrome_options.add_argument('--headless')
-    #chrome_options.add_argument('--disable-notifications')
-    #chrome_options.add_argument('--disable-infobars')
+    page = 1
+    query = quote(nama_barang)
     
-    #Customize chrome display huggingface
-    options = webdriver.ChromeOptions()
+    options = webdriver.FirefoxOptions()
     options.add_argument('--no-sandbox')
     options.add_argument('--headless')
     options.add_argument('--disable-notifications')
     options.add_argument('--disable-infobars')
     options.add_argument('--disable-dev-shm-usage')
+    
+    driver = webdriver.Firefox(options=options)
 
 
     try :
-        page = 1
-        query = quote(nama_barang)
-        #driver = webdriver.Chrome(executable_path = path, options = chrome_options)
-        driver = webdriver.Chrome(options = options)
+
         url = f'https://shopee.co.id/search?keyword={query}&page={page}'
         driver.get(url)
 
-        
-        # Cari elemen berdasarkan tagname HTML
-        html_element = driver.find_element(By.TAG_NAME, "html")
+        html = driver.page_source
 
-        # Dapatkan HTML dari elemen
-        html = html_element.get_attribute("innerHTML")
-        #html = driver.execute_script("return document.getElementsByTagName('html')[0].innerHTML")
         soup = BeautifulSoup(html, "html.parser")
         
         for i in soup.find_all('div', class_ = "ie3A+n bM+7UW Cve6sh" ):
@@ -206,15 +191,15 @@ def scrape_tokped_with_selenium(nama_barang, num_items):
     products = []
     page = 1
     query = quote(nama_barang)
-    driver = webdriver.Chrome(options = options)
 
-    
-    options = webdriver.ChromeOptions()
+    options = webdriver.FirefoxOptions()
     options.add_argument('--no-sandbox')
     options.add_argument('--headless')
     options.add_argument('--disable-notifications')
     options.add_argument('--disable-infobars')
     options.add_argument('--disable-dev-shm-usage')
+    
+    driver = webdriver.Firefox(options=options)
 
     while len(products) < num_items :
         try :