messages.py

# -*- coding: utf-8 -*-
"""
Генерация персональных сообщений для найденных компаний.
Запуск: python messages.py
Опционально через LM Studio: python messages.py --lm
"""
import argparse, json, textwrap

COMPANIES = [
    {
        "id": 1,
        "city": "Пенза",
        "name": "Барбершоп",
        "rating": 4.5,
        "phone": "+7 (902) 205-15-15",
        "whatsapp": "",
        "telegram": "",
        "url": "https://yandex.com/maps/org/barbershop/131199036165",
        "pain_quote": "да я попал (я же не на личный телефон звоню)…",
        "pain_type": "no_answer",  # не берут трубку
    },
    {
        "id": 2,
        "city": "Пенза",
        "name": "5th Avenue",
        "rating": 4.5,
        "phone": "+7 (8412) 59-44-99",
        "whatsapp": "",
        "telegram": "",
        "url": "https://yandex.com/maps/org/5th_avenue/34308132117",
        "pain_quote": "Записался на 19:30, пришел ко времени, администратор сказала подожда...",
        "pain_type": "wait_after_booking",  # записали, заставили ждать
    },
    {
        "id": 3,
        "city": "Пенза",
        "name": "Barber Blade",
        "rating": 4.3,
        "phone": "+7 (8412) 77-17-77",
        "whatsapp": "https://wa.me/79530263600",
        "telegram": "",
        "url": "https://yandex.com/maps/org/barber_blade/60939703730",
        "pain_quote": "Пришел без записи — мастера нет, позвонил по номеру, попросили зайти через час.",
        "pain_type": "no_booking_system",  # нет системы записи
    },
    {
        "id": 4,
        "city": "Смоленск",
        "name": "Чио Чио",
        "rating": 4.3,
        "phone": "+7 (951) 700-01-61",
        "whatsapp": "",
        "telegram": "",
        "url": "https://yandex.com/maps/org/chio_chio/126489996135",
        "pain_quote": "Записался на 9 утра, пришел — закрыто, хоть бы позвонили.",
        "pain_type": "forgot_booking",  # забыли про запись
    },
    {
        "id": 5,
        "city": "Смоленск",
        "name": "Чио Чио",
        "rating": 4.0,
        "phone": "+7 (951) 700-01-61",
        "whatsapp": "",
        "telegram": "",
        "url": "https://yandex.com/maps/org/chio_chio/10085805396",
        "pain_quote": "Пришла по заблаговременной записи — оказалось, что...",
        "pain_type": "booking_chaos",  # хаос с записями
    },
]

# Шаблоны по типу боли
TEMPLATES = {
    "no_answer": (
        "Здравствуйте! Видел ваш барбершоп {name} на Яндекс.Картах — "
        "в отзывах клиенты пишут, что не могут дозвониться. "
        "Из-за этого вы теряете записи. "
        "Есть решение: Telegram Mini App, через который клиент записывается сам в любое время — "
        "без звонков и пропущенных. Можно показать как это работает?"
    ),
    "wait_after_booking": (
        "Здравствуйте! Смотрел отзывы о {name} — клиенты жалуются, "
        "что приходят по записи и ждут. Это бьёт по репутации. "
        "С Telegram Mini App запись автоматически привязывается к слоту мастера — "
        "клиент видит точное время и путаницы нет. "
        "Хотите покажу как выглядит?"
    ),
    "no_booking_system": (
        "Здравствуйте! Нашёл {name} на картах. "
        "Вижу, что записи нет — клиенты приходят вживую или звонят. "
        "Это теряет людей, которые хотят записаться прямо сейчас, но не готовы звонить. "
        "Telegram Mini App для онлайн-записи — клиент выбирает мастера и время прямо в Telegram. "
        "Хотите посмотрите как это работает на примере?"
    ),
    "forgot_booking": (
        "Здравствуйте! Видел отзывы {name} — один клиент пришёл по записи, "
        "а салон был закрыт. Такие ситуации сразу идут в минус рейтингу. "
        "Telegram Mini App автоматически напоминает клиенту о записи за день и за час — "
        "отмены и незапланированные пропуски уходят. "
        "Интересно было бы попробовать?"
    ),
    "booking_chaos": (
        "Здравствуйте! Изучил отзывы {name} — вижу, что с записями бывают накладки. "
        "Когда запись ведётся через звонки или директ, это сложно контролировать. "
        "Telegram Mini App — это расписание мастеров + онлайн-запись клиентов в одном месте, "
        "всё автоматически. Можем показать как это выглядит?"
    ),
}

