Update stri.py

stri.py

@@ -28,7 +28,7 @@ length = 256
 # Определение запроса пользователя
 query = st.text_input("Введите запрос")
 
-col1, col2 = st.columns(2)
+col1, col2 = st.beta_columns(2)
 generate_button = col2.button('Сгенерировать')
 
 if generate_button:
@@ -60,14 +60,20 @@ if generate_button:
 
     num_books_per_page = col1.selectbox("Количество книг на странице:", [3, 5, 10], index=0)
 
-    for i in indices[:num_books_per_page]:
+    for i in indices:
         cols = st.beta_columns(2)  # Создание двух столбцов для размещения информации и изображения
         cols[1].write("## " + books['title'][i])
         cols[1].markdown("**Автор:** " + books['author'][i])
         cols[1].markdown("**Аннотация:** " + books['annotation'][i])
         image_url = books['image_url'][i]
+
         response = requests.get(image_url)
         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