Update app.py

app.py

@@ -121,20 +121,23 @@ if st.button("Sentiment Analysis", type="secondary"):
                 placeholder.text("Scrolling complete.")
                 progress_bar.empty()
 
-                data = []
-                try:
-                    wait.until(EC.presence_of_element_located((By.CSS_SELECTOR, "#contents #contents")))
-                    comments = driver.find_elements(By.CSS_SELECTOR, "#content #content-text")
-                    user_id = 1
-                    for comment in comments:
-                        timestamp_elements = comment.find_elements(By.CSS_SELECTOR, 'div.author-info > time')  
-                        timestamp = timestamp_elements[0].get_attribute('datetime') if timestamp_elements else None
-                        data.append({"User ID": user_id, "Comment": comment.text, "comment_date": timestamp})
-                        user_id += 1
-                    data = [dict(t) for t in {tuple(d.items()) for d in data}]  
-    
-                except Exception as e:
-                    st.error(f"Exception during comment extraction: {e}")
+               try:
+    wait.until(EC.presence_of_element_located((By.CSS_SELECTOR, "#contents #contents")))
+    comments = driver.find_elements(By.CSS_SELECTOR, "#content #content-text")
+    user_id = 1
+    for comment in comments:
+        timestamp = None
+        try:
+            timestamp_element = comment.find_element(By.XPATH, './ancestor::ytd-comment-renderer//yt-formatted-string[@class="published-time-text style-scope ytd-comment-renderer"]')
+            timestamp = timestamp_element.text
+        except Exception as e:
+            print(f"Timestamp not found for comment: {comment.text}. Error: {e}")
+        data.append({"User ID": user_id, "Comment": comment.text, "comment_date": timestamp})
+        user_id += 1
+    data = [dict(t) for t in {tuple(d.items()) for d in data}]
+except Exception as e:
+    st.error(f"Exception during comment extraction: {e}")
+
                 driver.quit()
                 df = pd.DataFrame(data, columns=["User ID", "Comment", "comment_date"])
                 st.dataframe(df)