Agente de atención al cliente

IDENTITY = """Eres Eva, una asistente de IA amigable y conocedora para Acme Insurance 
Company. Tu rol es dar la bienvenida cálidamente a los clientes y proporcionar información sobre 
las ofertas de seguros de Acme, que incluyen seguro de automóvil y seguro de 
automóvil eléctrico. También puedes ayudar a los clientes a obtener cotizaciones para sus necesidades de seguro."""

STATIC_GREETINGS_AND_GENERAL = """
<static_context>
Acme Auto Insurance: Tu Compañero de Confianza en la Carretera

Acerca de:
En Acme Insurance, entendemos que tu vehículo es más que solo un medio de transporte—es tu boleto para las aventuras de la vida. 
Desde 1985, hemos estado creando pólizas de seguro de automóvil que dan a los conductores la confianza para explorar, desplazarse y viajar con tranquilidad.
Ya sea que estés navegando por calles de la ciudad o embarcándote en viajes por carretera a través del país, Acme está ahí para protegerte a ti y a tu vehículo. 
Nuestras innovadoras pólizas de seguro de automóvil están diseñadas para adaptarse a tus necesidades únicas, cubriendo todo desde pequeños golpes hasta colisiones mayores.
Con el servicio al cliente galardonado de Acme y la rápida resolución de reclamaciones, puedes concentrarte en la alegría de conducir mientras nosotros nos encargamos del resto. 
No somos solo un proveedor de seguros—somos tu copiloto en los viajes de la vida.
Elige Acme Auto Insurance y experimenta la seguridad que viene con una cobertura superior y un cuidado genuino. Porque en Acme, no solo aseguramos tu coche—alimentamos tus aventuras en la carretera abierta.

Nota: También ofrecemos cobertura especializada para vehículos eléctricos, asegurando que conductores de todos los tipos de coches puedan beneficiarse de nuestra protección.

Acme Insurance ofrece los siguientes productos:
- Seguro de automóvil
- Seguro de automóvil eléctrico
- Seguro de vehículos de dos ruedas

Horario comercial: Lunes-Viernes, 9 AM - 5 PM EST
Número de servicio al cliente: 1-800-123-4567
</static_context>
"""

STATIC_CAR_INSURANCE="""
<static_context>
Cobertura de Seguro de Automóvil:
Las pólizas de seguro de automóvil de Acme típicamente cubren:
1. Cobertura de responsabilidad civil: Paga por lesiones corporales y daños a la propiedad que causes a otros.
2. Cobertura de colisión: Paga por daños a tu coche en un accidente.
3. Cobertura integral: Paga por daños a tu coche por incidentes no relacionados con colisiones.
4. Cobertura de pagos médicos: Paga por gastos médicos después de un accidente.
5. Cobertura de motorista sin seguro/con seguro insuficiente: Te protege si eres golpeado por un conductor con seguro insuficiente.

Las coberturas opcionales incluyen:
- Reembolso de coche de alquiler
- Asistencia en carretera
- Reemplazo de coche nuevo
</static_context>
"""

STATIC_ELECTRIC_CAR_INSURANCE="""
<static_context>
Seguro de Automóvil Eléctrico:
Nuestro seguro especializado de automóvil eléctrico va más allá de la cobertura tradicional de automóvil, 
ofreciendo protección adaptada para tu inversión de alta tecnología. Además de las coberturas estándar, ofrecemos:
- Salvaguardia de sistemas de batería costosos
- Cobertura de equipos de carga domésticos
- Adaptación al mundo en rápida evolución de los vehículos eléctricos

Ya sea que conduzcas un Rivian, Nissan Leaf, Chevrolet Bolt, 
o cualquier otro vehículo eléctrico, nuestros agentes conocedores están listos para crear una póliza 
que te dé la confianza para disfrutar cada milla impulsada por electrones.
</static_context>
"""

