vizdre's picture
Rename trail_7.py to app.py
657b55a verified
import pandas as pd
import streamlit as st
from streamlit_lottie import st_lottie
import requests
from bs4 import BeautifulSoup
import torch
from transformers import AutoTokenizer, AutoModelForSequenceClassification
import smtplib
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText
# Define Lottie animation URLs
animation_url1 = "https://lottie.host/dd7f2ccb-f1a4-46ab-9367-ac7a766c382f/1mpGXTnenM.json"
# Load Lottie animation data from URLs
def load_lottie_url(url):
r = requests.get(url)
if r.status_code != 200:
return None
return r.json()
st.header('Sentiment Analysis')
# Load RoBERTa model and tokenizer
roberta_model_name = "cardiffnlp/twitter-roberta-base-sentiment"
model = AutoModelForSequenceClassification.from_pretrained(roberta_model_name)
tokenizer = AutoTokenizer.from_pretrained(roberta_model_name)
labels = ['Negative', 'Neutral', 'Positive']
# Define function for sentiment analysis using RoBERTa
def analyze_sentiment(text):
encoded_text = tokenizer(text, return_tensors='pt', truncation=False, padding=True)
output = model(**encoded_text)
scores = torch.softmax(output.logits, dim=1).detach().numpy()[0]
sentiment_index = scores.argmax()
sentiment_label = labels[sentiment_index]
return sentiment_label, scores[sentiment_index]
# Define function to scrape reviews from an Amazon product page
def scrape_amazon_reviews(product_url):
response = requests.get(product_url)
soup = BeautifulSoup(response.content, 'html.parser')
review_elements = soup.find_all("div", class_="a-section review aok-relative")
reviews = []
for review_element in review_elements[:5]:
review_text = review_element.find("span", class_="review-text").text.strip()
reviews.append(review_text)
sentiment_label, sentiment_score = analyze_sentiment(review_text)
if sentiment_label == 'Negative':
send_email_alert(review_text) # Send email alert for negative review
return reviews
# Function to send email alert for negative review
def send_email_alert(review_text):
sender_email = "vaibhavdeori1@gmail.com"
receiver_email = "vaibhavdeori01@gmail.com"
password = "ejcp fpdq reek ewer"
message = MIMEMultipart()
message["From"] = sender_email
message["To"] = receiver_email
message["Subject"] = "Negative Review Alert"
body = f"The following review is negative:\n\n{review_text}"
message.attach(MIMEText(body, "plain"))
# Establish a connection with the SMTP server
with smtplib.SMTP("smtp.gmail.com", 587) as server:
server.starttls()
server.login(sender_email, password)
server.sendmail(sender_email, receiver_email, message.as_string())
# Analyze Text
with st.expander('Analyze Text'):
text = st.text_input('Text here: ')
if text:
sentiment_label, sentiment_score = analyze_sentiment(text)
st.write('Sentiment:', sentiment_label)
st.write('Confidence Score:', sentiment_score)
# Scrape Reviews from Amazon Product Page
with st.expander('URL of product'):
product_url = st.text_input('Enter the link to the Amazon product page:')
if product_url:
reviews = scrape_amazon_reviews(product_url)
if reviews:
st.write('Reviews scraped successfully:')
for review in reviews:
sentiment_label, sentiment_score = analyze_sentiment(review)
st.write('Review:', review)
st.write('Sentiment:', sentiment_label)
st.write('Confidence Score:', sentiment_score)
st.write('---')
else:
st.write('No reviews found.')
st_lottie(load_lottie_url(animation_url1), speed=1, height=200, key="lottie1")