collaborative_ui.py

import streamlit as st
import pandas as pd
import numpy as np
from posters_printer import posters_printer
from collaborative_filtering import collaborative_filter


class collaborative_ui:

    ########## INITIALISE REQUIRED CLASSES' OBJECTS AND DATA #######
    def __init__(self, links_data):
        self.recommender = collaborative_filter(
            links_data
        )  # collaborative recommendation class object
        self.user_ratings = list()  # list of ratings inputted by the user
        self.movies_list = (
            self.recommender.get_movies_list()
        )  # list of available movies
        self.posters_printer = posters_printer()  # posters grid printer class object

    ################################################################

    def add_preference(self, movie_name, rating):
        exists = False
        for i in self.user_ratings:
            if i[0] == movie_name:
                exists = True
        if not exists:
            self.user_ratings.append(
                [movie_name, rating]
            )  ### append a user inputted rating to the existing list

    def drop_preference(self, movie_name, dummy=0):
        for i in range(len(self.user_ratings)):
            if (
                self.user_ratings[i][0] == movie_name
            ):  ### find the inputted rating for the selected movie and drop it
                del self.user_ratings[i]
                return

    def render(self):
        st.sidebar.write(
            """
        #### Rate a few movies
        """
        )

        ###### "ADD A RATING" SECTION ###################################################
        movie_name = st.sidebar.selectbox("Select a movie to add", self.movies_list)
        rating = st.sidebar.slider("Select a rating", min_value=1, max_value=5)
        submit_movie = st.sidebar.button(
            "Add movie rating", on_click=self.add_preference, args=(movie_name, rating)
        )
        ###############################################################################

        st.sidebar.write(" ")
        st.sidebar.write(" ")

        ######### MAIN UI ################################################################
        if len(self.user_ratings) > 0:
            cols = st.columns([2, 1])
            with cols[0].expander("Your ratings"):
                st.write(
                    pd.DataFrame(self.user_ratings)
                    .rename(columns={0: "Title", 1: "Rating"})
                    .set_index("Title")
                )

            ##### "REMOVE A RATING" SECTION ########################
            with cols[1].expander("Want to undo a rating ?"):
                delete_movie_name = st.selectbox(
                    "",
                    pd.DataFrame(self.user_ratings).rename(
                        columns={0: "Title", 1: "Rating"}
                    )["Title"],
                )
                remove_rating = st.button(
                    "Remove",
                    on_click=self.drop_preference,
                    args=((delete_movie_name, 0)),
                )
            #########################################################

        else:
            st.write("#### Try adding some of your own ratings fom the sidebar")

        self.posters_printer.print(rec=self.recommender.recommend(self.user_ratings))
        ###################################################################################