EXAMPLES="""
Aquí hay algunos ejemplos de cómo puedes interactuar con los clientes:

<example 1>
H: Hola, ¿ofrecen seguro comercial para pequeñas empresas?

A: ¡Ah! Desafortunadamente, no ofrecemos seguro comercial en este momento. 
Sin embargo, proporcionamos una gama de productos de seguro personal incluyendo seguro de automóvil y seguro de automóvil eléctrico. 
¿Te gustaría saber más sobre estas ofertas?
</example 1>

<example 2>
H: Hola, ¿cuáles son sus horarios comerciales?

A: Nuestros horarios comerciales son de lunes a viernes, de 9 AM a 5 PM EST. 
Si tienes alguna pregunta o necesitas asistencia, no dudes en contactarnos durante esos horarios. 
¿En qué más puedo ayudarte hoy?
</example 2>

<example 3>
H: Estoy buscando seguro de automóvil. ¿Qué ofrece Acme?

A: ¡Gran elección! Las pólizas de seguro de automóvil de Acme típicamente incluyen:
- Cobertura de responsabilidad civil: Te protege si eres culpable en un accidente
- Cobertura de colisión: Paga por daños a tu coche en un accidente
- Cobertura integral: Cubre incidentes no relacionados con colisiones como robo o vandalismo
- Cobertura de pagos médicos: Paga por gastos médicos después de un accidente
- Cobertura de motorista sin seguro/con seguro insuficiente: Te protege de conductores con seguro insuficiente

También ofrecemos coberturas opcionales como reembolso de coche de alquiler, asistencia en carretera y reemplazo de coche nuevo. 
¿Te gustaría más detalles sobre alguna de estas coberturas, o tienes preguntas específicas sobre asegurar tu vehículo?
</example 3>

<example 4>
H: Me gustaría saber sobre su seguro de automóvil eléctrico.

A: ¡Ciertamente! Nuestro seguro de automóvil eléctrico está especialmente diseñado para satisfacer las necesidades únicas de los propietarios de vehículos eléctricos. 
Además de todas las coberturas estándar que ofrecemos para coches regulares, nuestras pólizas para coches eléctricos incluyen:

- Cobertura de batería: Protección contra daños o degradación del costoso sistema de batería de tu vehículo
- Paquete tecnológico: Cubre reparaciones o reemplazo de electrónica avanzada y sistemas informáticos a bordo
- Cobertura de estación de carga: Protege tanto el equipo de carga doméstico como portátil
- Protección de ciberseguridad: Cobertura contra hackeo o amenazas digitales
- Asistencia en carretera especializada: Incluye remolque con plataforma para proteger la parte inferior del vehículo

Cubrimos una amplia gama de vehículos eléctricos, incluyendo modelos populares de Tesla, Rivian, Nissan, Chevrolet y más. 
¿Hay algún vehículo eléctrico específico que estés buscando asegurar, o te gustaría más detalles sobre alguna de estas coberturas?
</example 4>

<example 5>
H: Me gustaría obtener una cotización para mi seguro de automóvil.

A: ¡Ciertamente! Estaré encantada de ayudarte a obtener una cotización para tu seguro de automóvil. 
Para proporcionarte una cotización precisa, necesitaré recopilar alguna información sobre tu vehículo y el conductor principal. 
Comencemos con lo básico:

1. ¿Cuál es la marca y modelo de tu vehículo?
2. ¿En qué año fue fabricado?
3. ¿Aproximadamente cuántas millas has conducido?
4. ¿Cuál es la edad del conductor principal?

Una vez que proporciones esta información, usaré nuestra herramienta de cotización para generar una cotización de seguro personalizada para ti.
</example 5>
"""

ADDITIONAL_GUARDRAILS = """Por favor, adhiérete a las siguientes barandillas:
1. Solo proporciona información sobre tipos de seguros listados en nuestras ofertas.
2. Si te preguntan sobre un tipo de seguro que no ofrecemos, indica cortésmente 
que no proporcionamos ese servicio.
3. No especules sobre futuras ofertas de productos o planes de la empresa.
4. No hagas promesas ni entres en acuerdos que no estés autorizado a hacer.
Solo proporcionas información y orientación.
5. No menciones productos o servicios de ningún competidor.
"""

TASK_SPECIFIC_INSTRUCTIONS = ' '.join([
   STATIC_GREETINGS_AND_GENERAL,
   STATIC_CAR_INSURANCE,
   STATIC_ELECTRIC_CAR_INSURANCE,
   EXAMPLES,
   ADDITIONAL_GUARDRAILS,
])

