Update app.py

app.py

@@ -13,7 +13,7 @@ from huggingface_hub import hf_hub_download
 # ==========================
 
 st.set_page_config(
-    page_title="Поиск диссертаций",
+    page_title="Поиск постдока",
     layout="wide",
 )
 
@@ -61,14 +61,12 @@ if HF_TOKEN is None:
     )
     st.stop()
 
-
 # ==========================
 # ЗАГРУЗКА ДАННЫХ И МОДЕЛИ
 # ==========================
 
 @st.cache_data(show_spinner="Скачиваю и загружаю данные из приватного датасета...")
 def load_data():
-    # Скачиваем файлы из приватного датасета с помощью токена
     all_data_path = hf_hub_download(
         repo_id=HF_DATA_REPO,
         filename=ALL_DATA_FILENAME,
@@ -83,12 +81,10 @@ def load_data():
         token=HF_TOKEN,
     )
 
-    # Загружаем Excel с метаданными
     df_all = pd.read_excel(all_data_path)
     df_all["registration_number"] = df_all["registration_number"].astype(str)
     df_all = df_all.set_index("registration_number", drop=False)
 
-    # Загружаем эмбеддинги
     with open(embeddings_path, "r", encoding="utf-8") as f:
         emb_data = json.load(f)
 
@@ -98,7 +94,6 @@ def load_data():
         dtype="float32",
     )
 
-    # Нормируем эмбеддинги
     norms = np.linalg.norm(emb_matrix, axis=1, keepdims=True)
     emb_matrix = emb_matrix / np.maximum(norms, 1e-8)
 
@@ -188,21 +183,17 @@ def run_search(query: str, top_k: int):
     if not query:
         empty_df = pd.DataFrame(columns=["№", "score"] + COLUMNS_ORDER)
         empty_df_ru = empty_df.rename(columns=COLUMN_LABELS_RU)
-        # для отображения: ставим № как индекс, чтобы не было двойной нумерации
-        df_display = empty_df_ru.set_index("№") if "№" in empty_df_ru.columns else empty_df_ru
-
+        df_display = (
+            empty_df_ru.set_index("№") if "№" in empty_df_ru.columns else empty_df_ru
+        )
         return df_display, None
 
     results = search_core(query, top_k)
     df_res = build_result_df(results)
 
-    # Переименовываем колонки в русский вариант
     df_res_ru = df_res.rename(columns=COLUMN_LABELS_RU)
-
-    # Для отображения: используем № как индекс, чтобы не было второй нумерации
     df_display = df_res_ru.set_index("№")
 
-    # Формируем Excel в памяти (с русскими заголовками и явным столбцом №)
     output = io.BytesIO()
     with pd.ExcelWriter(output, engine="xlsxwriter") as writer:
         df_res_ru.to_excel(writer, index=False)
@@ -215,9 +206,13 @@ def run_search(query: str, top_k: int):
 # UI НА STREAMLIT
 # ==========================
 
-st.title("Поиск диссертаций")
+# Заголовок по центру
+st.markdown(
+    "<h1 style='text-align: center; margin-bottom: 0.5rem;'>Поиск постдока</h1>",
+    unsafe_allow_html=True,
+)
 
-# 4) Слайдер между заголовком и полем ввода
+# Слайдер
 top_k = st.slider(
     "Сколько результатов показать",
     min_value=1,
@@ -226,28 +221,28 @@ top_k = st.slider(
     step=1,
 )
 
-# 5) Поле ввода: высота ~4 строки (text_area)
+# Поле ввода: ~4 строки
 query = st.text_area(
     "Введите запрос",
-    height=120,  # примерно 4 строки
+    height=120,
     placeholder="Например: пластификаторы для самоуплотняющихся бетонов",
     key="query",
 )
 
-# Кнопки под полем поиска
-col1, col2 = st.columns([1, 1])
-with col1:
+# Кнопки: Поиск по центру, Очистить справа
+btn_col1, btn_col2, btn_col3 = st.columns([1, 1, 1])
+with btn_col1:
+    pass
+with btn_col2:
     do_search = st.button("🔍 Поиск", type="primary")
-with col2:
+with btn_col3:
     clear = st.button("🧹 Очистить")
 
-# 3) Корректная очистка: не трогаем конкретные ключи после создания виджетов,
-# а чистим всё состояние и перезапускаем скрипт
 if clear:
     st.session_state.clear()
     st.experimental_rerun()
 
-# ---- ОСНОВНАЯ ОБЛАСТЬ РЕЗУЛЬТАТОВ ----
+# Результаты
 if do_search:
     with st.spinner("Идёт поиск по базе диссертаций..."):
         df_res, excel_bytes = run_search(query, top_k)
@@ -257,7 +252,6 @@ if do_search:
     else:
         st.success(f"Найдено записей: {len(df_res)}")
 
-        # 6) Без вкладок — просто таблица + кнопка скачивания
         st.dataframe(
             df_res,
             use_container_width=True,
@@ -275,3 +269,11 @@ if do_search:
             )
 else:
     st.info("Введите запрос выше и нажмите кнопку «Поиск».")
+
+# Футер
+st.markdown(
+    "<p style='font-size: 0.8rem; text-align: center; color: gray;'>"
+    "(с) Антон Лощилов, 2025"
+    "</p>",
+    unsafe_allow_html=True,
+)