Update stri.py

stri.py

@@ -28,6 +28,8 @@ length = 256
 # Определение запроса пользователя
 query = st.text_input("Введите запрос")
 
+num_books_per_page = st.selectbox("Количество книг на странице:", [3, 5, 10], index=0)
+
 col1, col2 = st.columns(2)
 generate_button = col2.button('Сгенерировать')
 
@@ -58,9 +60,7 @@ if generate_button:
 
     indices = np.argsort(cosine_similarities)[::-1]  # Сортировка по убыванию
 
-    num_books_per_page = col1.selectbox("Количество книг на странице:", [3, 5, 10], index=0)
-
-    for i in indices:
+    for i in indices[:num_books_per_page]:
         cols = st.columns(2)  # Создание двух столбцов для размещения информации и изображения
         cols[1].write("## " + books['title'][i])
         cols[1].markdown("**Автор:** " + books['author'][i])
@@ -71,9 +71,4 @@ if generate_button:
         image = Image.open(BytesIO(response.content))
         cols[0].image(image)
         cols[0].write(cosine_similarities[i])
-        cols[1].write("---")
-
-        num_books_per_page -= 1
-
-        if num_books_per_page == 0:
-            break
+        cols[1].write("---")