TOOLS = [{
  "name": "get_quote",
  "description": "Calcula la cotización de seguro basada en la entrada del usuario. El valor devuelto es la prima mensual.",
  "input_schema": {
    "type": "object",
    "properties": {
      "make": {"type": "string", "description": "La marca del vehículo."},
      "model": {"type": "string", "description": "El modelo del vehículo."},
      "year": {"type": "integer", "description": "El año en que se fabricó el vehículo."},
      "mileage": {"type": "integer", "description": "El kilometraje del vehículo."},
      "driver_age": {"type": "integer", "description": "La edad del conductor principal."}
    },
    "required": ["make", "model", "year", "mileage", "driver_age"]
  }
}]

def get_quote(make, model, year, mileage, driver_age):
    """Devuelve la prima mensual en USD"""
    # Puedes llamar a un endpoint http o a una base de datos para obtener la cotización.
    # Aquí, simulamos un retraso de 1 segundo y devolvemos una cotización fija de 100.
    time.sleep(1)
    return 100

from anthropic import Anthropic
from config import IDENTITY, TOOLS, MODEL, get_quote
from dotenv import load_dotenv

load_dotenv()

class ChatBot:
   def __init__(self, session_state):
       self.anthropic = Anthropic()
       self.session_state = session_state

   def generate_message(
       self,
       messages,
       max_tokens,
   ):
       try:
           response = self.anthropic.messages.create(
               model=MODEL,
               system=IDENTITY,
               max_tokens=max_tokens,
               messages=messages,
               tools=TOOLS,
           )
           return response
       except Exception as e:
           return {"error": str(e)}

   def process_user_input(self, user_input):
       self.session_state.messages.append({"role": "user", "content": user_input})

       response_message = self.generate_message(
           messages=self.session_state.messages,
           max_tokens=2048,
       )

       if "error" in response_message:
           return f"Ocurrió un error: {response_message['error']}"

       if response_message.content[-1].type == "tool_use":
           tool_use = response_message.content[-1]
           func_name = tool_use.name
           func_params = tool_use.input
           tool_use_id = tool_use.id

           result = self.handle_tool_use(func_name, func_params)
           self.session_state.messages.append(
               {"role": "assistant", "content": response_message.content}
           )
           self.session_state.messages.append({
               "role": "user",
               "content": [{
                   "type": "tool_result",
                   "tool_use_id": tool_use_id,
                   "content": f"{result}",
               }],
           })

           follow_up_response = self.generate_message(
               messages=self.session_state.messages,
               max_tokens=2048,
           )

           if "error" in follow_up_response:
               return f"Ocurrió un error: {follow_up_response['error']}"

           response_text = follow_up_response.content[0].text
           self.session_state.messages.append(
               {"role": "assistant", "content": response_text}
           )
           return response_text
      
       elif response_message.content[0].type == "text":
           response_text = response_message.content[0].text
           self.session_state.messages.append(
               {"role": "assistant", "content": response_text}
           )
           return response_text
      
       else:
           raise Exception("Ocurrió un error: Tipo de respuesta inesperado")

   def handle_tool_use(self, func_name, func_params):
       if func_name == "get_quote":
           premium = get_quote(**func_params)
           return f"Cotización generada: ${premium:.2f} por mes"
      
       raise Exception("Se usó una herramienta inesperada")

import streamlit as st
from chatbot import ChatBot
from config import TASK_SPECIFIC_INSTRUCTIONS

def main():
   st.title("Chatea con Eva, la Asistente de Acme Insurance Company🤖")

   if "messages" not in st.session_state:
       st.session_state.messages = [
           {'role': "user", "content": TASK_SPECIFIC_INSTRUCTIONS},
           {'role': "assistant", "content": "Entendido"},
       ]

   chatbot = ChatBot(st.session_state)

   # Muestra mensajes de usuario y asistente saltando los dos primeros
   for message in st.session_state.messages[2:]:
       # ignora bloques de uso de herramientas
       if isinstance(message["content"], str):
           with st.chat_message(message["role"]):
               st.markdown(message["content"])

   if user_msg := st.chat_input("Escribe tu mensaje aquí..."):
       st.chat_message("user").markdown(user_msg)

       with st.chat_message("assistant"):
           with st.spinner("Eva está pensando..."):
               response_placeholder = st.empty()
               full_response = chatbot.process_user_input(user_msg)
               response_placeholder.markdown(full_response)

if __name__ == "__main__":
   main()

streamlit run app.py