Delete dialogue.py

dialogue.py

@@ -1,237 +0,0 @@
-"""
-PlotWeaver Voice Agent — Dialogue Manager
-==========================================
-FSM for multi-turn Hausa conversations across 3 verticals.
-State lives in Gradio session state (dict) — no Redis needed in the Space.
-"""
-from __future__ import annotations
-from dataclasses import dataclass, field, asdict
-from enum import Enum
-from typing import Optional
-
-
-class Vertical(str, Enum):
-    BANK = "bank"
-    TELECOM = "telecom"
-    ECOMMERCE = "ecommerce"
-
-
-@dataclass
-class DialogueState:
-    session_id: str
-    vertical: str
-    current_state: str = "greeting"
-    slots: dict = field(default_factory=dict)
-    turn_count: int = 0
-    escalate_to_human: bool = False
-    history: list = field(default_factory=list)
-
-    def to_dict(self):
-        return asdict(self)
-
-    @classmethod
-    def from_dict(cls, d):
-        return cls(**d) if d else None
-
-
-SCENARIOS = {
-    "bank": {
-        "name": "PlotWeaver Bank",
-        "states": {
-            "greeting": {
-                "ha": "Sannu! Wannan shine mataimakin banki na PlotWeaver. Yaya zan taimake ka yau? Za ka iya ce 'duba ma'auni', 'toshe kati', ko 'canjin kuɗi'.",
-                "en": "Hello! This is the PlotWeaver banking assistant. How can I help you today? You can say 'check balance', 'block card', or 'transfer money'.",
-                "expects": "intent",
-                "transitions": {"check_balance": "ask_account_number", "block_card": "confirm_block_card", "transfer_money": "ask_recipient"},
-            },
-            "ask_account_number": {
-                "ha": "Don Allah ka faɗi lambobin ƙarshe huɗu na asusunka.",
-                "en": "Please say the last four digits of your account number.",
-                "expects": "digits",
-                "transitions": {"provide_digits": "return_balance"},
-            },
-            "return_balance": {
-                "ha": "Ma'aunin asusunka shine Naira dubu ɗari biyu da arba'in da biyar. Akwai wani abu?",
-                "en": "Your account balance is two hundred forty-five thousand Naira. Anything else?",
-                "expects": "yesno",
-                "transitions": {"yes": "greeting", "no": "exit"},
-            },
-            "confirm_block_card": {
-                "ha": "Don tabbatar, kana son toshe katinka? Ka ce 'i' ko 'a'a'.",
-                "en": "To confirm, you want to block your card? Say 'yes' or 'no'.",
-                "expects": "yesno",
-                "transitions": {"yes": "card_blocked", "no": "greeting"},
-            },
-            "card_blocked": {
-                "ha": "An toshe katinka. Sabon kati zai iso a cikin kwanaki uku zuwa biyar. Ana juya ka ga wakili don tabbatar.",
-                "en": "Your card is blocked. A new card will arrive in 3-5 days. Transferring you to an agent for confirmation.",
-                "expects": None, "terminal": True, "escalate": True,
-            },
-            "ask_recipient": {
-                "ha": "Zuwa wa kake son turawa? Ka faɗi sunan mai karɓa.",
-                "en": "Who do you want to transfer to? Say the recipient's name.",
-                "expects": "name",
-                "transitions": {"provide_name": "ask_amount"},
-            },
-            "ask_amount": {
-                "ha": "Nawa kake son turawa, a Naira?",
-                "en": "How much do you want to transfer, in Naira?",
-                "expects": "amount",
-                "transitions": {"provide_amount": "confirm_transfer"},
-            },
-            "confirm_transfer": {
-                "ha": "Zan tura kuɗin yanzu. Ka ce 'i' don ci gaba.",
-                "en": "I'll send the money now. Say 'yes' to continue.",
-                "expects": "yesno",
-                "transitions": {"yes": "transfer_done", "no": "greeting"},
-            },
-            "transfer_done": {
-                "ha": "An tura kuɗin. Godiya da zabar PlotWeaver Bank.",
-                "en": "Money sent. Thank you for choosing PlotWeaver Bank.",
-                "expects": None, "terminal": True,
-            },
-        },
-    },
-    "telecom": {
-        "name": "PlotWeaver Telecom",
-        "states": {
-            "greeting": {
-                "ha": "Sannu! Wannan shine PlotWeaver Telecom. Kana son 'saya airtime', 'saya bundle', ko 'yin korafi'?",
-                "en": "Hello! This is PlotWeaver Telecom. Would you like to 'buy airtime', 'buy bundle', or 'file a complaint'?",
-                "expects": "intent",
-                "transitions": {"buy_airtime": "ask_airtime_amount", "buy_bundle": "ask_bundle_type", "complaint": "ask_complaint"},
-            },
-            "ask_airtime_amount": {
-                "ha": "Nawa na airtime kake son saya? Misali, Naira ɗari ko dubu.",
-                "en": "How much airtime? For example 100 or 1000 Naira.",
-                "expects": "amount",
-                "transitions": {"provide_amount": "airtime_done"},
-            },
-            "airtime_done": {
-                "ha": "An kara airtime. Ma'aunin ka sabo shine Naira dubu ɗaya da ɗari biyar.",
-                "en": "Airtime loaded. Your new balance is 1500 Naira.",
-                "expects": None, "terminal": True,
-            },
-            "ask_bundle_type": {
-                "ha": "Wane irin bundle? Muna da 'rana', 'mako', ko 'wata'.",
-                "en": "Which bundle type? 'day', 'week', or 'month'.",
-                "expects": "bundle",
-                "transitions": {"provide_bundle": "bundle_done"},
-            },
-            "bundle_done": {
-                "ha": "An kunna bundle ɗinka. Za ka iya yin amfani da shi yanzu.",
-                "en": "Your bundle is active. You can use it now.",
-                "expects": None, "terminal": True,
-            },
-            "ask_complaint": {
-                "ha": "Me ya faru? Ka bayyana matsalar da kake fuskanta.",
-                "en": "What happened? Please describe the issue.",
-                "expects": "text",
-                "transitions": {"provide_text": "escalate"},
-            },
-            "escalate": {
-                "ha": "Nagode. Zan juya ka ga wakili na mutum yanzu.",
-                "en": "Thank you. I'll transfer you to a human agent now.",
-                "expects": None, "terminal": True, "escalate": True,
-            },
-        },
-    },
-    "ecommerce": {
-        "name": "PlotWeaver Delivery",
-        "states": {
-            "greeting": {
-                "ha": "Sannu! Wannan shine PlotWeaver Delivery. Kana son 'bincika oda', 'sake tsara lokaci', ko 'mayar da kaya'?",
-                "en": "Hello! This is PlotWeaver Delivery. Would you like to 'check order', 'reschedule', or 'return'?",
-                "expects": "intent",
-                "transitions": {"check_order": "ask_order_id", "reschedule": "ask_order_id_reschedule", "return_item": "ask_order_id_return"},
-            },
-            "ask_order_id": {
-                "ha": "Ka faɗi lambar oda naka.",
-                "en": "Say your order number.",
-                "expects": "digits",
-                "transitions": {"provide_digits": "order_status"},
-            },
-            "order_status": {
-                "ha": "Oda ɗinka yana kan hanya. Za a isar gobe da yamma.",
-                "en": "Your order is on the way. It will be delivered tomorrow evening.",
-                "expects": None, "terminal": True,
-            },
-            "ask_order_id_reschedule": {
-                "ha": "Ka faɗi lambar oda da kake son sake tsarawa.",
-                "en": "Say the order number you want to reschedule.",
-                "expects": "digits",
-                "transitions": {"provide_digits": "ask_new_date"},
-            },
-            "ask_new_date": {
-                "ha": "Wace rana kake so? Misali 'jumma'a' ko 'asabar'.",
-                "en": "Which day? For example 'Friday' or 'Saturday'.",
-                "expects": "date",
-                "transitions": {"provide_date": "reschedule_done"},
-            },
-            "reschedule_done": {
-                "ha": "An sake tsara isar. Za ka sami SMS na tabbatarwa.",
-                "en": "Delivery rescheduled. You'll receive a confirmation SMS.",
-                "expects": None, "terminal": True,
-            },
-            "ask_order_id_return": {
-                "ha": "Ka faɗi lambar oda da kake son mayarwa.",
-                "en": "Say the order number you want to return.",
-                "expects": "digits",
-                "transitions": {"provide_digits": "return_reason"},
-            },
-            "return_reason": {
-                "ha": "Me ya sa kake son mayarwa?",
-                "en": "Why do you want to return it?",
-                "expects": "text",
-                "transitions": {"provide_reason": "return_done"},
-            },
-            "return_done": {
-                "ha": "An karɓi buƙatarka. Wakili zai tattara kaya a gobe.",
-                "en": "Your request is received. An agent will collect the item tomorrow.",
-                "expects": None, "terminal": True,
-            },
-        },
-    },
-}
-
-
-def get_prompt(vertical: str, state_name: str) -> dict:
-    if state_name == "escalate_virtual":
-        return {"ha": "Zan juya ka ga wakili na mutum yanzu. Ka jira ɗan lokaci.",
-                "en": "I'll transfer you to a human agent now. Please hold."}
-    if state_name == "exit":
-        return {"ha": "Nagode. Sai watan.", "en": "Thank you. Goodbye."}
-    s = SCENARIOS[vertical]["states"].get(state_name)
-    if not s:
-        return {"ha": "Ban fahimci abin da ka ce ba.", "en": "I didn't understand."}
-    return {"ha": s["ha"], "en": s["en"]}
-
-
-def get_expected_slot(vertical: str, state_name: str) -> Optional[str]:
-    s = SCENARIOS[vertical]["states"].get(state_name)
-    return s.get("expects") if s else None
-
-
-def transition(state: DialogueState, intent: str, entities: dict) -> DialogueState:
-    state.turn_count += 1
-    for k, v in entities.items():
-        state.slots[k] = v
-
-    if intent == "human_agent" or state.turn_count > 12:
-        state.current_state = "escalate_virtual"
-        state.escalate_to_human = True
-        return state
-
-    current = SCENARIOS[state.vertical]["states"].get(state.current_state)
-    if not current:
-        state.current_state = "greeting"
-        return state
-
-    next_state = current.get("transitions", {}).get(intent)
-    if next_state:
-        state.current_state = next_state
-        target = SCENARIOS[state.vertical]["states"].get(next_state, {})
-        if target.get("escalate"):
-            state.escalate_to_human = True
-
-    return state