app & Helpers

Helpers.py

@@ -84,7 +84,13 @@ def get_user_recommendation(DataBase, Matrix,user_id,l=10):
     return recommendations
 
 
-
+def get_recommendation_item(dataBase,matrix, movie_name, n=10):
+  similar_scores = matrix[movie_name]
+  similar_scores = similar_scores.sort_values(ascending=False)
+  
+  # only return movies with more than 100 ratings
+  similar_scores = similar_scores[similar_scores.index.isin(dataBase[dataBase['number_of_ratings'] > 100].index)][:n]
+  return similar_scores
 
 
 if __name__ == '__main__':

app.py

@@ -4,7 +4,7 @@ import pandas as pd
 import pickle
 import gdown
 import os
-from Helpers import get_user_recommendation , train_model , get_user_recommendation_XGBoost
+from Helpers import get_user_recommendation , train_model , get_user_recommendation_XGBoost ,get_recommendation_item
 
 
 # Set page configuration
@@ -462,7 +462,8 @@ def main():
         if button:
             st.write("The rating bar here is token from our dataset and it's between 0 and 5.")
             if selected_movie:
-                recommendations = recommend(selected_movie, similarity_df, movies_df, ratings_df, links_df, k)
+                recommendations = get_recommendation_item(DB_df, similarity_df, selected_movie , k)
+                # recommendations = recommend(selected_movie, similarity_df, movies_df, ratings_df, links_df, k)
                 if recommendations:
                     st.write(f"Similar movies to '{selected_movie}':")
                     num_cols = 2