Customer Support Agent

from typing import Iterator, Dict, Any
import json

class CustomerSupportAgent:
    def __init__(self):
        # Knowledge base - in production, this would be a database
        self.knowledge_base = {
            "pricing": {
                "basic": "$9/month - Up to 1000 requests",
                "pro": "$29/month - Up to 10,000 requests", 
                "enterprise": "Custom pricing - Unlimited requests"
            },
            "features": [
                "Multi-language SDKs (Python, JavaScript, Rust, Go)",
                "Real-time streaming responses",
                "Auto-scaling infrastructure",
                "Framework integrations (LangGraph, CrewAI, etc.)"
            ],
            "troubleshooting": {
                "connection": "Check your API key and agent ID",
                "timeout": "Increase timeout settings or check network",
                "streaming": "Ensure your entrypoint tag ends with '_stream'"
            }
        }
    
    def categorize_query(self, message: str) -> str:
        """Categorize the customer query"""
        message_lower = message.lower()
        
        if any(word in message_lower for word in ["price", "cost", "pricing", "plan"]):
            return "pricing"
        elif any(word in message_lower for word in ["feature", "capability", "what can"]):
            return "features"
        elif any(word in message_lower for word in ["help", "problem", "error", "issue", "troubleshoot"]):
            return "troubleshooting"
        else:
            return "general"

def customer_support_sync(message: str, user_id: str = "anonymous") -> Dict[str, Any]:
    """Synchronous customer support response"""
    agent = CustomerSupportAgent()
    category = agent.categorize_query(message)
    
    response = {
        "message": "",
        "category": category,
        "user_id": user_id,
        "timestamp": "2024-01-01T00:00:00Z"  # In production, use actual timestamp
    }
    
    if category == "pricing":
        response["message"] = f"""Here are our current pricing plans:

**Basic Plan**: {agent.knowledge_base['pricing']['basic']}
- Perfect for small projects and testing

**Pro Plan**: {agent.knowledge_base['pricing']['pro']}
- Great for growing applications

**Enterprise Plan**: {agent.knowledge_base['pricing']['enterprise']}
- Custom solutions for large organizations

Would you like to know more about any specific plan?"""
    
    elif category == "features":
        features_list = "\n".join([f"• {feature}" for feature in agent.knowledge_base['features']])
        response["message"] = f"""RunAgent offers these key features:

{features_list}

Is there a specific feature you'd like to learn more about?"""
    
    elif category == "troubleshooting":
        response["message"] = """I'd be happy to help troubleshoot! Here are some common solutions:

• **Connection issues**: Check your API key and agent ID
• **Timeout errors**: Increase timeout settings or check your network
• **Streaming problems**: Ensure your entrypoint tag ends with '_stream'

Can you describe the specific issue you're experiencing?"""
    
    else:
        response["message"] = """Hello! I'm here to help with any questions about RunAgent. 

I can help you with:
• Pricing and plans
• Features and capabilities  
• Technical troubleshooting
• Getting started guides

What would you like to know?"""
    
    return response

def customer_support_stream(message: str, user_id: str = "anonymous") -> Iterator[str]:
    """Streaming customer support response"""
    agent = CustomerSupportAgent()
    category = agent.categorize_query(message)
    
    # Stream the response with typing effect
    if category == "pricing":
        yield "Let me check our current pricing for you...\n\n"
        yield "**Basic Plan**: $9/month - Up to 1000 requests\n"
        yield "Perfect for small projects and testing\n\n"
        yield "**Pro Plan**: $29/month - Up to 10,000 requests\n"
        yield "Great for growing applications\n\n"
        yield "**Enterprise Plan**: Custom pricing - Unlimited requests\n"
        yield "Custom solutions for large organizations\n\n"
        yield "Would you like to know more about any specific plan?"
    
    elif category == "features":
        yield "Here are RunAgent's key features:\n\n"
        for feature in agent.knowledge_base['features']:
            yield f"• {feature}\n"
        yield "\nIs there a specific feature you'd like to learn more about?"
    
    elif category == "troubleshooting":
        yield "I'd be happy to help troubleshoot!\n\n"
        yield "Here are some common solutions:\n\n"
        yield "• **Connection issues**: Check your API key and agent ID\n"
        yield "• **Timeout errors**: Increase timeout settings or check your network\n"
        yield "• **Streaming problems**: Ensure your entrypoint tag ends with '_stream'\n\n"
        yield "Can you describe the specific issue you're experiencing?"
    
    else:
        yield "Hello! I'm here to help with any questions about RunAgent.\n\n"
        yield "I can help you with:\n"
        yield "• Pricing and plans\n"
        yield "• Features and capabilities\n"
        yield "• Technical troubleshooting\n"
        yield "• Getting started guides\n\n"
        yield "What would you like to know?"

