queue_worker.py

import logging
import os
from datetime import datetime, timezone
from typing import Any

import openai

import agent
import copywriter
import db
import whatsapp
from models import ConversationMessage, IncomingMessage, Intent, LeadData, Tenant


logger = logging.getLogger("propbot")

LEAD_NOTIFICATION_THRESHOLD = 75

AGENT_PLACEHOLDER_REPLIES: dict[Intent, str] = {
    Intent.CMA_REQUEST: "קיבלתי — ניתוח שוק (CMA) יהיה זמין בגרסה הבאה.",
    Intent.PROPERTY_UPDATE: "קיבלתי — עדכון מלאי נכסים יהיה זמין בגרסה הבאה.",
    Intent.STATS_REQUEST: "קיבלתי — דוחות וסטטיסטיקות יהיו זמינים בגרסה הבאה.",
    Intent.LEAD_QUALIFICATION: (
        "קיבלתי — ההודעה הזו נראית כמו שיחה עם לקוח, "
        "לא ממך. אם זו טעות תכתוב לי שוב."
    ),
}


def process_message(tenant_id: str, role: str, incoming_dict: dict[str, Any]) -> None:
    try:
        incoming = IncomingMessage(**incoming_dict)
        tenant = db.get_tenant_by_id(tenant_id)
        if tenant is None:
            logger.warning("tenant לא נמצא: %s — מדלג", tenant_id)
            return

        if incoming.audio_id is not None:
            try:
                audio_bytes = whatsapp.download_media(incoming.audio_id)
                client = openai.OpenAI(api_key=os.environ["OPENAI_API_KEY"])
                response = client.audio.transcriptions.create(
                    model="whisper-1",
                    language="he",
                    file=("voice.ogg", audio_bytes, "audio/ogg"),
                )
                transcript = response.text
                logger.info("🎤 תמלול קולי: %s", transcript)
                incoming.body = f"[הודעה קולית] {transcript}"
            except Exception:
                logger.warning(
                    "תמלול הודעה קולית נכשל (audio_id=%s)",
                    incoming.audio_id,
                    exc_info=True,
                )
                incoming.body = (
                    "[הודעה קולית — לא הצלחתי לתמלל. "
                    "אנא שלח כהודעת טקסט]"
                )

        intent, intent_usage = agent.classify_intent(incoming.body)
        db.log_usage(tenant_id, intent_usage, intent.value)
        logger.info(
            "📩 מעבד הודעה (tenant=%s, role=%s, intent=%s)",
            tenant_id,
            role,
            intent.value,
        )

        if role == "customer":
            _handle_customer(tenant, incoming, intent)
        elif role == "agent":
            _handle_agent(tenant, incoming, intent)
        else:
            logger.warning("תפקיד לא ידוע: %s", role)
    except Exception:
        logger.exception(
            "שגיאה בעיבוד הודעה (tenant=%s, role=%s)", tenant_id, role
        )
        _notify_agent_of_error(tenant_id)


def _handle_customer(
    tenant: Tenant, incoming: IncomingMessage, intent: Intent
) -> None:
    tenant_id = str(tenant.id)
    conversation = db.get_or_create_conversation(tenant_id, incoming.from_phone)
    properties_summary = db.get_active_properties_summary(tenant_id)

    existing_lead = conversation.lead_data
    if agent.is_bant_complete(existing_lead) and not existing_lead.meeting_proposed:
        bant_hint = (
            "הלקוח השלים את כל פרטי החיפוש. שלב בסוף התשובה הצעה חמה וטבעית "
            "לקבוע פגישה או ביקור בנכס — קצר, לא לחוץ."
        )
    else:
        bant_hint = agent.get_bant_gap_question(existing_lead)

    reply, resp_usage = agent.get_response(
        tenant=tenant,
        history=conversation.messages,
        new_message=incoming.body,
        properties_summary=properties_summary,
        bant_gap_hint=bant_hint,
    )
    db.log_usage(tenant_id, resp_usage, intent.value)

    whatsapp.send_text_message(incoming.from_phone, reply)

    now = datetime.now(timezone.utc)
    updated_messages = [
        *conversation.messages,
        ConversationMessage(
            role="user", content=incoming.body, timestamp=incoming.timestamp
        ),
        ConversationMessage(role="assistant", content=reply, timestamp=now),
    ]

    lead_data, lead_usage = agent.extract_lead_data(updated_messages)
    db.log_usage(tenant_id, lead_usage, intent.value)

    # extract_lead_data לא מחלץ meeting_proposed (state פנימי) —
    # שומרים את הדגל כדי לא להציע פגישה שוב בטורן הבא.
    if existing_lead.meeting_proposed:
        lead_data.meeting_proposed = True
    elif agent.is_bant_complete(existing_lead):
        # BANT היה שלם בכניסה לטורן → bant_hint כלל את הצעת הפגישה → כבר הוצע.
        lead_data.meeting_proposed = True

    score = agent.compute_lead_score(lead_data)
    notified = conversation.notified_agent
    if score >= LEAD_NOTIFICATION_THRESHOLD and not notified:
        _notify_agent_of_lead(tenant, incoming.from_phone, lead_data, score)
        from crm import send_crm_webhook
        send_crm_webhook(tenant, lead_data, score, customer_phone=incoming.from_phone)
        notified = True

    db.save_conversation(
        conversation_id=conversation.id,
        messages=updated_messages,
        lead_data=lead_data,
        lead_score=score,
        notified_agent=notified,
    )
    logger.info(
        "✅ תגובה ללקוח נשלחה (tenant=%s, customer=%s, score=%d, notified=%s)",
        tenant_id,
        incoming.from_phone,
        score,
        notified,
    )


