Spaces:
Sleeping
Sleeping
Update programmeRecommendation.py
Browse files
programmeRecommendation.py
CHANGED
@@ -60,16 +60,16 @@ def recommendMoviesByKeywordsAndGenres(keywordToSearchBy, genres, referenceProgr
|
|
60 |
if keywordToSearchBy:
|
61 |
#search for event name in the programme title first
|
62 |
filtered_movies = movieslist[movieslist['programmetitle'].str.contains(keywordToSearchBy, case=False)]
|
|
|
|
|
63 |
if len(filtered_movies) == 0:
|
64 |
# Step 1: Filter the DataFrame to find movies that match the given keyword
|
65 |
filtered_movies = movieslist[movieslist['Name'].str.contains(keywordToSearchBy, case=False)]
|
|
|
|
|
66 |
|
67 |
if len(filtered_movies) == 0:
|
68 |
-
return recommendMoviesByTitle(referenceProgrammeTitle)
|
69 |
-
|
70 |
-
# Step 2: Further filter the movies based on the list of genres
|
71 |
-
filtered_movies = filtered_movies[filtered_movies['genredescription'].apply(
|
72 |
-
lambda genre_list: all(genre.lower() in [g.lower().strip() for g in genre_list.split(',')] for genre in genres))]
|
73 |
|
74 |
if filtered_movies.empty:
|
75 |
return("No movies found with the given keyword and genres.")
|
|
|
60 |
if keywordToSearchBy:
|
61 |
#search for event name in the programme title first
|
62 |
filtered_movies = movieslist[movieslist['programmetitle'].str.contains(keywordToSearchBy, case=False)]
|
63 |
+
filtered_movies = filtered_movies[filtered_movies['genredescription'].apply(
|
64 |
+
lambda genre_list: all(genre.lower() in [g.lower().strip() for g in genre_list.split(',')] for genre in genres))]
|
65 |
if len(filtered_movies) == 0:
|
66 |
# Step 1: Filter the DataFrame to find movies that match the given keyword
|
67 |
filtered_movies = movieslist[movieslist['Name'].str.contains(keywordToSearchBy, case=False)]
|
68 |
+
filtered_movies = filtered_movies[filtered_movies['genredescription'].apply(
|
69 |
+
lambda genre_list: all(genre.lower() in [g.lower().strip() for g in genre_list.split(',')] for genre in genres))]
|
70 |
|
71 |
if len(filtered_movies) == 0:
|
72 |
+
return recommendMoviesByTitle(referenceProgrammeTitle)
|
|
|
|
|
|
|
|
|
73 |
|
74 |
if filtered_movies.empty:
|
75 |
return("No movies found with the given keyword and genres.")
|