<!DOCTYPE html>
<html>
<head>
    <title>RunAgent Customer Support</title>
    <style>
        body { font-family: Arial, sans-serif; max-width: 800px; margin: 0 auto; padding: 20px; }
        .chat-container { border: 1px solid #ccc; height: 400px; overflow-y: auto; padding: 10px; margin: 20px 0; }
        .message { margin: 10px 0; padding: 10px; border-radius: 5px; }
        .user { background-color: #e3f2fd; text-align: right; }
        .bot { background-color: #f5f5f5; }
        .input-container { display: flex; gap: 10px; }
        input[type="text"] { flex: 1; padding: 10px; border: 1px solid #ccc; border-radius: 5px; }
        button { padding: 10px 20px; background-color: #007bff; color: white; border: none; border-radius: 5px; cursor: pointer; }
        button:hover { background-color: #0056b3; }
        .typing { color: #666; font-style: italic; }
    </style>
</head>
<body>
    <h1>RunAgent Customer Support</h1>
    <div id="chat" class="chat-container"></div>
    <div class="input-container">
        <input type="text" id="messageInput" placeholder="Ask me anything about RunAgent..." />
        <button onclick="sendMessage()">Send</button>
    </div>

    <script>
        // Replace with your actual agent details
        const AGENT_ID = "support_agent_123";
        const API_BASE = "http://127.0.0.1:8451";
        
        function addMessage(content, isUser = false) {
            const chat = document.getElementById('chat');
            const messageDiv = document.createElement('div');
            messageDiv.className = `message ${isUser ? 'user' : 'bot'}`;
            messageDiv.textContent = content;
            chat.appendChild(messageDiv);
            chat.scrollTop = chat.scrollHeight;
        }
        
        function addTyping() {
            const chat = document.getElementById('chat');
            const typingDiv = document.createElement('div');
            typingDiv.className = 'message bot typing';
            typingDiv.id = 'typing';
            typingDiv.textContent = 'Bot is typing...';
            chat.appendChild(typingDiv);
            chat.scrollTop = chat.scrollHeight;
        }
        
        function removeTyping() {
            const typing = document.getElementById('typing');
            if (typing) typing.remove();
        }
        
        async function sendMessage() {
            const input = document.getElementById('messageInput');
            const message = input.value.trim();
            if (!message) return;
            
            // Add user message
            addMessage(message, true);
            input.value = '';
            
            // Add typing indicator
            addTyping();
            
            try {
                // Call the support agent
                const response = await fetch(`${API_BASE}/api/v1/agents/${AGENT_ID}/execute/support`, {
                    method: 'POST',
                    headers: {
                        'Content-Type': 'application/json',
                    },
                    body: JSON.stringify({
                        message: message,
                        user_id: "web_user"
                    })
                });
                
                const data = await response.json();
                removeTyping();
                addMessage(data.message);
                
            } catch (error) {
                removeTyping();
                addMessage("Sorry, I'm having trouble connecting. Please try again later.");
                console.error('Error:', error);
            }
        }
        
        // Allow sending with Enter key
        document.getElementById('messageInput').addEventListener('keypress', function(e) {
            if (e.key === 'Enter') {
                sendMessage();
            }
        });
        
        // Welcome message
        addMessage("Hello! I'm your RunAgent support assistant. How can I help you today?");
    </script>
</body>
</html>