app/main.py

"""
main.py
-------
Primary entry point for Notiflow.

Exposes run_notiflow(message) for programmatic use (dashboard, API, tests)
and supports CLI testing directly from the terminal.

Public API
----------
run_notiflow(message: str) -> dict

    Runs the full pipeline and returns:
    {
        "message": str,
        "intent":  str,
        "data":    dict,
        "event":   dict,
    }

CLI usage
---------
    python app/main.py "rahul ne 15000 bheja"

    Prints the result as formatted JSON to stdout.

Demo mode
---------
    Controlled by DEMO_MODE in app/config.py or the environment variable
    NOTIFLOW_DEMO_MODE=false (set to disable demo mode).

    When DEMO_MODE is True, a local simulation is used so the app works
    without AWS credentials. The dashboard's DEMO_MODE toggle maps to this.
"""

from __future__ import annotations

import json
import logging
import sys
from typing import Any

from app.config import DEMO_MODE

logger = logging.getLogger(__name__)

# ---------------------------------------------------------------------------
# Demo pipeline (no AWS needed)
# ---------------------------------------------------------------------------

_DEMO_RESPONSES: dict[str, dict] = {
    "rahul ne 15000 bheja": {
        "intent": "payment",
        "data":   {"customer": "Rahul", "amount": 15000, "payment_type": None},
        "event":  {"event": "payment_recorded",
                   "payment": {"customer": "Rahul", "amount": 15000,
                               "payment_type": None, "status": "received"}},
    },
    "bhaiya 3 kurti bhej dena": {
        "intent": "order",
        "data":   {"customer": None, "item": "kurti", "quantity": 3},
        "event":  {"event": "order_received",
                   "order":  {"customer": None, "item": "kurti",
                              "quantity": 3, "status": "pending"},
                   "invoice": {"invoice_id": "INV-DEMO-0001", "total_amount": 0.0}},
    },
    "priya ke liye 2 kilo aata bhej dena": {
        "intent": "order",
        "data":   {"customer": "Priya", "item": "aata", "quantity": 2},
        "event":  {"event": "order_received",
                   "order":  {"customer": "Priya", "item": "aata",
                              "quantity": 2, "status": "pending"},
                   "invoice": {"invoice_id": "INV-DEMO-0002", "total_amount": 0.0}},
    },
    "size chota hai exchange karna hai": {
        "intent": "return",
        "data":   {"customer": None, "item": None, "reason": "size issue"},
        "event":  {"event": "return_requested",
                   "return": {"customer": None, "item": None,
                              "reason": "size issue", "status": "pending_review"}},
    },
    "udhar me de dijiye": {
        "intent": "credit",
        "data":   {"customer": None, "item": None, "quantity": None, "amount": None},
        "event":  {"event": "credit_recorded",
                   "credit": {"customer": None, "amount": None, "status": "open"}},
    },
    "suresh ko 500 ka maal udhar dena": {
        "intent": "credit",
        "data":   {"customer": "Suresh", "item": "goods", "quantity": None, "amount": 500},
        "event":  {"event": "credit_recorded",
                   "credit": {"customer": "Suresh", "amount": 500, "status": "open"}},
    },
    "3 kurti ka set ready rakhna": {
        "intent": "preparation",
        "data":   {"item": "kurti", "quantity": 3},
        "event":  {"event": "preparation_queued",
                   "preparation": {"item": "kurti", "quantity": 3, "status": "queued"}},
    },
    "amit bhai ka 8000 gpay se aaya": {
        "intent": "payment",
        "data":   {"customer": "Amit", "amount": 8000, "payment_type": "upi"},
        "event":  {"event": "payment_recorded",
                   "payment": {"customer": "Amit", "amount": 8000,
                               "payment_type": "upi", "status": "received"}},
    },
}


def _fallback_intent(message: str) -> str:
    m = message.lower()
    if any(w in m for w in ["bheja", "aaya", "cash", "gpay", "upi", "paytm", "online"]):
        return "payment"
    if any(w in m for w in ["exchange", "wapas", "return", "vapas", "size"]):
        return "return"
    if any(w in m for w in ["udhar", "credit", "baad"]):
        return "credit"
    if any(w in m for w in ["ready", "pack", "rakhna", "taiyar"]):
        return "preparation"
    if any(w in m for w in ["bhej", "dena", "chahiye", "kilo", "piece"]):
        return "order"
    return "other"


def _run_demo(message: str) -> dict[str, Any]:
    key      = message.strip().lower()
    response = _DEMO_RESPONSES.get(key)
    if response is None:
        intent   = _fallback_intent(message)
        response = {
            "intent": intent,
            "data":   {"note": f"Demo: classified as '{intent}'"},
            "event":  {"event": f"{intent}_recorded",
                       "note":  "Demo fallback — no exact match"},
        }
    return {
        "message": message,
        "intent":  response["intent"],
        "data":    response["data"],
        "event":   response["event"],
    }


# ---------------------------------------------------------------------------
# Public API
# ---------------------------------------------------------------------------

def run_notiflow(message: str, demo_mode: bool | None = None) -> dict[str, Any]:
    """
    Run a business message through the full Notiflow pipeline.

    This is the single function the dashboard and any external caller
    should use. It never calls agents, skills, or Excel directly.

    Args:
        message:   Raw Hinglish or English business message.
        demo_mode: Override DEMO_MODE from config. If None, uses config value.

    Returns:
        {
            "message": str,
            "intent":  str,
            "data":    dict,
            "event":   dict,
        }

    Raises:
        ValueError:   Empty message.
        RuntimeError: Pipeline failure (live mode only).
    """
    if not message or not message.strip():
        raise ValueError("Message cannot be empty.")

    use_demo = DEMO_MODE if demo_mode is None else demo_mode

    if use_demo:
        logger.info("run_notiflow [demo] ← %r", message)
        return _run_demo(message.strip())
    else:
        logger.info("run_notiflow [live] ← %r", message)
        from agent.orchestrator import process_message
        return process_message(message.strip())


# ---------------------------------------------------------------------------
# CLI entry point
# ---------------------------------------------------------------------------

if __name__ == "__main__":
    logging.basicConfig(level=logging.WARNING)

    if len(sys.argv) < 2:
        print("Usage: python app/main.py \"<business message>\"")
        print('Example: python app/main.py "rahul ne 15000 bheja"')
        sys.exit(1)

    input_message = " ".join(sys.argv[1:])

    try:
        result = run_notiflow(input_message)
        print(json.dumps(result, indent=2, ensure_ascii=False))
    except Exception as exc:
        print(json.dumps({"error": str(exc)}, indent=2))
        sys.exit(1)