import pandas as pd
import pickle 
from pathlib import Path
import joblib
import os
import requests
from dotenv import load_dotenv

load_dotenv()

os.environ['CALENDARIFIC_API_KEY'] = os.getenv("CALENDARIFIC_API_KEY")

currentPath = Path(__file__).resolve().parent
modelfilePath = currentPath / "similarity.pkl"
moviesfilePath = currentPath / "movielist_images.pkl"

with open(modelfilePath, 'rb') as file:
    model = pickle.load(file)

with open(moviesfilePath,'rb') as file:
    movieslist = joblib.load(file)

def recommendMoviesByTitle(programmeTitle):
  recommended_movies = []
  if programmeTitle in movieslist['programmetitle'].values:
    movie_index = movieslist[movieslist['programmetitle'] == programmeTitle].index[0]
    sim_score = model[movie_index]
    #Take the top 10 results ONLY
    movie_list = sorted(list(enumerate(sim_score)), reverse = True, key = lambda x: x[1])[1:11]
    for i in movie_list:      
      recommended_movies.append({'programmetitle':movieslist.iloc[i[0]].programmetitle, 'imageurl': movieslist.iloc[i[0]].imagepath})
  else:
    return("No movies to recommend by title...")
  return recommended_movies

COUNTRY_CODE = "US"
def get_holidaysByScheduleDate(scheduleDate):    
    key = os.environ["CALENDARIFIC_API_KEY"]   
    current_date = pd.to_datetime(scheduleDate,errors="coerce")  
    year = current_date.year
    month = current_date.month
    holidaylist = []

    endpoint_url = f"https://calendarific.com/api/v2/holidays?api_key={key}&country={COUNTRY_CODE}&year={year}&month={month}"
    response = requests.get(endpoint_url)
    if response.status_code == 200:
        holidays = response.json()['response']['holidays']
        holidaylist = list(map(lambda x: x['description'],holidays))
        
        return list(set(holidaylist))
    else:        
        return []

def recommendMoviesByKeywordsAndGenres(keywordToSearchBy, genres, referenceProgrammeTitle):
    recommended_movies = []
    filtered_movies = []
    # Convert the list of genres to lowercase for consistent comparison
    genres = [genre.lower().strip() for genre in genres]

    if keywordToSearchBy:
        #search for event name in the programme title first 
        filtered_movies = movieslist[movieslist['programmetitle'].str.contains(keywordToSearchBy, case=False)]
        filtered_movies = filtered_movies[filtered_movies['genredescription'].apply(
            lambda genre_list: all(genre.lower() in [g.lower().strip() for g in genre_list.split(',')] for genre in genres))]
        if len(filtered_movies) == 0:
        # Step 1: Filter the DataFrame to find movies that match the given keyword
            filtered_movies = movieslist[movieslist['Name'].str.contains(keywordToSearchBy, case=False)]
            filtered_movies = filtered_movies[filtered_movies['genredescription'].apply(
            lambda genre_list: all(genre.lower() in [g.lower().strip() for g in genre_list.split(',')] for genre in genres))]

    if  len(filtered_movies) == 0:
        return recommendMoviesByTitle(referenceProgrammeTitle) 

    if filtered_movies.empty:
        return("No movies found with the given keyword and genres.")

    movies = filtered_movies.sample(len(filtered_movies))
    
    recommended_movies.append({'programmetitle':movies.iloc[0]['programmetitle'],'imageurl': movies.iloc[0]['imagepath']})

    movie_index = movies.index[0]    
    sim_score = model[movie_index]

    movie_list = sorted(list(enumerate(sim_score)), reverse=True, key=lambda x: x[1])[1:10]

    for i in movie_list:
        recommended_movies.append({'programmetitle':movieslist.iloc[i[0]].programmetitle,'imageurl': movieslist.iloc[i[0]].imagepath}) 
    return recommended_movies


def recommendProgrammes(scheduleDate : str, genres: list, referenceProgrammeTitle : str):
    holidaylist = get_holidaysByScheduleDate(scheduleDate)
    eventName = ""
    if any("Christmas" in sentence for sentence in holidaylist):
       eventName = "Christmas"
    elif any("New year" in sentence for sentence in holidaylist):
       eventName = "New year"
    elif any("Halloween" in sentence for sentence in holidaylist):
       eventName = "Halloween"
    elif any("Valentine" in sentence for sentence in holidaylist):
       eventName = "Valentine"
    elif any("Black Friday" in sentence for sentence in holidaylist):
       eventName = "Thanksgiving"

    return recommendMoviesByKeywordsAndGenres(eventName,genres=genres, referenceProgrammeTitle = referenceProgrammeTitle)