-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathapplication.py
More file actions
67 lines (57 loc) · 2.28 KB
/
Copy pathapplication.py
File metadata and controls
67 lines (57 loc) · 2.28 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
import logging
from flask import Flask, render_template, request
import joblib
import pandas as pd
from sklearn.feature_extraction.text import TfidfVectorizer
import pickle as pkl
# Initialize Flask application
application = Flask(__name__)
app = application
# Configure logging
logging.basicConfig(filename='app.log', level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
# Load the model data
model_path = "book_recommendation_model.pkl"
try:
# file_path = 'book_recommendation_model.pkl'
# with open(file_path , 'rb') as f:
# model_data = pkl.load(f)
model_data = joblib.load(model_path)
kmeans_model = model_data['kmeans_model']
cluster_books = model_data['cluster_books']
except Exception as e:
logging.error(f"Error loading model data: {e}")
# Load the filtered dataset
filtered_df = pd.read_csv("filtered_books.csv")
# Create TF-IDF vectorizer
tfidf_vectorizer = TfidfVectorizer(stop_words='english')
tfidf_matrix = tfidf_vectorizer.fit_transform(filtered_df['genres'])
# Define function for book recommendation
def get_recommendations(genre, n_recommendations=5):
try:
genre_vector = tfidf_vectorizer.transform([genre])
cluster = kmeans_model.predict(genre_vector)[0]
cluster_books_titles = cluster_books[cluster]
recommendations = filtered_df[filtered_df['title'].isin(cluster_books_titles)].sample(n=n_recommendations)['title'].tolist()
return recommendations
except Exception as e:
logging.error(f"Error getting recommendations: {e}")
return []
# Define routes
@app.route('/')
def home():
genres = sorted(filtered_df['genres'].unique())
return render_template('index.html', genres=genres)
@app.route('/recommendation', methods=['POST'])
def recommendation():
if request.method == 'POST':
try:
genre = request.form['genre']
recommendations = get_recommendations(genre)
return render_template('recommendation.html', genre=genre, recommendations=recommendations)
except Exception as e:
logging.error(f"Error processing recommendation request: {e}")
return render_template('error.html')
else:
return render_template('index.html', genres=genres)
if __name__ == '__main__':
app.run("0.0.0.0")