Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import io
 from typing import Optional, Tuple, List
+from datetime import datetime, timezone
 
 import numpy as np
 import pandas as pd
@@ -22,7 +23,6 @@ MODEL_NAME = os.getenv("MODEL_NAME")
 
 SLIDER_MIN_YEAR = 2005  # фиксированный минимум
 
-# Для Excel (полный набор)
 COLUMN_LABELS_RU_EXCEL = {
     "№": "№",
     "score": "Сходство",
@@ -97,7 +97,6 @@ DEFAULT_SCIENCES = {
     "Биологические",
 }
 
-# Паттерны для сопоставления с degree_pursued (на случай разных формулировок)
 SCIENCE_PATTERNS = {
     "Архитектура": ["архитектур"],
     "Биологические": ["биолог"],
@@ -123,7 +122,6 @@ SCIENCE_PATTERNS = {
 
 
 def _keyify(label: str) -> str:
-    # стабильный ключ для streamlit
     return "k_" + "".join(ch if ch.isalnum() else "_" for ch in label).strip("_")
 
 
@@ -133,13 +131,11 @@ def _keyify(label: str) -> str:
 
 @st.cache_data(show_spinner="Загрузка данных...")
 def load_data():
-    # 1) META
     ds_meta = load_dataset(HF_MERGED_REPO, split="train")
     df_meta = ds_meta.to_pandas()
     df_meta["registration_number"] = df_meta["registration_number"].astype(str)
     df_meta = df_meta.set_index("registration_number", drop=False)
 
-    # 2) EMB
     ds_emb = load_dataset(HF_EMB_REPO, split="train")
     df_emb = ds_emb.to_pandas()
     df_emb["registration_number"] = df_emb["registration_number"].astype(str)
@@ -150,7 +146,6 @@ def load_data():
     norms = np.linalg.norm(emb_matrix, axis=1, keepdims=True)
     emb_matrix = emb_matrix / np.maximum(norms, 1e-8)
 
-    # выравниваем meta под reg_nums
     meta_aligned = df_meta.reindex(reg_nums)
 
     # dissertation_type masks
@@ -208,15 +203,13 @@ def build_filter_mask(
 ) -> np.ndarray:
     mask = np.ones(len(reg_nums), dtype=bool)
 
-    # 1) Тип диссертации
     type_mask = np.zeros(len(reg_nums), dtype=bool)
     if candidate_selected:
         type_mask |= is_candidate
     if doctor_selected:
         type_mask |= is_doctor
-    mask &= type_mask  # если оба выключены -> всё False
+    mask &= type_mask
 
-    # 2) Науки
     if science_selected:
         sci_mask = np.zeros(len(reg_nums), dtype=bool)
         for label in science_selected:
@@ -224,7 +217,6 @@ def build_filter_mask(
             sci_mask |= _contains_any(degree_lower, patterns)
         mask &= sci_mask
 
-    # 3) Годы (NaN пропускаем)
     if year_range is not None:
         y0, y1 = int(year_range[0]), int(year_range[1])
         yr = year_arr
@@ -321,18 +313,11 @@ def run_search(
     science_selected: List[str],
     year_range: Optional[Tuple[int, int]],
 ):
-    query = query.strip()
-    if not query:
-        return pd.DataFrame(), None, pd.DataFrame()
-
     mask = build_filter_mask(candidate_selected, doctor_selected, science_selected, year_range)
     results = search_core(query, top_k, mask=mask)
     df_raw = build_result_df(results)
 
-    if df_raw.empty:
-        return pd.DataFrame(), None, df_raw
-
-    # --- UI dataframe (скрываем №, рег.номер, тип) ---
+    # UI: скрываем №, reg, type
     df_ui = df_raw.copy()
     df_ui = df_ui.rename(
         columns={
@@ -344,11 +329,9 @@ def run_search(
             "vak_link": "ВАК",
         }
     )
-
-    # Убираем не нужные колонки (как вы просили)
     df_ui = df_ui.drop(columns=["№", "registration_number", "dissertation_type"])
 
-    # --- Excel: полный набор ---
+    # Excel: полный набор
     df_excel_ru = df_raw.rename(columns=COLUMN_LABELS_RU_EXCEL)
     output = io.BytesIO()
     with pd.ExcelWriter(output, engine="xlsxwriter") as writer:
@@ -367,6 +350,16 @@ st.markdown(
     unsafe_allow_html=True,
 )
 
+# Подготовка session_state
+if "last_df_ui" not in st.session_state:
+    st.session_state.last_df_ui = None
+if "last_df_raw" not in st.session_state:
+    st.session_state.last_df_raw = None
+if "last_excel" not in st.session_state:
+    st.session_state.last_excel = None
+if "requests_log" not in st.session_state:
+    st.session_state.requests_log = []  # заглушка "БД"
+
 data_has_years = np.isfinite(year_arr).any()
 year_max = int(np.nanmax(year_arr)) if data_has_years else None
 
@@ -417,74 +410,135 @@ with st.form("search_form"):
     with c2:
         do_search = st.form_submit_button("🔍 Поиск", type="primary", use_container_width=True)
 
+# Выполняем поиск и сохраняем результаты (чтобы они не исчезали на rerun)
 if do_search:
     if not candidate_selected and not doctor_selected:
         st.warning("Выключены оба типа диссертаций. Включите «Кандидатские» и/или «Докторские».")
-        st.stop()
-
-    with st.spinner("Идёт поиск по базе диссертаций..."):
-        df_ui, excel_bytes, df_raw = run_search(
-            query=query,
-            top_k=top_k,
-            candidate_selected=candidate_selected,
-            doctor_selected=doctor_selected,
-            science_selected=science_selected,
-            year_range=year_range,
+    else:
+        with st.spinner("Идёт поиск по базе диссертаций..."):
+            df_ui, excel_bytes, df_raw = run_search(
+                query=query,
+                top_k=top_k,
+                candidate_selected=candidate_selected,
+                doctor_selected=doctor_selected,
+                science_selected=science_selected,
+                year_range=year_range,
+            )
+        st.session_state.last_df_ui = df_ui
+        st.session_state.last_df_raw = df_raw
+        st.session_state.last_excel = excel_bytes
+
+# ======= БЛОК РЕЗУЛЬТАТОВ (показывается, если есть сохранённые данные) =======
+df_ui_saved = st.session_state.last_df_ui
+df_raw_saved = st.session_state.last_df_raw
+excel_saved = st.session_state.last_excel
+
+if isinstance(df_ui_saved, pd.DataFrame) and not df_ui_saved.empty:
+    st.success(f"Найдено записей: {len(df_ui_saved)}")
+
+    df_ui_show = df_ui_saved[["Сходство", "ФИО", "Название диссертации", "Организация", "Год", "ВАК"]].copy()
+
+    # Сортировка по клику на заголовок + чекбоксы выбора строк
+    event = st.dataframe(
+        df_ui_show,
+        use_container_width=True,
+        hide_index=True,
+        column_order=["Сходство", "ФИО", "Название диссертации", "Организация", "Год", "ВАК"],
+        column_config={
+            "Сходство": st.column_config.NumberColumn("Сходство", format="%.4f", width="small"),
+            "ФИО": st.column_config.TextColumn("ФИО", width="medium"),
+            "Название диссертации": st.column_config.TextColumn("Название диссертации", width="large"),
+            "Организация": st.column_config.TextColumn("Организация", width="medium"),
+            "Год": st.column_config.NumberColumn("Год", width="small"),
+            "ВАК": st.column_config.LinkColumn("ВАК", display_text="открыть", width="small"),
+        },
+        on_select="rerun",
+        selection_mode="multi-row",
+    )
+
+    selected_rows = getattr(event, "selection", None).rows if getattr(event, "selection", None) else []
+    selected_rows = list(selected_rows or [])
+
+    # Детали (перенос текста) для выбранных
+    if selected_rows:
+        st.markdown("### Выбранные записи (предпросмотр)")
+        for i in selected_rows[:10]:
+            r = df_ui_show.iloc[i]
+            st.markdown(f"- **{r['ФИО']}** — {r['Название диссертации']} ({r['Год']})")
+        if len(selected_rows) > 10:
+            st.caption(f"Показаны первые 10 из {len(selected_rows)} выбранных.")
+
+    # Кнопка Excel
+    if excel_saved is not None:
+        st.download_button(
+            label="💾 Скачать результаты в Excel",
+            data=excel_saved,
+            file_name="search_results.xlsx",
+            mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
         )
 
-    if df_ui.empty:
-        st.warning("Ничего не найдено. Попробуйте изменить запрос и/или ослабьте фильтры.")
-    else:
-        st.success(f"Найдено записей: {len(df_ui)}")
-
-        # ВАЖНО:
-        # st.dataframe даёт сортировку по клику, но перенос текста в ячейках не поддерживает.
-        # Чтобы читать длинные названия без горизонтального скролла — ниже блок "Детали по выбранной строке".
-        df_ui_show = df_ui[["Сходство", "ФИО", "Название диссертации", "Организация", "Год", "ВАК"]].copy()
-
-        event = st.dataframe(
-            df_ui_show,
-            use_container_width=True,
-            hide_index=True,
-            column_order=["Сходство", "ФИО", "Название диссертации", "Организация", "Год", "ВАК"],
-            column_config={
-                "Сходство": st.column_config.NumberColumn("Сходство", format="%.4f", width="small"),
-                "ФИО": st.column_config.TextColumn("ФИО", width="medium"),
-                "Название диссертации": st.column_config.TextColumn("Название диссертации", width="large"),
-                "Организация": st.column_config.TextColumn("Организация", width="medium"),
-                "Год": st.column_config.NumberColumn("Год", width="small"),
-                "ВАК": st.column_config.LinkColumn("ВАК", display_text="открыть", width="small"),
-            },
-            on_select="rerun",
-            selection_mode="single-row",
+    # ======= ЗАГЛУШКА "Запрос расширенной информации" =======
+    st.markdown("---")
+    st.subheader("Запрос расширенной информации")
+
+    with st.form("request_form"):
+        requester_fio = st.text_input("Ваше ФИО", placeholder="Иванов Иван Иванович")
+        requester_email = st.text_input("Email", placeholder="name@example.com")
+        requester_note = st.text_area(
+            "Дополнительная информация",
+            height=120,
+            placeholder="Что именно вы хотите уточнить/получить по выбранным диссертациям?",
         )
 
-        # Детали с переносом строк (без горизонтального скролла)
-        sel = getattr(event, "selection", None)
-        sel_rows = sel.rows if sel is not None else []
-        if sel_rows:
-            i = sel_rows[0]
-            row = df_ui_show.iloc[i].to_dict()
-            st.markdown("### Детали по выбранной записи")
-            st.markdown(f"**ФИО:** {row.get('ФИО', '')}")
-            st.markdown(f"**Организация:** {row.get('Организация', '')}")
-            st.markdown(f"**Год:** {row.get('Год', '')}")
-            st.markdown(f"**Сходство:** {row.get('Сходство', '')}")
-            st.markdown("**Название диссертации:**")
-            st.write(row.get("Название диссертации", ""))  # переносится естественно
-            if row.get("ВАК"):
-                st.markdown(f"**ВАК:** {row.get('ВАК')}")
-
-        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",
-            )
+        st.caption(f"Выбрано записей: {len(selected_rows)}")
+        send_request = st.form_submit_button("📨 Отправить запрос", type="primary", use_container_width=True)
+
+    if send_request:
+        if not requester_fio.strip() or not requester_email.strip():
+            st.warning("Заполните «Ваше ФИО» и «Email».")
+        elif len(selected_rows) == 0:
+            st.warning("Выберите хотя бы одну запись в таблице (чекбоксами слева).")
+        else:
+            # Собираем payload (заглушка для отправки в БД)
+            # Берём данные из df_raw_saved по тем же индексам.
+            selected_payload_items = []
+            for i in selected_rows:
+                raw = df_raw_saved.iloc[i].to_dict()
+                selected_payload_items.append(
+                    {
+                        "author_fio": raw.get("fio"),
+                        "title": raw.get("title"),
+                        "org": raw.get("author_org_short"),
+                        "year": raw.get("protection_year"),
+                        "vak_link": raw.get("vak_link"),
+                        "registration_number": raw.get("registration_number"),
+                    }
+                )
+
+            payload = {
+                "created_at_utc": datetime.now(timezone.utc).isoformat(),
+                "requester": {
+                    "fio": requester_fio.strip(),
+                    "email": requester_email.strip(),
+                    "note": requester_note.strip(),
+                },
+                "items": selected_payload_items,
+            }
+
+            # "Отправка в БД" — заглушка: сохраняем в session_state и показываем
+            st.session_state.requests_log.append(payload)
+
+            st.success("Запрос принят (заглушка). Ниже — данные, которые будут отправляться в БД.")
+            st.json(payload)
+
+            with st.expander("История запросов (заглушка)", expanded=False):
+                st.write(f"Всего запросов в текущей сессии: {len(st.session_state.requests_log)}")
+                st.json(st.session_state.requests_log[-1])
+
 else:
-    st.info("Введите запрос выше и нажмите кнопку «Поиск» или нажмите Ctrl+Enter в поле ввода.")
+    st.info("Введите запрос и нажмите «Поиск». После этого результаты будут доступны для выбора и отправки запроса.")
 
+# Футер
 st.markdown(
     "<p style='font-size: 0.8rem; text-align: right; color: gray;'>(с) Антон Лощилов, 2025</p>",
     unsafe_allow_html=True,