Update stri.py

stri.py

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