Update app.py

app.py

@@ -107,7 +107,7 @@ if st.button("Sentiment Analysis", type="secondary"):
                 placeholder = st.empty()
                 progress_bar = st.progress(0)
 
-                for item in range(150):
+                for item in range(30):
                     try:
                         body = WebDriverWait(driver, 30).until(EC.visibility_of_element_located((By.TAG_NAME, "body")))
                         body.send_keys(Keys.END)
@@ -124,15 +124,15 @@ if st.button("Sentiment Analysis", type="secondary"):
                 data = []
                 try:
                     wait.until(EC.presence_of_element_located((By.CSS_SELECTOR, "#contents #contents")))
-                    comments = driver.find_elements(By.CSS_SELECTOR, "#content #content-text")
-                    comments = driver.find_element(By.CSS_SELECTOR,'div.author-info time')
-                    comments = comments.get_attribute('datetime')
+                    comment_elements = driver.find_elements(By.CSS_SELECTOR, "#content #content-text")
+                    comment_date_element = driver.find_element(By.CSS_SELECTOR,'div.author-info time')
+                    timestamp = comment_date_element.get_attribute('datetime')  # Get timestamp
                     user_id = 1
-                    for comment in comments:
-                        
-                        data.append({"User ID": user_id, "Comment": comment.text, "comment_date": timestamp})
+                    for comment_element in comment_elements:  # Iterate through comment elements
+                        data.append({"User ID": user_id, "Comment": comment_element.text, "comment_date": timestamp})
                         user_id += 1
-                    data = [dict(t) for t in {tuple(d.items()) for d in data}]
+                        data = [dict(t) for t in {tuple(d.items()) for d in data}]  # Remove duplicates
+    
                 except Exception as e:
                     st.error(f"Exception during comment extraction: {e}")
                 driver.quit()