Update app.py

app.py

@@ -70,9 +70,6 @@ with st.sidebar:
 ''')
 
 
-
-
-
 st.subheader("YouTube Comments Sentiment Analysis", divider="red")
 tokenizer = transformers.DistilBertTokenizer.from_pretrained("tabularisai/robust-sentiment-analysis")
 model = transformers.DistilBertForSequenceClassification.from_pretrained("tabularisai/robust-sentiment-analysis")
@@ -110,12 +107,13 @@ if st.button("Sentiment Analysis", type="secondary"):
                 placeholder = st.empty()
                 progress_bar = st.progress(0)
 
-                for item in range(30): #Reduced to 30 scrolls.
+                for item in range(30):
                     try:
-                        driver.execute_script("window.scrollBy(0, 500);") #javascript scroll
-                        WebDriverWait(driver, 10).until(
-                            EC.presence_of_element_located((By.CSS_SELECTOR, "#content #content-text"))
-                        )
+                        body = WebDriverWait(driver, 30).until(EC.visibility_of_element_located((By.TAG_NAME, "body")))
+                        body.send_keys(Keys.END)
+                        placeholder.text(f"Scrolled {item + 1} times")
+                        progress_bar.progress((item + 1) / 150)
+                        time.sleep(0.5)
                     except Exception as e:
                         st.error(f"Exception during scrolling: {e}")
                         break
@@ -134,7 +132,7 @@ if st.button("Sentiment Analysis", type="secondary"):
                         timestamp_text = timestamp_element.get_attribute('href')
                         timestamp_text = timestamp_text.split("lc=")[-1]
                         try:
-                            timestamp = datetime.fromtimestamp(int(timestamp_text,36)/1000).strftime("%Y-%m-%d %H:%M:%S")
+                            timestamp = datetime.fromtimestamp(int(timestamp_text,36)/1000).strftime("%Y-%m-%d")
                         except:
                             timestamp = "Timestamp not found"
                         data.append({"User ID": user_id, "Comment": comment.text, "comment_date": timestamp})
@@ -179,19 +177,19 @@ if st.button("Sentiment Analysis", type="secondary"):
                         file_name='Summary of the results.csv',
                         mime='text/csv',
                     )
+                
         else:
             st.warning("Please enter a URL.")
     else:
         st.warning(f"You have reached the maximum URL attempts ({max_attempts}).")
 
-if 'url_count' in st.session_state:
+if 'url_count' in st.session_state: #added if statement.
     st.write(f"URL pasted {st.session_state['url_count']} times.")
 
-            
 
-    
             
 
+