def _handle_agent(
    tenant: Tenant, incoming: IncomingMessage, intent: Intent
) -> None:
    tenant_id = str(tenant.id)
    if intent == Intent.AD_REQUEST:
        try:
            ad = copywriter.generate_ad(incoming.body, tenant_id)
            reply = (
                "✍️ *מודעת פייסבוק:*\n"
                f"{ad['facebook']}\n\n"
                "🏠 *מודעת יד2:*\n"
                f"{ad['yad2']}"
            )
        except Exception:
            logger.exception("generate_ad נכשל (tenant=%s)", tenant_id)
            reply = (
                "מצטער, לא הצלחתי לנסח את המודעה. "
                "נסה שוב או פנה לתמיכה."
            )
    elif intent == Intent.GENERAL_QUESTION:
        properties_summary = db.get_active_properties_summary(tenant_id)
        reply, usage = agent.get_response(
            tenant=tenant,
            history=[],
            new_message=incoming.body,
            properties_summary=properties_summary,
        )
        db.log_usage(tenant_id, usage, intent.value)
    else:
        reply = AGENT_PLACEHOLDER_REPLIES.get(
            intent, "קיבלתי — זה יגיע בגרסה הבאה."
        )
    whatsapp.send_text_message(incoming.from_phone, reply)
    logger.info(
        "✅ תגובה למתווך נשלחה (tenant=%s, intent=%s)", tenant_id, intent.value
    )


def _notify_agent_of_lead(
    tenant: Tenant, customer_phone: str, lead: LeadData, score: int
) -> None:
    lines = [
        f"🔥 ליד חם מ-{customer_phone}",
        f"ציון: {score}/100",
        "",
    ]
    if lead.customer_name:
        lines.append(f"שם: {lead.customer_name}")
    if lead.budget:
        lines.append(f"תקציב: {lead.budget}")
    if lead.area_or_asset:
        lines.append(f"אזור/נכס: {lead.area_or_asset}")
    if lead.rooms:
        lines.append(f"חדרים: {lead.rooms}")
    if lead.condition_preference:
        pref_map = {"renovated": "משופצת", "needs_work": "מוכן לשיפוץ", "both": "לא אכפת"}
        lines.append(f"מצב דירה: {pref_map.get(lead.condition_preference, lead.condition_preference)}")
    if lead.meeting_ready:
        lines.append("מוכן לפגישה: כן")
    if lead.gender:
        gender_map = {"male": "זכר", "female": "נקבה", "couple": "זוג"}
        lines.append(f"מגדר: {gender_map.get(lead.gender, lead.gender)}")

    try:
        whatsapp.send_text_message(tenant.phone, "\n".join(lines))
    except Exception:
        logger.exception(
            "שליחת התראת ליד למתווך נכשלה (tenant=%s)", tenant.id
        )


def _notify_agent_of_error(tenant_id: str) -> None:
    try:
        tenant = db.get_tenant_by_id(tenant_id)
        if tenant is None:
            return
        whatsapp.send_text_message(
            tenant.phone, "⚠️ שגיאה בטיפול בהודעה האחרונה. צוות התמיכה יבדוק."
        )
    except Exception:
        logger.exception("גם הודעת השגיאה למתווך נכשלה (tenant=%s)", tenant_id)