HaggiVaggi commited on
Commit
be3d993
·
1 Parent(s): 2545913

Update pages/1 Подбор фильмов по описанию✏️🔍.py

Browse files
pages/1 Подбор фильмов по описанию✏️🔍.py CHANGED
@@ -90,11 +90,11 @@ if st.button("Искать🔍🎦"):
90
  recs.index = recs.index + 1
91
 
92
  if user_select_genre:
93
- genres_selected = user_select_genre.values()
94
  genre_mask = df['genre'].str.contains('')
95
  for i in range(len(genres_selected)):
96
- genre_mask_i = df['genre'].str.contains(genres_selected[i])
97
- genre_mask = genre_mask * genre_mask_i
98
  recs = recs[genre_mask]
99
 
100
  if recs.empty:
 
90
  recs.index = recs.index + 1
91
 
92
  if user_select_genre:
93
+ genres_selected = pd.Series(user_select_genre)
94
  genre_mask = df['genre'].str.contains('')
95
  for i in range(len(genres_selected)):
96
+ genre_mask_i = df['genre'].str.contains(genres_selected.iloc[i])
97
+ genre_mask = genre_mask & genre_mask_i
98
  recs = recs[genre_mask]
99
 
100
  if recs.empty: