ammar00200's picture
Upload 13 files
3aaa0a3 verified
from flask import Flask, jsonify, request
import pickle
import pandas as pd
import numpy as np
app = Flask(__name__)
# Load pickled data
# Open the file using 'with' statement
with open('popular1.pkl', 'rb') as f:
popular_df = pd.read_pickle(f)
with open('pt1.pkl', 'rb') as fi:
pt = pd.read_pickle(fi)
with open('banquet.pkl', 'rb') as fil:
banquets = pd.read_pickle(fil)
with open('similarity_scores1.pkl', 'rb') as file:
similarity_scores = pd.read_pickle(file)
# Define a route to get recommendations
@app.route('/recommend/<string:n>',methods=['GET'])
def recommend(n):
# Get user input from request
# Perform recommendation logic
if n not in pt.index:
return "Banquet not found in the index"
index = np.where(pt.index == n)[0][0]
similar_items = sorted(enumerate(similarity_scores[index]), key=lambda x: x[1], reverse=True)[1:5]
data = []
for i in similar_items:
item = []
temp_df = banquets[banquets['Hall-Name'] == pt.index[i[0]]]
item.extend(temp_df.drop_duplicates('Hall-Name')['Hall-Name'].values)
item.extend(temp_df.drop_duplicates('Hall-Name')['Address'].values)
item.extend(temp_df.drop_duplicates('Hall-Name')['Contact'].values)
item.extend(temp_df.drop_duplicates('Hall-Name')['Rating'].values)
item.extend(temp_df.drop_duplicates('Hall-Name')['Jn12ke src'].values)
data.append(item)
return jsonify({'recommendations': data})
if __name__ == "__main__":
app.run(debug=True)