from flask import Flask, jsonify, render_template, request
import tmdbsimple as tmdb
from yamr.dataset import EnhancedDataset
from yamr.recommender import RecommendationEngine
import tmdb_util
import util
app = Flask(__name__)
tmdb.API_KEY = tmdb._get_env_key('TMDB_API_KEY')
dataset_path = 'datasets/ml-latest-enhanced'
ds = EnhancedDataset(dataset_path)
recommender = RecommendationEngine(ds)
recommender.load_model('item_sim_model')
@app.route('/')
def index():
return render_template('index.html')
@app.route('/api/movies/<int:movieId>')
def movie_detail(movieId):
movie = ds.find_movie_by_id(movieId)
movie = tmdb_util.add_trailer(movie)
return jsonify(movie)
@app.route('/api/movies/top_rated')
from yamr.dataset import EnhancedDataset
from yamr.recommender import RecommendationEngine, ItemSimilarityRecommender, FactorizationRecommender, PopularityRecommender
dataset_path = 'datasets/ml-latest-enhanced'
ds = EnhancedDataset(dataset_path)
#recommender = ItemSimilarityRecommender()
#recommender = FactorizationRecommender()
recommender = PopularityRecommender()
re = RecommendationEngine(ds, recommender=recommender)
re.train_model()
re.save_model('popularity_model')
