def identify():
form = RatingForm()
if form.validate_on_submit():
rating = form.rating.data
print session['user_id'], 1, rating
cur = mysql.connection.cursor()
cur.execute("INSERT INTO user_article VALUES(%s, %s, %s)",
(session['user_id'], 1, rating))
mysql.connection.commit()
cur.close()
flash('Rating is submitted for this article', 'success')
return redirect(url_for('identify'))
return render_template('identify.html', form=form)
def home():
form = RatingForm()
# def clean_text(text):
# text = text.lower()
# clean = re.compile('<.*?>')
# text = re.sub(clean,'',text)
# text = re.sub(r"""[,.;@#?!&$]+\ *"""," ",text, flags=re.VERBOSE)
# text = text.replace('\r\n', '-').replace('\n', ' ')
# check = text.replace(" ", "").isalpha()
# if check:
# text = [word for word in word_tokenize(text)]
# else:
# text = [word for word in word_tokenize(text) if word.isalpha()]
# return text
# def stem(text):
# text = ' '.join([porter.stem(word) for word in text.split()])
# return text
if form.validate_on_submit():
review = form.review.data
#review = clean_text(review)
review = review.lower()
clean = re.compile('<.*?>')
review = re.sub(clean,'',review)
review = re.sub(r"""[,.;@#?!&$]+\ *"""," ",review, flags=re.VERBOSE)
review = review.replace('\r\n', '-').replace('\n', ' ')
review = [word for word in review.split() if word.isalpha()]
STOPWORDS= set(["a","about","above","after","again","against","am","an","and","any","are","as","at","be","been","before","being","below","between","both","by","can","could","did","do","does","doing","down","during","each","few","for","from","further","had","has","have","having","he","he'd","he'll","he's","her","here","here's","hers","herself","him","himself","his","how","how's","i","i'd","i'll","i'm","i've","in","into","is","isn't","it","it's","its","itself","let's","me","more","myself","of","off","on","once","only","or","other","ought","our","ours" ,"ourselves","out","over","own","same","she","she'd","she'll","she's","should","so","some","such","than","that","that's","the","their","theirs","them","hemselves","then","there","there's","these","they","they'd","they'll","they're","they've","this","those","through","to","too","under","until","up","very","was","we","we'd","we'll","we're","we've","were","what","what's","when","when's","where","where's","which","while","who","who's","whom","why","why's","with","would","you","you'd","you'll","you're","you've","your","yours","yourself","yourselves","return","hes","heres","hows","im","its","lets","shes","thats","theres","theyll","theyre","theyve","were","whats","whens","wheres","whos","whys","youd","youll","youre","youve"])
STOPWORDS_dict = dict.fromkeys(STOPWORDS, 0)
review = [word for word in review if word not in STOPWORDS_dict]
review = ' '.join(word for word in review)
# porter=PorterStemmer()
# review = ' '.join([porter.stem(word) for word in review.split()])
vocab = open('models/Vocab.pickle', 'rb')
vocab_pickled = pickle.load(vocab)
vectorizer = TfidfVectorizer(vocabulary=vocab_pickled.vocabulary_)
model = open('models/LinearRegression.pickle', 'rb')
linearregression_pickled = pickle.load(model)
p = vectorizer.fit_transform([review])
rate = round(linearregression_pickled.predict(p)[0],2)
flash(f'YOUR REVIEW : {review}','success')
flash(f'MY MODEL RATING : { rate }','success')
return redirect(url_for('predict'))
return render_template('PredictReviewRating.html',form=form)
def movie_rating(id):
form = RatingForm()
movie = mg.db.movie.find_one({'_id': id})
form.uid.data = current_user.uid
form.mid.data = movie['lens_id']
form.name.data = movie['title']
if form.validate_on_submit() and request.method == 'POST':
Rating(uid=int(form.uid.data),
mid=int(form.mid.data),
name=form.name.data,
rating=float(form.rating.data),
title=form.title.data,
content=form.content.data).save()
recommender.add_ratings(
[[int(form.uid.data),
int(form.mid.data),
float(form.rating.data)]])
flash('rating recorded, retraining model...')
return redirect(url_for('main.movie_rating', id=id))
return render_template('rating.html', movie=movie, form=form)
