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",
 )
 
@@ -45,12 +45,12 @@ COLUMN_LABELS_RU = {
     "score": "Сходство",
     "registration_number": "Регистрационный номер",
     "name": "Название работы",
-    "author_surname": "Фамилия автора",
-    "author_name": "Имя автора",
-    "author_patronymic": "Отчество автора",
-    "author_organization__short_name": "Организация автора",
-    "dissertation_type": "Тип диссертации",
-    "created_date": "Дата создания / регистрации",
+    "author_surname": "Фамилия",
+    "author_name": "Имя",
+    "author_patronymic": "Отчество",
+    "author_organization__short_name": "Организация",
+    "dissertation_type": "Тип",
+    "created_date": "Дата создания",
     "abstract": "Аннотация",
 }
 
@@ -187,8 +187,11 @@ 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
+        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
+
+        return df_display, None
 
     results = search_core(query, top_k)
     df_res = build_result_df(results)
@@ -196,106 +199,79 @@ def run_search(query: str, top_k: int):
     # Переименовываем колонки в русский вариант
     df_res_ru = df_res.rename(columns=COLUMN_LABELS_RU)
 
-    # Формируем Excel в памяти (тоже с русскими заголовками)
+    # Для отображения: используем № как индекс, чтобы не было второй нумерации
+    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)
     output.seek(0)
 
-    return df_res_ru, output
+    return df_display, output
 
 
 # ==========================
 # UI НА STREAMLIT
 # ==========================
 
-st.title("Поиск релевантных диссертаций")
-st.caption(
-    "Введите текстовый запрос на естественном языке, например: "
-    "«пластификаторы для самоуплотняющихся бетонов»."
+st.title("Поиск диссертаций")
+
+# 4) Слайдер между заголовком и полем ввода
+top_k = st.slider(
+    "Сколько результатов показать",
+    min_value=1,
+    max_value=100,
+    value=20,
+    step=1,
 )
 
-# Инициализация значений по умолчанию для состояния
-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="Например: пластификаторы для самоуплотняющихся бетонов",
-        label_visibility="collapsed",  # скрываем текст "Запрос" сверху
-        key="query",
-    )
+# 5) Поле ввода: высота ~4 строки (text_area)
+query = st.text_area(
+    "Введите запрос",
+    height=120,  # примерно 4 строки
+    placeholder="Например: пластификаторы для самоуплотняющихся бетонов",
+    key="query",
+)
 
-    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 col3:
-        clear = st.button("🧹 Очистить")
+# Кнопки под полем поиска
+col1, col2 = st.columns([1, 1])
+with col1:
+    do_search = st.button("🔍 Поиск", type="primary")
+with col2:
+    clear = st.button("🧹 Очистить")
 
-    if clear:
-        st.session_state["query"] = ""
-        st.session_state["top_k"] = 20
-        st.experimental_rerun()
+# 3) Корректная очистка: не трогаем конкретные ключи после создания виджетов,
+# а чистим всё состояние и перезапускаем скрипт
+if clear:
+    st.session_state.clear()
+    st.experimental_rerun()
 
 # ---- ОСНОВНАЯ ОБЛАСТЬ РЕЗУЛЬТАТОВ ----
 if do_search:
     with st.spinner("Идёт поиск по базе диссертаций..."):
-        df_res, excel_bytes = run_search(st.session_state["query"], st.session_state["top_k"])
+        df_res, excel_bytes = run_search(query, top_k)
 
     if df_res.empty:
         st.warning("Ничего не найдено. Попробуйте изменить формулировку запроса.")
     else:
         st.success(f"Найдено записей: {len(df_res)}")
 
-        tab_table, tab_info = st.tabs(["Таблица результатов", "Описание полей"])
-
-        with tab_table:
-            st.dataframe(
-                df_res,
-                use_container_width=True,
-            )
+        # 6) Без вкладок — просто таблица + кнопка скачивания
+        st.dataframe(
+            df_res,
+            use_container_width=True,
+        )
 
-            if excel_bytes is not None:
-                st.download_button(
-                    label="💾 Скачать результаты в Excel",
-                    data=excel_bytes,
-                    file_name="search_results.xlsx",
-                    mime=(
-                        "application/vnd.openxmlformats-officedocument."
-                        "spreadsheetml.sheet"
-                    ),
-                )
-
-        with tab_info:
-            st.markdown(
-                """
-**Пояснения к полям:**
-
-- **№** — порядковый номер результата  
-- **Сходство** — косинусное сходство запроса с документом (чем выше, тем релевантнее)  
-- **Регистрационный номер** — регистрационный номер диссертации  
-- **Название работы** — полное название диссертации  
-- **Фамилия / Имя / Отчество автора** — ФИО автора  
-- **Организация автора** — организация, с которой связан автор  
-- **Тип диссертации** — вид диссертации  
-- **Дата создания / регистрации** — дата создания / регистрации работы  
-- **Аннотация** — краткое содержание диссертации
-                """
+        if excel_bytes is not None:
+            st.download_button(
+                label="💾 Скачать результаты в Excel",
+                data=excel_bytes,
+                file_name="search_results.xlsx",
+                mime=(
+                    "application/vnd.openxmlformats-officedocument."
+                    "spreadsheetml.sheet"
+                ),
             )
 else:
-    st.info("Введите запрос в поле выше и нажмите кнопку «Поиск».")
+    st.info("Введите запрос выше и нажмите кнопку «Поиск».")