SoumyaJ commited on
Commit
038e989
·
verified ·
1 Parent(s): 68d35c8

Update programmeRecommendation.py

Browse files
Files changed (1) hide show
  1. programmeRecommendation.py +5 -5
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.")