Update pages/mpnet_base/mpnet_base_app.py

pages/mpnet_base/mpnet_base_app.py

@@ -55,75 +55,3 @@ def mpnet_base_page():
             st.write(f'**Время поиска:** {search_time:.4f} секунд')
             st.markdown(f"[Читать подробнее]({book['page_url']})", unsafe_allow_html=True)
 
-
-
-
-
-# # Загрузка данных
-# @st.cache_data
-# def load_data():
-#     data = pd.read_csv('Data/books_data2.csv')
-#     with open('pages/MiniLM/MiniLM_embeddings.pkl', 'rb') as f:
-#         book_embeddings = pickle.load(f)
-#     index = faiss.read_index('pages/MiniLM/MiniLM_index.index')
-
-#     embedder = SentenceTransformer("all-MiniLM-L6-v2")
-#     return data, index, embedder
-
-# # Функция поиска
-# def search_books(embedder, index, data, query, n_results):
-#     query_embedding = embedder.encode([query])
-#     D, I = index.search(query_embedding, n_results)
-    
-#     result_books = []
-#     for i in range(n_results):
-#         book_index = I[0][i]
-#         book_info = data.iloc[book_index]
-#         book_link = book_info['page_url']
-#         book_image = book_info['image_url']
-#         book_genre = book_info['genre']
-#         book_title = book_info['title']
-#         book_description = book_info['annotation']
-#         book_author = book_info['author']
-#         similarity_score = 1 / (1 + D[0][i])
-        
-#         result_books.append({
-#             'Link': book_link,
-#             'Image': book_image,
-#             'Genre': book_genre,
-#             'Title': book_title,
-#             'Description': book_description,
-#             'Author': book_author,
-#             'Similarity': similarity_score
-#         })
-    
-#     return result_books
-
-# # Streamlit
-# def MiniLm_page():
-#     st.title('Поиск книг')
-#     data, index, embedder = load_data()
-#     query = st.text_input('Введите цитату или автора:')
-#     n_results = st.slider("Количество результатов", min_value=1, max_value=20, value=5)
-
-#     if st.button("Искать"):
-#         if query:
-#             start_time = time.time() 
-#             result_books = search_books(embedder, index, data, query, n_results)
-#             end_time = time.time()  
-#             search_time = end_time - start_time 
-#             st.write("Результаты поиска:")
-            
-#             for book in result_books:
-#                 st.write('---')
-#                 st.image(book['Image'], width=250)
-#                 st.write(f"**Название:** {book['Title']}")
-#                 st.write(f"**Автор:** {book['Author']}")
-#                 st.write(f"**Жанр:** {book['Genre']}")
-#                 if len(book['Description']) > 50:
-#                     book['Description'] = ' '.join(book['Description'].split()[:50]) + '...'
-#                 st.write(f"**Описание:** {book['Description']}")
-#                 st.write(f"**Сходство:** {book['Similarity']:.2f}")
-#                 st.write(f'**Время поиска:** {search_time:.4f} секунд')
-#                 st.write(f"[Читать подробнее]({book['Link']})")
-#                 st.text("")