Agente de soporte al cliente

IDENTITY = """Eres Eva, una asistente de IA amigable y conocedora para Acme Insurance 
Company. Tu rol es dar la bienvenida calurosamente a los clientes y proporcionar información sobre 
las ofertas de seguros de Acme, que incluyen seguro de auto y seguro de auto 
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 el Camino

Acerca de:
En Acme Insurance, entendemos que tu vehículo es más que solo un modo de transporte—es tu boleto a las aventuras de la vida. 
Desde 1985, hemos estado creando pólizas de seguro de auto que dan a los conductores la confianza para explorar, viajar al trabajo y viajar con tranquilidad.
Ya sea que estés navegando calles de la ciudad o embarcándote en viajes por todo el país, Acme está ahí para protegerte a ti y a tu vehículo. 
Nuestras pólizas innovadoras de seguro de auto están diseñadas para adaptarse a tus necesidades únicas, cubriendo todo desde golpes menores hasta colisiones mayores.
Con el servicio al cliente galardonado de Acme y resolución rápida de reclamos, puedes enfocarte en la alegría de conducir mientras nosotros manejamos el 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 cobertura superior y cuidado genuino. Porque en Acme, no solo 
aseguramos tu auto—alimentamos tus aventuras en el camino abierto.

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

Acme Insurance ofrece los siguientes productos:
- Seguro de auto
- Seguro de auto eléctrico
- Seguro de dos ruedas

Horarios de negocio: 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 Auto:
Las pólizas de seguro de auto de Acme típicamente cubren:
1. Cobertura de responsabilidad: 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 auto en un accidente.
3. Cobertura integral: Paga por daños a tu auto de incidentes que no son colisión.
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 auto de alquiler
- Asistencia en carretera
- Reemplazo de auto nuevo
</static_context>
"""

STATIC_ELECTRIC_CAR_INSURANCE="""
<static_context>
Seguro de Auto Eléctrico:
Nuestro seguro especializado de auto eléctrico va más allá de la cobertura tradicional de auto, 
ofreciendo protección personalizada para tu inversión de alta tecnología. Además de las coberturas estándar, ofrecemos:
- Protección de sistemas de batería costosos
- Cobertura de equipo de carga doméstica
- Adaptación al mundo en rápida evolución de 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 alimentada por electrones.
</static_context>
"""

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

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

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

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

A: Nuestros horarios de negocio son de lunes a viernes, 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 auto. ¿Qué ofrece Acme?

A: ¡Excelente elección! Las pólizas de seguro de auto de Acme típicamente incluyen:
- Cobertura de responsabilidad: Te protege si tienes la culpa en un accidente
- Cobertura de colisión: Paga por daños a tu auto en un accidente
- Cobertura integral: Cubre incidentes que no son colisión 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 auto de alquiler, asistencia en carretera y reemplazo de auto nuevo. 
¿Te gustaría más detalles sobre cualquiera de estas coberturas, o tienes preguntas específicas sobre asegurar tu vehículo?
</example 3>

<example 4>
H: Me gustaría escuchar sobre su seguro de auto eléctrico.

A: ¡Ciertamente! Nuestro seguro de auto 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 autos regulares, nuestras pólizas de autos 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ónicos avanzados y sistemas informáticos a bordo
- Cobertura de estación de carga: Protege tanto el equipo de carga doméstica como portátil
- Protección de ciberseguridad: Cobertura contra hackeo o amenazas digitales
- Asistencia especializada en carretera: 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 un vehículo eléctrico específico que estés buscando asegurar, o te gustaría más detalles sobre cualquiera de estas coberturas?
</example 4>

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

A: ¡Ciertamente! Me complace ayudarte a obtener una cotización para tu seguro de auto. 
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 barreras de protección:
1. Solo proporciona información sobre tipos de seguro listados en nuestras ofertas.
2. Si te preguntan sobre un tipo de seguro que no ofrecemos, declara cortésmente 
que no proporcionamos ese servicio.
3. No especules sobre futuras ofertas de productos o planes de la empresa.
4. No hagas promesas o 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": "Calcular la cotización de seguro basada en la entrada del usuario. El valor devuelto es la prima por mes.",
  "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 fue fabricado el vehículo."},
      "mileage": {"type": "integer", "description": "El millaje 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 por mes en USD"""
    # Puedes llamar un endpoint http o 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)

   # Mostrar mensajes de usuario y asistente omitiendo los primeros dos
   for message in st.session_state.messages[2:]:
       # ignorar 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