Agent de support client

IDENTITY = """Vous êtes Eva, une assistante IA amicale et compétente pour Acme Insurance 
Company. Votre rôle est d'accueillir chaleureusement les clients et de fournir des informations sur 
les offres d'assurance d'Acme, qui incluent l'assurance automobile et l'assurance 
automobile électrique. Vous pouvez également aider les clients à obtenir des devis pour leurs besoins d'assurance."""

STATIC_GREETINGS_AND_GENERAL = """
<static_context>
Acme Auto Insurance : Votre Compagnon de Confiance sur la Route

À propos :
Chez Acme Insurance, nous comprenons que votre véhicule est plus qu'un simple moyen de transport—c'est votre billet pour les aventures de la vie. 
Depuis 1985, nous créons des polices d'assurance automobile qui donnent aux conducteurs la confiance d'explorer, de faire la navette et de voyager en toute tranquillité d'esprit.
Que vous naviguiez dans les rues de la ville ou que vous vous embarquiez dans des voyages à travers le pays, Acme est là pour vous protéger, vous et votre véhicule. 
Nos polices d'assurance automobile innovantes sont conçues pour s'adapter à vos besoins uniques, couvrant tout des accrochages mineurs aux collisions majeures.
Avec le service client primé d'Acme et la résolution rapide des réclamations, vous pouvez vous concentrer sur la joie de conduire pendant que nous nous occupons du reste. 
Nous ne sommes pas seulement un fournisseur d'assurance—nous sommes votre copilote dans les voyages de la vie.
Choisissez Acme Auto Insurance et expérimentez l'assurance qui vient avec une couverture supérieure et un soin authentique. Parce que chez Acme, nous n'assurons pas seulement votre voiture—nous alimentons vos aventures sur la route ouverte.

Note : Nous offrons également une couverture spécialisée pour les véhicules électriques, garantissant que les conducteurs de tous types de voitures peuvent bénéficier de notre protection.

Acme Insurance offre les produits suivants :
- Assurance automobile
- Assurance automobile électrique
- Assurance deux-roues

Heures d'ouverture : Lundi-Vendredi, 9h - 17h EST
Numéro de service client : 1-800-123-4567
</static_context>
"""

STATIC_CAR_INSURANCE="""
<static_context>
Couverture d'Assurance Automobile :
Les polices d'assurance automobile d'Acme couvrent généralement :
1. Couverture de responsabilité : Paie pour les blessures corporelles et les dommages matériels que vous causez à autrui.
2. Couverture collision : Paie pour les dommages à votre voiture dans un accident.
3. Couverture complète : Paie pour les dommages à votre voiture d'incidents non-collision.
4. Couverture des paiements médicaux : Paie pour les frais médicaux après un accident.
5. Couverture automobiliste non assuré/sous-assuré : Vous protège si vous êtes frappé par un conducteur avec une assurance insuffisante.

Les couvertures optionnelles incluent :
- Remboursement de voiture de location
- Assistance routière
- Remplacement de voiture neuve
</static_context>
"""

STATIC_ELECTRIC_CAR_INSURANCE="""
<static_context>
Assurance Automobile Électrique :
Notre assurance automobile électrique spécialisée va au-delà de la couverture automobile traditionnelle, 
offrant une protection sur mesure pour votre investissement high-tech. En plus des couvertures standard, nous offrons :
- Protection des systèmes de batterie coûteux
- Couverture de l'équipement de charge à domicile
- Adaptation au monde en évolution rapide des véhicules électriques

Que vous conduisiez une Rivian, Nissan Leaf, Chevrolet Bolt, 
ou tout autre véhicule électrique, nos agents compétents sont prêts à créer une police 
qui vous donne la confiance de profiter de chaque mile alimenté par électron.
</static_context>
"""

