bugfix

pages/1 - Popularity-Based Recommender.py

@@ -56,14 +56,13 @@ def get_popular_recommendations_streaming(n, genres, country, url, headers):
             [lambda df: df["genres"].str.contains(genres, regex=True)]
             .sort_values('combined_rating', ascending=False)
             .head(n)
-            [['title', 'avg_rating', 'genres']]
+            [['title', 'avg_rating', 'genres', 'movieId']]
     )
     # merge recommendations with links df to get imdbIds for the API calls
     recommendations_ids =   (
                             recommendations
-                                .merge(movies, how = 'left', on = 'title')
                                 .merge(links, how = 'left', on = 'movieId')
-                                [['title', 'genres', 'imdbId']]
+                                # [['title', 'genres', 'imdbId']]
                             )
     recommendations_ids['imdbId'] = 'tt0' + recommendations_ids['imdbId'].astype('str')
     imdb_ids = list(recommendations_ids['imdbId'])
@@ -82,7 +81,7 @@ def get_popular_recommendations_streaming(n, genres, country, url, headers):
         for streaming_service in streaming_info['streamingInfo']:
             recommendations_ids.loc[recommendations_ids['imdbId'] == id, 'Streaming Availability'] += f"{streaming_service}: {streaming_info['streamingInfo'][streaming_service][country]['link']} \n" 
 
-        return recommendations_ids[['title', 'genres', 'Streaming Availability']]
+    return recommendations_ids[['title', 'genres', 'Streaming Availability']]
 
 def transform_genre_to_regex(genres):
     regex = ""