from models.search_model import MovieSearch
from models.recommendation_model import Model
import gradio
import re
import pandas as pd

df = pd.read_csv('movie_data/movie_data.csv')

recommender = Model(df)
recommender.load('movie_data/_similarity')

movie = [[df['id'].iloc[i], df['title'].iloc[i], df['year'].iloc[i]] for i in range(len(df))]
corpus = df['title'].values.tolist()
stopwords = 'for a of the and to in - , is'.split()
search_model = MovieSearch(movie, corpus, stopwords)

def search_movie(title):
    search_res = re.compile(r'--res=\d+').findall(title)
    if(len(search_res) > 0):
        search_res = int(search_res[0].replace('--res=', ''))
    else:
        search_res = 10
    s_res = search_model.search(title, search_res)
    s_res = [i[0] for i in s_res]
    return(f'Search Results For "{title}"\n' + "\n".join([f"[{i[0]}] {i[1]} ({i[2]})" for i in s_res]))

def get_recommendation(ids):
    id = [int(id) for id in ids.split()]
    rec = recommender.forward(id)
    return(f'Movies That You Might Like\n' + "\n".join([f"- {i[0]} ({int(i[1])})" for i in rec]))

interface = gradio.Blocks()

with interface:
    gradio.Markdown('<center><h2>Movie Recommendation System</h3></center>')
    with gradio.Row():
        with gradio.Column():
            gradio.Markdown('Find the ID of the movie that you like')
            input = gradio.Textbox(
                label = 'Movie Title',
                placeholder = 'Example: batman'
            )
            output = gradio.Textbox(
                label = 'Search Result',
                lines = 15
            )
            search_button = gradio.Button('Search')
            search_button.click(
                fn = search_movie,
                inputs = input,
                outputs = output
            )
        with gradio.Column():
            gradio.Markdown('Get a movie recommendation')
            input = gradio.Textbox(
                label = 'Movie IDs, separated by space',
                placeholder = 'Example: 0 1 2'
            )
            output = gradio.Textbox(
                label = 'Movie Recommendation',
                lines = 15
            )
            search_button = gradio.Button('Get Recommendation')
            search_button.click(
                fn = get_recommendation,
                inputs = input,
                outputs = output
            )

def main():
    interface.launch()

if __name__ == '__main__':
    main()