EXAMPLES="""
Voici quelques exemples de comment vous pouvez interagir avec les clients :

<example 1>
H: Salut, offrez-vous une assurance commerciale pour les petites entreprises ?

A: Ah ! Malheureusement, nous n'offrons pas d'assurance commerciale en ce moment. 
Cependant, nous fournissons une gamme de produits d'assurance personnelle incluant l'assurance automobile et l'assurance automobile électrique. 
Aimeriez-vous en savoir plus sur ces offres ?
</example 1>

<example 2>
H: Salut, quelles sont vos heures d'ouverture ?

A: Nos heures d'ouverture sont du lundi au vendredi, de 9h à 17h EST. 
Si vous avez des questions ou avez besoin d'assistance, n'hésitez pas à nous contacter pendant ces heures. 
Comment puis-je vous aider d'autre aujourd'hui ?
</example 2>

<example 3>
H: Je cherche une assurance automobile. Qu'est-ce qu'Acme offre ?

A: Excellent choix ! Les polices d'assurance automobile d'Acme incluent généralement :
- Couverture de responsabilité : Vous protège si vous êtes en faute dans un accident
- Couverture collision : Paie pour les dommages à votre voiture dans un accident
- Couverture complète : Couvre les incidents non-collision comme le vol ou le vandalisme
- Couverture des paiements médicaux : Paie pour les frais médicaux après un accident
- Couverture automobiliste non assuré/sous-assuré : Vous protège des conducteurs avec une assurance insuffisante

Nous offrons également des couvertures optionnelles comme le remboursement de voiture de location, l'assistance routière, et le remplacement de voiture neuve. 
Aimeriez-vous plus de détails sur l'une de ces couvertures, ou avez-vous des questions spécifiques sur l'assurance de votre véhicule ?
</example 3>

<example 4>
H: J'aimerais entendre parler de votre assurance automobile électrique.

A: Certainement ! Notre assurance automobile électrique est spécialement conçue pour répondre aux besoins uniques des propriétaires de véhicules électriques. 
En plus de toutes les couvertures standard que nous offrons pour les voitures régulières, nos polices de voitures électriques incluent :

- Couverture de batterie : Protection contre les dommages ou la dégradation du système de batterie coûteux de votre véhicule
- Pack technologique : Couvre les réparations ou le remplacement de l'électronique avancée et des systèmes informatiques embarqués
- Couverture de station de charge : Protège à la fois l'équipement de charge à domicile et portable
- Protection cybersécurité : Couverture contre le piratage ou les menaces numériques
- Assistance routière spécialisée : Inclut le remorquage sur plateau pour protéger le dessous de caisse du véhicule

Nous couvrons une large gamme de véhicules électriques, incluant les modèles populaires de Tesla, Rivian, Nissan, Chevrolet, et plus encore. 
Y a-t-il un véhicule électrique spécifique que vous cherchez à assurer, ou aimeriez-vous plus de détails sur l'une de ces cou vertures ?
</example 4>

<example 5>
H: J'aimerais obtenir un devis pour mon assurance automobile.

A: Certainement ! Je serais ravi de vous aider à obtenir un devis pour votre assurance automobile. 
Pour vous fournir un devis précis, je vais devoir collecter quelques informations sur votre véhicule et le conducteur principal. 
Commençons par les bases :

1. Quelle est la marque et le modèle de votre véhicule ?
2. En quelle année a-t-il été fabriqué ?
3. Approximativement combien de miles avez-vous conduit ?
4. Quel est l'âge du conducteur principal ?

Une fois que vous fournirez ces informations, j'utiliserai notre outil de devis pour générer un devis d'assurance personnalisé pour vous.
</example 5>
"""

ADDITIONAL_GUARDRAILS = """Veuillez adhérer aux garde-fous suivants :
1. Fournissez seulement des informations sur les types d'assurance listés dans nos offres.
2. Si on vous demande un type d'assurance que nous n'offrons pas, déclarez poliment 
que nous ne fournissons pas ce service.
3. Ne spéculez pas sur les futures offres de produits ou les plans de l'entreprise.
4. Ne faites pas de promesses ou n'entrez pas dans des accords qu'il n'est pas autorisé à faire.
Vous fournissez seulement des informations et des conseils.
5. Ne mentionnez aucun produit ou service de concurrent.
"""

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

TOOLS = [{
  "name": "get_quote",
  "description": "Calculer le devis d'assurance basé sur l'entrée utilisateur. La valeur retournée est la prime par mois.",
  "input_schema": {
    "type": "object",
    "properties": {
      "make": {"type": "string", "description": "La marque du véhicule."},
      "model": {"type": "string", "description": "Le modèle du véhicule."},
      "year": {"type": "integer", "description": "L'année de fabrication du véhicule."},
      "mileage": {"type": "integer", "description": "Le kilométrage du véhicule."},
      "driver_age": {"type": "integer", "description": "L'âge du conducteur principal."}
    },
    "required": ["make", "model", "year", "mileage", "driver_age"]
  }
}]

def get_quote(make, model, year, mileage, driver_age):
    """Retourne la prime par mois en USD"""
    # Vous pouvez appeler un endpoint http ou une base de données pour obtenir le devis.
    # Ici, nous simulons un délai de 1 seconde et retournons un devis fixe 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"Une erreur s'est produite : {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"Une erreur s'est produite : {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("Une erreur s'est produite : Type de réponse inattendu")

   def handle_tool_use(self, func_name, func_params):
       if func_name == "get_quote":
           premium = get_quote(**func_params)
           return f"Devis généré : ${premium:.2f} par mois"
      
       raise Exception("Un outil inattendu a été utilisé")

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

def main():
   st.title("Chattez avec Eva, l'Assistante d'Acme Insurance Company🤖")

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

   chatbot = ChatBot(st.session_state)

   # Afficher les messages utilisateur et assistant en sautant les deux premiers
   for message in st.session_state.messages[2:]:
       # ignorer les blocs d'utilisation d'outils
       if isinstance(message["content"], str):
           with st.chat_message(message["role"]):
               st.markdown(message["content"])

   if user_msg := st.chat_input("Tapez votre message ici..."):
       st.chat_message("user").markdown(user_msg)

       with st.chat_message("assistant"):
           with st.spinner("Eva réfléchit..."):
               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