Update app.py

app.py

@@ -18,21 +18,15 @@ st.set_page_config(
 )
 
 # ---- Hugging Face ----
-# Секрет HF_TOKEN нужно задать в Settings → Variables and secrets вашего Space.
 HF_TOKEN = os.getenv("HF_TOKEN")
-
-# repo_id приватного датасета с файлами ALL_data.xlsx и embeddings_full.json
-# Можешь зашить строкой или задать в переменной окружения HF_DATA_REPO.
 HF_DATA_REPO = os.getenv("HF_DATA_REPO", "yogl/dissertation-search-data")
 
-# Имена файлов в датасете
 ALL_DATA_FILENAME = "ALL_data.xlsx"
 EMBEDDINGS_FILENAME = "embeddings_full.json"
 
-# Модель для эмбеддингов
 MODEL_NAME = os.getenv("MODEL_NAME", "intfloat/multilingual-e5-base")
 
-# Порядок колонок в выдаче
+# Порядок колонок в выдаче (внутренние имена)
 COLUMNS_ORDER = [
     "registration_number",
     "name",
@@ -45,6 +39,20 @@ COLUMNS_ORDER = [
     "abstract",
 ]
 
+# Отображаемые заголовки (русские имена колонок)
+COLUMN_LABELS_RU = {
+    "№": "№",
+    "score": "Сходство",
+    "registration_number": "Регистрационный номер",
+    "name": "Название работы",
+    "author_surname": "Фамилия автора",
+    "author_name": "Имя автора",
+    "author_patronymic": "Отчество автора",
+    "author_organization__short_name": "Организация автора",
+    "dissertation_type": "Тип диссертации",
+    "created_date": "Дата создания / регистрации",
+    "abstract": "Аннотация",
+}
 
 if HF_TOKEN is None:
     st.error(
@@ -179,18 +187,22 @@ def run_search(query: str, top_k: int):
     query = query.strip()
     if not query:
         empty_df = pd.DataFrame(columns=["№", "score"] + COLUMNS_ORDER)
+        empty_df = empty_df.rename(columns=COLUMN_LABELS_RU)
         return empty_df, None
 
     results = search_core(query, top_k)
     df_res = build_result_df(results)
 
-    # Формируем Excel в памяти
+    # Переименовываем колонки в русский вариант
+    df_res_ru = df_res.rename(columns=COLUMN_LABELS_RU)
+
+    # Формируем Excel в памяти (тоже с русскими заголовками)
     output = io.BytesIO()
     with pd.ExcelWriter(output, engine="xlsxwriter") as writer:
-        df_res.to_excel(writer, index=False)
+        df_res_ru.to_excel(writer, index=False)
     output.seek(0)
 
-    return df_res, output
+    return df_res_ru, output
 
 
 # ==========================
@@ -203,43 +215,53 @@ st.caption(
     "«пластификаторы для самоуплотняющихся бетонов»."
 )
 
-with st.sidebar:
-    st.subheader("Параметры поиска")
-    query = st.text_area(
-        "Запрос",
-        height=120,
+# Инициализация значений по умолчанию для состояния
+if "query" not in st.session_state:
+    st.session_state["query"] = ""
+if "top_k" not in st.session_state:
+    st.session_state["top_k"] = 20
+
+# ---- ПОИСКОВАЯ ПАНЕЛЬ СВЕРХУ ----
+with st.container():
+    # Имитация "поисковика": большое поле ввода по центру
+    query = st.text_input(
+        label="Запрос",
+        value=st.session_state["query"],
         placeholder="Например: пластификаторы для самоуплотняющихся бетонов",
-    )
-    top_k = st.slider(
-        "Сколько результатов показать",
-        min_value=1,
-        max_value=100,
-        value=20,
-        step=1,
+        label_visibility="collapsed",  # скрываем текст "Запрос" сверху
+        key="query",
     )
 
-    col_btn1, col_btn2 = st.columns(2)
-    with col_btn1:
+    col1, col2, col3 = st.columns([2, 1, 1])
+    with col1:
+        top_k = st.slider(
+            "Сколько результатов показать",
+            min_value=1,
+            max_value=100,
+            value=st.session_state["top_k"],
+            step=1,
+            key="top_k",
+        )
+    with col2:
         do_search = st.button("🔍 Поиск", type="primary")
-    with col_btn2:
+    with col3:
         clear = st.button("🧹 Очистить")
 
     if clear:
-        # Полный сброс состояния
-        st.session_state.clear()
+        st.session_state["query"] = ""
+        st.session_state["top_k"] = 20
         st.experimental_rerun()
 
-# Основная область
+# ---- ОСНОВНАЯ ОБЛАСТЬ РЕЗУЛЬТАТОВ ----
 if do_search:
     with st.spinner("Идёт поиск по базе диссертаций..."):
-        df_res, excel_bytes = run_search(query, top_k)
+        df_res, excel_bytes = run_search(st.session_state["query"], st.session_state["top_k"])
 
     if df_res.empty:
         st.warning("Ничего не найдено. Попробуйте изменить формулировку запроса.")
     else:
         st.success(f"Найдено записей: {len(df_res)}")
 
-        # Можно разделить на вкладки: таблица и детали по столбцам
         tab_table, tab_info = st.tabs(["Таблица результатов", "Описание полей"])
 
         with tab_table:
@@ -264,14 +286,16 @@ if do_search:
                 """
 **Пояснения к полям:**
 
-- `registration_number` — регистрационный номер диссертации  
-- `name` — название работы  
-- `author_surname`, `author_name`, `author_patronymic` — ФИО автора  
-- `author_organization__short_name` — организация автора  
-- `dissertation_type` — тип диссертации  
-- `created_date` — дата создания / регистрации  
-- `abstract` — аннотация
+- **№** — порядковый номер результата  
+- **Сходство** — косинусное сходство запроса с документом (чем выше, тем релевантнее)  
+- **Регистрационный номер** — регистрационный номер диссертации  
+- **Название работы** — полное название диссертации  
+- **Фамилия / Имя / Отчество автора** — ФИО автора  
+- **Организация автора** — организация, с которой связан автор  
+- **Тип диссертации** — вид диссертации  
+- **Дата создания / регистрации** — дата создания / регистрации работы  
+- **Аннотация** — краткое содержание диссертации
                 """
             )
 else:
-    st.info("Введите запрос в левом сайдбаре и нажмите кнопку «Поиск».")
+    st.info("Введите запрос в поле выше и нажмите кнопку «Поиск».")