Graceless Call Ending

class ClosingManager:
    """Manage graceful call closing"""
    
    CLOSING_TEMPLATES = {
        "qualified": [
            "Cool, all noted! Take care, bye!",
            "Great, got it! Best of luck, bye!",
            "Nice, that's noted. Bye!",
        ],
        "not_interested": [
            "All good, totally understand. All the best!",
            "No worries! Take care, bye!",
        ],
        "callback": [
            "Got it, {time} callback. Talk then, bye!",
            "Noted—{time} works. Bye!",
        ],
        "dnc": [
            "Understood, won't reach out again. Bye!",
            "Got it, removed. Take care!",
        ],
        "wrong_number": [
            "Oh sorry, my bad! Bye!",
        ]
    }
    
    def __init__(self):
        self.goodbye_said = False
        self.closing_started = False
    
    def get_closing(self, outcome: str, 
                    context: dict = None) -> str:
        """Get appropriate closing phrase"""
        templates = self.CLOSING_TEMPLATES.get(
            outcome, ["Bye!"]
        )
        closing = random.choice(templates)
        
        # Fill in context if needed
        if context and "{time}" in closing:
            closing = closing.format(
                time=context.get("callback_time", "later")
            )
        
        return closing
    
    def validate_closing(self, response: str) -> dict:
        """Validate closing doesn't have issues"""
        issues = []
        
        # Check for new questions
        if "?" in response and self.closing_started:
            issues.append("Question in closing")
        
        # Check for content after bye
        bye_patterns = ["bye", "bye!", "goodbye", "take care"]
        response_lower = response.lower()
        
        for pattern in bye_patterns:
            idx = response_lower.find(pattern)
            if idx >= 0:
                after_bye = response[idx + len(pattern):].strip()
                # Allow only punctuation or very short additions
                if len(after_bye) > 10:
                    issues.append(f"Content after goodbye: '{after_bye}'")
        
        # Check for multiple goodbyes
        bye_count = sum(response_lower.count(p) 
                        for p in bye_patterns)
        if bye_count > 1:
            issues.append("Multiple goodbyes")
        
        return {
            "valid": len(issues) == 0,
            "issues": issues
        }
    
    def should_disconnect_now(self) -> bool:
        """Check if we should disconnect immediately"""
        return self.goodbye_said


class ClosingProtocol:
    """Execute proper closing protocol"""
    
    def __init__(self):
        self.manager = ClosingManager()
    
    def execute_close(self, outcome: str, 
                       captured_data: dict) -> dict:
        """Execute closing protocol"""
        # Step 1: Get closing phrase
        closing = self.manager.get_closing(outcome, captured_data)
        
        # Step 2: Validate
        validation = self.manager.validate_closing(closing)
        if not validation["valid"]:
            # Use fallback
            closing = self.get_fallback_closing(outcome)
        
        return {
            "closing_phrase": closing,
            "disconnect_after_ms": 500,  # Brief pause then disconnect
            "allow_caller_response": False,  # Don't wait for caller
            "no_further_content": True
        }
    
    def get_fallback_closing(self, outcome: str) -> str:
        """Simple fallback closings"""
        fallbacks = {
            "qualified": "Got it! Bye!",
            "not_interested": "No worries, bye!",
            "callback": "Talk later, bye!",
            "default": "Bye!"
        }
        return fallbacks.get(outcome, fallbacks["default"])

instructions: |
  ## CLOSING RULES
  
  When closing the call:
  
  1. ACKNOWLEDGE briefly (if qualified/callback):
     - "Cool, noted!" or "Got it!"
     - Keep under 5 words
  
  2. WELL-WISH (optional, brief):
     - "Take care!" or "All the best!"
  
  3. SAY BYE:
     - "Bye!" — that's it
  
  4. STOP TALKING:
     - No new information after bye
     - No new questions after bye
     - Don't wait for caller to respond
  
  EXAMPLES:
  ✓ "Cool, all noted! Bye!"
  ✓ "Got it! Take care, bye!"
  ✓ "All good, understood. Bye!"
  
  ✗ "Bye! Oh and the playbook will have..."
  ✗ "Bye! ... ... ... Bye?"
  ✗ "Your query has been registered. A representative..."
  
  CLOSE ONCE:
  - Say bye ONCE
  - Don't add "one more thing"
  - Don't ask questions in closing
  
  Keep it casual, brief, and final.