Update app.py

app.py

@@ -70,23 +70,15 @@ if st.button("Sentiment Analysis", type="secondary"):
                     for comment in comments:
                         timestamp = None
                         try:
-                            # Try a more direct XPath
-                            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:
-                            try:
-                                # Try a more general XPath
-                                timestamp_element = comment.find_element(By.XPATH, './ancestor::ytd-comment-renderer//yt-formatted-string[contains(@class, "time-text")]')
-                                timestamp = timestamp_element.text
-                            except Exception:
-                                try:
-                                    #try grabbing the a tag.
-                                    timestamp_element = comment.find_element(By.XPATH, './ancestor::ytd-comment-renderer//a[@id="time"]')
-                                    timestamp = timestamp_element.text
-                                except Exception as inner_e:
-                                    print(f"Timestamp not found for comment: {comment.text}. Error: {inner_e}") #debug
+                            comment_text = comment.text
+                            date_match = re.search(r'\d+ (day|week|month|year)s? ago', comment_text) #Example regex.
+                            if date_match:
+                                timestamp = date_match.group(0)
+                        except Exception as e:
+                            st.error(f"Error extracting date with regex: {e}")
                         data.append({"Comment": comment.text, "comment_date": timestamp})
 
+
                 except Exception as e:
                     st.error(f"Exception during comment extraction: {e}")
                 driver.quit()