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

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

@@ -36,6 +36,41 @@ model = load_model()
 # Пользовательский ввод
 user_input = st.text_input("Введите описание фильма:", value="", help="Чем подробнее будет ваше описание, тем точнее мы сможем подобрать для вас фильм 🤗'")
 
+genre_list = ['анимация',
+    'аниме',
+    'балет',
+    'биография',
+    'боевик',
+    'вестерн',
+    'военный',
+    'детектив',
+    'детский',
+    'документальный',
+    'драма',
+    'исторический',
+    'катастрофа',
+    'комедия',
+    'концерт',
+    'короткометражный',
+    'криминал',
+    'мелодрама',
+    'мистика',
+    'музыка',
+    'мюзикл',
+    'нуар',
+    'приключения',
+    'сборник',
+    'семейный',
+    'сказка',
+    'спорт',
+    'триллер',
+    'ужасы',
+    'фантастика',
+    'фэнтези',
+    'эротика']
+
+user_select_genre = st.multiselect('Выберите жанр', genre_list)
+
 if st.button("Искать🔍🎦"):
     if user_input:
         def encode_description(description, tokenizer, model):
@@ -54,31 +89,43 @@ if st.button("Искать🔍🎦"):
         recs = df.iloc[sorted_indices[0]].reset_index(drop=True)
         recs.index = recs.index + 1
 
+        if user_select_genre: 
+            genres_selected = user_select_genre.values()
+            genre_mask = df['genre'].str.contains('')
+            for i in range(len(genres_selected)):
+                genre_mask_i = df['genre'].str.contains(genres_selected[i])
+                genre_mask = genre_mask * genre_mask_i
+            recs = recs[genre_mask]
+        
+        if recs.empty: 
+            st.subheader("Подходящих фильмов не найдено, ослабьте фильтры 😔:")
+        else: 
+
 
 # Вывод рекомендованных фильмов с изображениями
-        st.subheader("Рекомендованные фильмы 🎉:")
-        for i in range(5):
-            st.markdown(f"<span style='font-size:{20}px; color:purple'>{recs['movie_title'].iloc[i]}</span>", unsafe_allow_html=True)
-        # Создаем две колонки: одну для текста, другую для изображения
-            col1, col2 = st.columns([2, 1])
-    
-        # В  колонке отображаем название фильма, описание, роли и ссылку
-            col1.info(recs['description'].iloc[i])
-            col1.markdown(f"**В ролях:** {recs['actors'].iloc[i]}")
-            col1.markdown(f"**Фильм можно посмотреть [здесь]({recs['page_url'].iloc[i]})**")
-    
-        # В  колонке отображаем изображение
-            col2.image(recs['image_url'].iloc[i], caption=recs['movie_title'].iloc[i], width=200)
-        with st.sidebar:
-            st.info("""
-                 #### Мы смогли помочь вам с выбором? 
-                """)
-            feedback = st.text_input('Поделитесь с нами вашим мнением')
+            st.subheader("Рекомендованные фильмы 🎉:")
+            for i in range(5):
+                st.markdown(f"<span style='font-size:{20}px; color:purple'>{recs['movie_title'].iloc[i]}</span>", unsafe_allow_html=True)
+            # Создаем две колонки: одну для текста, другую для изображения
+                col1, col2 = st.columns([2, 1])
         
-            feedback_button = st.button("Отправить отзыв", key="feedback_button")
+            # В  колонке отображаем название фильма, описание, роли и ссылку
+                col1.info(recs['description'].iloc[i])
+                col1.markdown(f"**В ролях:** {recs['actors'].iloc[i]}")
+                col1.markdown(f"**Фильм можно посмотреть [здесь]({recs['page_url'].iloc[i]})**")
         
-        if feedback_button and feedback:
-            feedback_container.success("Спасибо, каждый день мы стараемся быть лучше для вас 💟")
-        elif feedback_button:
-            feedback_container.warning("Пожалуйста, введите отзыв перед отправкой.")
+            # В  колонке отображаем изображение
+                col2.image(recs['image_url'].iloc[i], caption=recs['movie_title'].iloc[i], width=200)
+            with st.sidebar:
+                st.info("""
+                     #### Мы смогли помочь вам с выбором? 
+                    """)
+                feedback = st.text_input('Поделитесь с нами вашим мнением')
+            
+                feedback_button = st.button("Отправить отзыв", key="feedback_button")
+            
+            if feedback_button and feedback:
+                feedback_container.success("Спасибо, каждый день мы стараемся быть лучше для вас 💟")
+            elif feedback_button:
+                feedback_container.warning("Пожалуйста, введите отзыв перед отправкой.")