CONTACT_NOTE = {
    "whatsapp": "Написать в WhatsApp: {wa}",
    "telegram": "Написать в Telegram: {tg}",
    "phone": "Написать в WhatsApp на номер: {phone} (или позвонить)",
}


def make_message(c: dict) -> str:
    tmpl = TEMPLATES[c["pain_type"]]
    msg = tmpl.format(name=c["name"])

    # Add contact call-to-action
    if c["whatsapp"]:
        contact = f"(ответьте здесь или напишите в WA: {c['whatsapp']})"
    elif c["telegram"]:
        contact = f"(ответьте здесь или в TG: {c['telegram']})"
    else:
        contact = f"(ответьте на это сообщение)"

    return msg + "\n" + contact


def make_message_via_lm(c: dict, host: str, model: str) -> str:
    import httpx, json as _json
    prompt = (
        f"Ты продавец Telegram Mini App для онлайн-записи в барбершопы и салоны красоты.\n"
        f"Напиши короткое холодное сообщение (3-4 предложения) владельцу барбершопа '{c['name']}' "
        f"в городе {c['city']}.\n"
        f"Рейтинг заведения: {c['rating']}.\n"
        f"Реальная жалоба клиента из отзыва: «{c['pain_quote']}»\n"
        f"Упомяни эту конкретную проблему. Предложи решение через Telegram Mini App для онлайн-записи. "
        f"Не используй слово 'бот'. Обращайся на 'вы'. Без приветствия вроде 'Добрый день'.\n"
        f"Только текст сообщения, без кавычек и пояснений."
    )
    try:
        r = httpx.post(
            f"{host.rstrip('/')}/v1/chat/completions",
            json={
                "model": model,
                "messages": [{"role": "user", "content": prompt}],
                "max_tokens": 300,
                "temperature": 0.7,
            },
            timeout=60,
        )
        r.raise_for_status()
        return r.json()["choices"][0]["message"]["content"].strip()
    except Exception as e:
        print(f"  [LM Studio недоступна: {e}] — использую шаблон")
        return make_message(c)


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("--lm", action="store_true", help="Генерировать через LM Studio")
    parser.add_argument("--host", default="http://localhost:1234", help="LM Studio URL")
    parser.add_argument("--model", default="local-model", help="Модель")
    parser.add_argument("--json", action="store_true", help="Вывод в JSON")
    args = parser.parse_args()

    results = []

    for c in COMPANIES:
        print(f"\n{'='*60}")
        print(f"#{c['id']} {c['name']} | {c['city']} | ★{c['rating']}")
        print(f"Тел: {c['phone']}")
        if c["whatsapp"]:
            print(f"WA:  {c['whatsapp']}")
        if c["telegram"]:
            print(f"TG:  {c['telegram']}")
        print(f"URL: {c['url']}")
        print(f"Жалоба: «{c['pain_quote']}»")
        print()

        if args.lm:
            print("Генерирую через LM Studio...")
            msg = make_message_via_lm(c, args.host, args.model)
        else:
            msg = make_message(c)

        print("СООБЩЕНИЕ:")
        print(textwrap.fill(msg, width=70))

        results.append({**c, "message": msg})

    if args.json:
        with open("messages_out.json", "w", encoding="utf-8") as f:
            json.dump(results, f, ensure_ascii=False, indent=2)
        print(f"\nСохранено в messages_out.json")


if __name__ == "__main__":
    main()