| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778 |
- import re
- import requests
- from bs4 import BeautifulSoup
- import sentiment
- from imdb import Cinemagoer
- def fetch_movie(input):
- ia = Cinemagoer()
- search = ia.search_movie(title=input)
- if(len(search)==0):
- print("Not Found")
- id = "tt"+search[0].movieID
- data = search[0].data
- title = data['title']
- poster_url = data['cover url']
- url = f'https://www.imdb.com/title/{id}/reviews/'
- return url, title, poster_url
- def html_removal(text):
- text = re.compile(r'<[^>]+>').sub('', text)
- return text
- def reviews_func(url):
-
- page = requests.get(url=url)
- soup = BeautifulSoup(page.content, "html.parser")
- reviews = soup.find_all("div",class_="text show-more__control")
- reviews = list(reviews)
- for i in range(len(reviews)):
- reviews[i] = html_removal(str(reviews[i]))
- return reviews
- def sentiment_analysis(movie, language = "English"):
- analysis = dict()
- url, title, poster_url = fetch_movie(movie)
- analysis['title'] = title
- reviews = reviews_func(url)
- total = len(reviews)
- positive = 0
- negative = 0
- for review in reviews:
- score = sentiment.calc_score(str(review))
- if score == 'Positive':
- positive+=1
- elif score == 'Negative':
- negative+=1
-
- if(positive>negative):
- overall_sentiment = 'Positive'
- percentage = f"{round((positive/total),2)*100}%"
- elif(positive<negative):
- overall_sentiment = 'Negative'
- percentage = f"{round((negative/total),2)*100}%"
- else:
- overall_sentiment = 'Neutral'
- percentage = "50%"
-
- analysis['sentiment'] = overall_sentiment
- analysis['percentage'] = percentage
- if(language == "English"):
- return analysis
- elif(language == "Hindi"):
- if(analysis['sentiment'] == "Positive"):
- analysis['sentiment'] = 'उत्तम'
- elif(analysis['sentiment'] == "Negative"):
- analysis['sentiment'] = 'खराब'
- return analysis
|
