Подготовить AI Lead Processor и проверить обработку заявок

app.py

@@ -0,0 +1,119 @@
+"""Рабочее место менеджера сервисной компании."""
+import os
+from datetime import datetime
+from pathlib import Path
+from uuid import uuid4
+
+import streamlit as st
+from dotenv import load_dotenv
+
+from src.economics import calculate, estimate_request_usd
+from src.llm_service import classify
+from src.storage import append, snapshot
+
+
+ROOT = Path(__file__).resolve().parent
+load_dotenv(ROOT / ".env")
+JOURNAL = Path(os.getenv("LEADS_FILE", ROOT / "data" / "processed_leads.xlsx"))
+LABELS = {"urgent": "Срочно", "normal": "Не срочно", "spam": "Спам"}
+COLORS = {"urgent": "red", "normal": "green", "spam": "gray"}
+
+st.set_page_config(page_title="AI Lead Processor", layout="wide")
+st.title("AI Lead Processor — обработка клиентских заявок")
+st.write("ИИ анализирует входящие обращения, определяет их приоритет и подготавливает черновики ответов.")
+mode = st.radio("Режим обработки", ["REAL AI", "DEMO"], horizontal=True, key="mode")
+if mode == "DEMO":
+    st.info("DEMO · Локальные правила. Можно проверить весь процесс без ключа.")
+else:
+    st.caption("REAL AI · Текст передаётся в OpenAI. Для демонстрации используйте синтетические обращения.")
+
+def discard_pending():
+    st.session_state.pop("pending", None)
+
+text = st.text_area("Введите текст заявки клиента", key="lead", max_chars=4000,
+                    placeholder="Например, из стиральной машины течёт вода на пол.",
+                    on_change=discard_pending)
+if st.button("Обработать заявку", key="process", type="primary"):
+    discard_pending()
+    try:
+        with st.spinner("Обработка заявки…"):
+            result = classify(text, mode)
+        st.session_state.pending = (str(uuid4()), text.strip(), mode, result)
+    except ValueError as exc:
+        st.error(str(exc))
+        if mode == "REAL AI":
+            st.caption("Для демонстрации выберите DEMO и снова нажмите «Обработать заявку».")
+
+pending = st.session_state.get("pending")
+if pending and pending[2] != mode:
+    discard_pending()
+    pending = None
+if pending:
+    uid, original, used_mode, result = pending
+    st.subheader("Проверка заявки")
+    st.caption("Ожидает проверки менеджером")
+    st.markdown(f":{COLORS[result.category]}-background[**{LABELS[result.category]}**]")
+    st.text(result.reason)
+    usage = f"{used_mode} · {result.model}"
+    if used_mode == "REAL AI" and result.input_tokens is not None:
+        usage += f" · Входные токены: {result.input_tokens} · Выходные токены: {result.output_tokens}"
+    st.caption(usage)
+    category = st.selectbox("Категория после проверки", list(LABELS),
+                            index=list(LABELS).index(result.category), format_func=LABELS.get, key=f"category-{uid}")
+    reply = ""
+    if category != "spam":
+        reply = st.text_area("Черновик ответа клиенту", value=result.draft_reply, max_chars=4000, key=f"reply-{uid}")
+    if st.button("Подтвердить и сохранить", key="save"):
+        try:
+            if category != "spam" and not reply.strip():
+                raise ValueError("Заполните черновик ответа перед сохранением.")
+            cost = (0.0 if used_mode == "DEMO" else estimate_request_usd(
+                result.model, result.input_tokens, result.output_tokens, result.cached_tokens or 0))
+            reason = result.reason
+            if category != result.category:
+                reason = f"Менеджер изменил категорию «{LABELS[result.category]}» на «{LABELS[category]}». Исходная оценка: {reason}"
+            append(JOURNAL, {
+                "ID": uid, "Дата и время": datetime.now().astimezone().isoformat(timespec="seconds"),
+                "Текст заявки": original, "Категория": LABELS[category],
+                "Причина классификации": reason, "Черновик ответа": reply.strip(),
+                "Статус проверки": "Проверено менеджером", "Режим обработки": used_mode,
+                "Модель": result.model, "Входные токены": result.input_tokens,
+                "Выходные токены": result.output_tokens, "Оценочная стоимость API-запроса, USD": cost,
+            })
+            discard_pending()
+            st.session_state.saved = True
+            st.rerun()
+        except ValueError as exc:
+            st.error(str(exc))
+        except OSError:
+            st.error("Не удалось сохранить журнал. Закройте книгу в Excel, проверьте права на папку и повторите сохранение.")
+
+if st.session_state.pop("saved", False):
+    st.success("Заявка сохранена. Ответ клиенту отправляет менеджер.")
+st.subheader("История обработанных заявок")
+try:
+    rows, workbook = snapshot(JOURNAL)
+    if rows:
+        st.dataframe(rows[::-1], hide_index=True, width="stretch")
+    else:
+        st.caption("Здесь появятся заявки, которые вы подтвердите.")
+    st.download_button("Скачать Excel", workbook, file_name="processed_leads.xlsx",
+                       mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet")
+except (ValueError, OSError):
+    st.error("Не удалось открыть журнал. Проверьте права доступа, целостность книги и заголовки листа «Заявки».")
+
+with st.expander("Экономический эффект"):
+    st.caption("Иллюстративный сценарий для gpt-5-mini. Один запрос на заявку, 600 входных и 250 выходных токенов, без кэша.")
+    a, b, c = st.columns(3)
+    count = a.number_input("Заявок в неделю", min_value=0, value=120)
+    manual = b.number_input("Ручная обработка, мин", min_value=0.0, value=4.0)
+    review = c.number_input("Проверка и завершение, мин", min_value=0.0, value=1.5)
+    hourly = a.number_input("Стоимость часа, ₽", min_value=0.0, value=500.0)
+    rate = b.number_input("Условный курс, ₽ за доллар", min_value=0.0, value=90.0)
+    weeks = c.number_input("Недель в месяце", min_value=0.0, value=4.33)
+    e = calculate(count, manual, review, weeks, hourly, rate)
+    st.write(f"Время за неделю: {e.before_hours:g} ч вручную → {e.after_hours:g} ч с проверкой. "
+             f"Экономия: {e.saved_hours:g} ч в неделю и {e.monthly_saved_hours:.2f} ч в месяц.")
+    st.write(f"API: {e.request_usd:.5f} USD за заявку · {e.weekly_api_usd:.4f} USD в неделю · {e.monthly_api_usd:.4f} USD в месяц.")
+    st.write(f"Оценка высвобождённого времени за вычетом API: {e.weekly_net_rub:.2f} ₽ в неделю "
+             f"и {e.monthly_net_rub:.2f} ₽ в месяц. Фактическая экономия фонда оплаты труда зависит от загрузки.")