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

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

@@ -20,9 +20,6 @@ def embedding_and_index():
 def load_model():
     model = AutoModel.from_pretrained("DeepPavlov/rubert-base-cased-sentence")
     return model
-    
-
-
 
 
 st.header("Подбор фильмов по описанию ✏️🔍")
@@ -35,39 +32,7 @@ model = load_model()
 
 # Пользовательский ввод
 user_input = st.text_input("Введите описание фильма:", value="", help="Чем подробнее будет ваше описание, тем точнее мы сможем подобрать для вас фильм 🤗'")
-
-genre_list = ['анимация',
-    'аниме',
-    'балет',
-    'биография',
-    'боевик',
-    'вестерн',
-    'военный',
-    'детектив',
-    'детский',
-    'документальный',
-    'драма',
-    'исторический',
-    'катастрофа',
-    'комедия',
-    'концерт',
-    'короткометражный',
-    'криминал',
-    'мелодрама',
-    'мистика',
-    'музыка',
-    'мюзикл',
-    'нуар',
-    'приключения',
-    'сборник',
-    'семейный',
-    'сказка',
-    'спорт',
-    'триллер',
-    'ужасы',
-    'фантастика',
-    'фэнтези',
-    'эротика']
+genre_list = ['анимация', 'аниме', 'балет', 'биография', 'боевик', 'вестерн', 'военный', 'детектив', 'детский', 'документальный', 'драма', 'исторический', 'катастрофа', 'комедия', 'концерт', 'короткометражный', 'криминал', 'мелодрама', 'мистика', 'музыка', 'мюзикл', 'нуар', 'приключения', 'сборник', 'семейный', 'сказка', 'спорт', 'триллер', 'ужасы', 'фантастика', 'фэнтези', 'эротика']
 
 user_select_genre = st.multiselect('Выберите жанр', genre_list)
 
@@ -79,6 +44,7 @@ if st.button("Искать🔍🎦"):
                 outputs = model(**tokens)
             embeddings = outputs.last_hidden_state.mean(dim=1)
             return embeddings.cpu().numpy().astype('float32')
+
         # Векторизация введенного запроса с использованием переданных tokenizer и model
         input_embedding = encode_description(user_input, tokenizer, model)
 
@@ -96,36 +62,34 @@ if st.button("Искать🔍🎦"):
                 genre_mask_i = df['genre'].str.contains(genres_selected.iloc[i])
                 genre_mask = genre_mask & genre_mask_i
             recs = recs[genre_mask]
-        
-        if recs.empty: 
-            st.subheader("Подходящих фильмов не найдено, ослабьте фильтры 😔:")
-        else: 
-
 
-# Вывод рекомендованных фильмов с изображениями
+        if not recs.empty:
+            # Вывод рекомендованных фильмов с изображениями
             st.subheader("Рекомендованные фильмы 🎉:")
-            for i in range(5):
+            for i in range(min(5, len(recs))):
                 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('Поделитесь с нами вашим мнением')
             
                 feedback_button = st.button("Отправить отзыв", key="feedback_button")
             
             if feedback_button and feedback:
-                feedback_container.success("Спасибо, каждый день мы стараемся быть лучше для вас 💟")
+                st.success("Спасибо, каждый день мы стараемся быть лучше для вас 💟")
             elif feedback_button:
-                feedback_container.warning("Пожалуйста, введите отзыв перед отправкой.")
-
+                st.warning("Пожалуйста, введите отзыв перед отправкой.")
+        else:
+            st.subheader("Подходящих фильмов не найдено, ослабьте фильтры 😔:")