ИИ обрабатывает заявки с сайта: квалификация, автоответ и задача менеджеру за 30 секунд

import asyncio
import smtplib
from email.mime.text import MIMEText
from email.mime.multipart import MIMEMultipart
from fastapi import FastAPI, HTTPException
from pydantic import BaseModel
import anthropic
import json
from notion_client import AsyncClient as NotionClient

app = FastAPI()

CLAUDE_API_KEY   = "YOUR_CLAUDE_KEY"
NOTION_TOKEN     = "YOUR_NOTION_TOKEN"
NOTION_DB_ID     = "YOUR_DATABASE_ID"
TG_BOT_TOKEN     = "YOUR_TG_BOT_TOKEN"
SMTP_HOST        = "smtp.yandex.ru"
SMTP_USER        = "noreply@yourcompany.ru"
SMTP_PASS        = "YOUR_SMTP_PASS"

# Маршрутизация по теме → Telegram chat_id менеджера
MANAGERS = {
  "юридические":   "@legal_manager",
  "бухгалтерия":   "@finance_manager",
  "hr":            "@hr_manager",
  "общее":         "@duty_manager",
}

class LeadRequest(BaseModel):
  name: str
  phone: str
  email: str
  message: str
  utm_source: str = ""
  utm_campaign: str = ""

@app.post("/webhook/lead")
async def handle_lead(lead: LeadRequest):
  # 1. Отправляем в Claude — получаем структурированный JSON
  analysis = await analyze_lead(lead)

  if analysis.get("is_spam"):
      return {"status": "filtered"}

  # 2. Три параллельных действия
  await asyncio.gather(
      send_email_reply(lead.email, lead.name, analysis["email_reply"]),
      create_notion_task(lead, analysis),
      notify_manager(lead, analysis),
  )
  return {"status": "processed", "classification": analysis["classification"]}


async def analyze_lead(lead: LeadRequest) -> dict:
  client = anthropic.Anthropic(api_key=CLAUDE_API_KEY)

  prompt = f"""Ты — ассистент по обработке входящих заявок. Проанализируй заявку и верни ТОЛЬКО валидный JSON.

Заявка:
- Имя: {lead.name}
- Телефон: {lead.phone}
- Email: {lead.email}
- Сообщение: {lead.message}
- UTM-источник: {lead.utm_source}
- UTM-кампания: {lead.utm_campaign}

Верни JSON со следующими полями:
{{
"classification": "hot|warm|cold",
"priority": "high|medium|low",
"topic": "юридические|бухгалтерия|hr|общее",
"is_spam": false,
"email_reply": "персональный текст ответа клиенту на русском, 3-5 предложений. Обратись по имени. Упомяни конкретику из их сообщения. Объясни что произойдёт дальше. Без общих фраз.",
"manager_note": "краткая пометка для менеджера: почему горячий/тёплый/холодный, на что обратить внимание"
}}

Критерии классификации:
- hot: есть конкретика (сроки, бюджет, детали задачи), срочность, готовность обсуждать
- warm: интерес есть, деталей мало, надо уточнять
- cold: очень общий запрос, нет деталей, или просто смотрит варианты
- is_spam: true если это явная рассылка, тест формы, мусор, или запрос не по теме бизнеса"""

  message = client.messages.create(
      model="claude-sonnet-4-6",
      max_tokens=800,
      messages=[{"role": "user", "content": prompt}]
  )
  text = message.content[0].text.strip()
  # Claude иногда обворачивает в markdown — вычищаем
  if text.startswith("```"):
      text = text.split("\n", 1)[1].rsplit("```", 1)[0]
  return json.loads(text)


async def send_email_reply(to_email: str, name: str, reply_text: str):
  msg = MIMEMultipart("alternative")
  msg["Subject"] = f"Ваша заявка получена — ответим скоро"
  msg["From"]    = SMTP_USER
  msg["To"]      = to_email

  html = f"""<div style="font-family:sans-serif;max-width:600px;color:#222">
    <p>Здравствуйте, {name}!</p>
    <p>{reply_text.replace(chr(10), '<br>')}</p>
    <p style="color:#888;font-size:12px;margin-top:24px">
      Это письмо сформировано автоматически. Если что-то срочно — звоните напрямую.
    </p>
  </div>"""
  msg.attach(MIMEText(html, "html", "utf-8"))

  with smtplib.SMTP_SSL(SMTP_HOST, 465) as server:
      server.login(SMTP_USER, SMTP_PASS)
      server.sendmail(SMTP_USER, to_email, msg.as_string())


async def create_notion_task(lead: LeadRequest, analysis: dict):
  notion = NotionClient(auth=NOTION_TOKEN)
  priority_map = {"high": "Высокий", "medium": "Средний", "low": "Низкий"}

  # Важно: заголовок X-Notion-Version обязателен — без него 400 ошибка
  await notion.pages.create(
      parent={"database_id": NOTION_DB_ID},
      properties={
          "Name":           {"title": [{"text": {"content": f"{lead.name} — {analysis['topic']}"}}]},
          "Статус":         {"select": {"name": "Новая"}},
          "Приоритет":      {"select": {"name": priority_map.get(analysis["priority"], "Средний")}},
          "Телефон":        {"phone_number": lead.phone},
          "Email":          {"email": lead.email},
          "Классификация":  {"select": {"name": analysis["classification"]}},
          "Источник":       {"rich_text": [{"text": {"content": lead.utm_source or "—"}}]},
          "Заметка":        {"rich_text": [{"text": {"content": analysis.get("manager_note", "")}}]},
      }
  )


async def notify_manager(lead: LeadRequest, analysis: dict):
  import httpx
  manager = MANAGERS.get(analysis.get("topic", "общее"), MANAGERS["общее"])
  cls_emoji = {"hot": "🔥", "warm": "🟡", "cold": "🔵"}.get(analysis["classification"], "⚪")

  text = (
      f"{cls_emoji} Новая заявка [{analysis['classification'].upper()}]\n\n"
      f"👤 {lead.name}\n"
      f"📞 {lead.phone}\n"
      f"✉️ {lead.email}\n\n"
      f"💬 {lead.message[:300]}\n\n"
      f"📌 {analysis.get('manager_note', '')}\n"
      f"🏷 UTM: {lead.utm_source or '—'}"
  )
  async with httpx.AsyncClient() as client:
      await client.post(
          f"https://api.telegram.org/bot{TG_BOT_TOKEN}/sendMessage",
          json={"chat_id": manager, "text": text, "parse_mode": "HTML"}
      )