from flask import Flask, render_template, request, redirect, url_for, session, flash
import pandas as pd
import joblib
from fuzzywuzzy import process
from flask_bcrypt import Bcrypt
from functools import wraps
import os
from supabase import create_client, Client
from dotenv import load_dotenv


load_dotenv()

app = Flask(__name__)
app.secret_key = os.getenv("SECRET_KEY")
bcrypt = Bcrypt(app)



movies = pd.read_csv('core/data/processed_movies_with_posters.csv')
cosine_sim = joblib.load('core/model/cosine_sim.pkl')



def get_recommendations(title, cosine_sim=cosine_sim):
    title = title.lower()
    if title not in movies['title'].str.lower().values:
        close_matches = process.extract(title, movies['title'].str.lower().values, limit=5)
        return None, [movies[movies['title'].str.lower() == match[0]].iloc[0] for match in close_matches]
    idx = movies[movies['title'].str.lower() == title].index[0]
    sim_scores = list(enumerate(cosine_sim[idx]))
    sim_scores = sorted(sim_scores, key=lambda x: x[1], reverse=True)
    sim_scores = sim_scores[1:11]
    movie_indices = [i[0] for i in sim_scores]
    return movies.iloc[movie_indices], None

def get_recommendations_by_id(movie_id, cosine_sim=cosine_sim):
    idx = movies[movies['id'] == movie_id].index[0]
    sim_scores = list(enumerate(cosine_sim[idx]))
    sim_scores = sorted(sim_scores, key=lambda x: x[1], reverse=True)
    sim_scores = sim_scores[1:6]
    movie_indices = [i[0] for i in sim_scores]
    return movies.iloc[movie_indices]

@app.route('/')

def home():
    return render_template('recommendation.html', movies=movies.sample(20).to_dict(orient='records'))

@app.route('/movie/<int:id>')
def movie_details(id):
    movie = movies[movies['id'] == id].iloc[0]
    recommendations = get_recommendations_by_id(id).to_dict(orient='records')
    return render_template('movie_details.html', movie=movie, recommendations=recommendations)

@app.route('/recommend', methods=['POST'])
def recommend():
    title = request.form['title']
    recommendations, close_matches = get_recommendations(title)
    if recommendations is None:
        flash("Movie title not found. Did you mean one of these?", 'warning')
        return render_template('recommendation.html', movies=[match.to_dict() for match in close_matches])
    return render_template('recommendation.html', movies=recommendations.to_dict(orient='records'))

@app.route('/search', methods=['GET', 'POST'])
def search():
    if request.method == 'POST':
        query = request.form['query']
        results = movies[movies['title'].str.contains(query, case=False, na=False)]
        return render_template('search.html', movies=results.to_dict(orient='records'))
    return render_template('search.html', movies=None)

@app.route('/filter', methods=['GET', 'POST'])
def filter():
    genres = sorted(movies['genre'].str.split(',', expand=True).stack().dropna().unique())
    languages = sorted(movies['original_language'].dropna().unique())

    if request.method == 'POST':
        selected_genre = request.form.get('genre')
        selected_language = request.form.get('language')

        if not selected_genre and not selected_language:
            return render_template('filter.html', movies=None, genres=genres, languages=languages,
                                   error_message="No movies found. Please adjust your filters.")

        filtered_movies = movies.copy()

        if selected_genre:
            filtered_movies = filtered_movies[filtered_movies['genre'].str.contains(selected_genre, na=False)]

        if selected_language:
            filtered_movies = filtered_movies[filtered_movies['original_language'] == selected_language]

        filtered_movies['genre'] = filtered_movies['genre'].fillna('').astype(str)

        sample_size = min(50, len(filtered_movies))

        filtered_movies = filtered_movies.sample(sample_size).to_dict(orient='records')

        return render_template('filter.html', movies=filtered_movies, genres=genres, languages=languages,
                               error_message=None)

    return render_template('filter.html', movies=None, genres=genres, languages=languages, error_message=None)

if __name__ == '__main__':
    app.run(host='0.0.0.0', port=7860)