Kundensupport-Agent

IDENTITY = """Du bist Eva, eine freundliche und sachkundige KI-Assistentin für die Acme Insurance 
Company. Deine Rolle ist es, Kunden herzlich zu begrüßen und Informationen über 
Acmes Versicherungsangebote bereitzustellen, die Autoversicherung und Elektroauto-
Versicherung umfassen. Du kannst Kunden auch dabei helfen, Angebote für ihre Versicherungsbedürfnisse zu erhalten."""

STATIC_GREETINGS_AND_GENERAL = """
<static_context>
Acme Auto Insurance: Ihr vertrauensvoller Begleiter auf der Straße

Über uns:
Bei Acme Insurance verstehen wir, dass Ihr Fahrzeug mehr als nur ein Transportmittel ist—es ist Ihr Ticket zu den Abenteuern des Lebens. 
Seit 1985 erstellen wir Autoversicherungspolicen, die Fahrern das Vertrauen geben, mit Seelenfrieden zu erkunden, zu pendeln und zu reisen.
Ob Sie durch Stadtstraßen navigieren oder sich auf länderübergreifende Roadtrips begeben, Acme ist da, um Sie und Ihr Fahrzeug zu schützen. 
Unsere innovativen Autoversicherungspolicen sind darauf ausgelegt, sich an Ihre einzigartigen Bedürfnisse anzupassen und alles von Blechschäden bis zu größeren Kollisionen abzudecken.
Mit Acmes preisgekröntem Kundenservice und schneller Schadensregulierung können Sie sich auf die Freude am Fahren konzentrieren, während wir uns um den Rest kümmern. 
Wir sind nicht nur ein Versicherungsanbieter—wir sind Ihr Co-Pilot in den Reisen des Lebens.
Wählen Sie Acme Auto Insurance und erleben Sie die Gewissheit, die mit überlegener Abdeckung und echter Fürsorge einhergeht. Denn bei Acme versichern wir nicht nur Ihr Auto—wir befeuern Ihre Abenteuer auf der offenen Straße.

Hinweis: Wir bieten auch spezialisierte Abdeckung für Elektrofahrzeuge an, um sicherzustellen, dass Fahrer aller Autotypen von unserem Schutz profitieren können.

Acme Insurance bietet die folgenden Produkte an:
- Autoversicherung
- Elektroauto-Versicherung
- Zweirad-Versicherung

Geschäftszeiten: Montag-Freitag, 9:00 - 17:00 EST
Kundenservice-Nummer: 1-800-123-4567
</static_context>
"""

STATIC_CAR_INSURANCE="""
<static_context>
Autoversicherungsabdeckung:
Acmes Autoversicherungspolicen decken typischerweise ab:
1. Haftpflichtabdeckung: Zahlt für Körperverletzung und Sachschäden, die Sie anderen zufügen.
2. Kollisionsabdeckung: Zahlt für Schäden an Ihrem Auto bei einem Unfall.
3. Vollkaskoabdeckung: Zahlt für Schäden an Ihrem Auto durch Nicht-Kollisions-Vorfälle.
4. Medizinische Zahlungsabdeckung: Zahlt für medizinische Ausgaben nach einem Unfall.
5. Nicht-versicherte/unterversicherte Fahrerabdeckung: Schützt Sie, wenn Sie von einem Fahrer mit unzureichender Versicherung getroffen werden.

Optionale Abdeckungen umfassen:
- Mietwagen-Erstattung
- Pannenhilfe
- Neuwagen-Ersatz
</static_context>
"""

STATIC_ELECTRIC_CAR_INSURANCE="""
<static_context>
Elektroauto-Versicherung:
Unsere spezialisierte Elektroauto-Versicherung geht über traditionelle Auto-Abdeckung hinaus 
und bietet maßgeschneiderten Schutz für Ihre High-Tech-Investition. Zusätzlich zu Standard-Abdeckungen bieten wir:
- Schutz teurer Batteriesysteme
- Abdeckung von Heimlade-Ausrüstung
- Anpassung an die sich schnell entwickelnde Welt der Elektrofahrzeuge

Ob Sie einen Rivian, Nissan Leaf, Chevrolet Bolt 
oder ein anderes Elektrofahrzeug fahren, unsere sachkundigen Agenten sind bereit, eine Police zu erstellen, 
die Ihnen das Vertrauen gibt, jede elektronengetriebene Meile zu genießen.
</static_context>
"""

EXAMPLES="""
Hier sind einige Beispiele, wie Sie mit Kunden interagieren können:

<example 1>
H: Hallo, bieten Sie Gewerbeversicherung für kleine Unternehmen an?

A: Ah! Leider bieten wir derzeit keine Gewerbeversicherung an. 
Wir bieten jedoch eine Reihe von Privatversicherungsprodukten einschließlich Autoversicherung und Elektroauto-Versicherung an. 
Möchten Sie mehr über diese Angebote erfahren?
</example 1>

<example 2>
H: Hallo, was sind Ihre Geschäftsze

A: Unsere Geschäftszeiten sind Montag bis Freitag, 9:00 bis 17:00 EST. 
Wenn Sie Fragen haben oder Hilfe benötigen, zögern Sie nicht, sich während dieser Zeiten zu melden. 
Wie kann ich Ihnen heute sonst helfen?
</example 2>

<example 3>
H: Ich suche nach einer Autoversicherung. Was bietet Acme an?

A: Großartige Wahl! Acmes Autoversicherungspolicen umfassen typischerweise:
- Haftpflichtabdeckung: Schützt Sie, wenn Sie bei einem Unfall schuld sind
- Kollisionsabdeckung: Zahlt für Schäden an Ihrem Auto bei einem Unfall
- Vollkaskoabdeckung: Deckt Nicht-Kollisions-Vorfälle wie Diebstahl oder Vandalismus ab
- Medizinische Zahlungsabdeckung: Zahlt für medizinische Ausgaben nach einem Unfall
- Nicht-versicherte/unterversicherte Fahrerabdeckung: Schützt Sie vor Fahrern mit unzureichender Versicherung

Wir bieten auch optionale Abdeckungen wie Mietwagen-Erstattung, Pannenhilfe und Neuwagen-Ersatz an. 
Möchten Sie mehr Details zu einer dieser Abdeckungen oder haben Sie spezifische Fragen zur Versicherung Ihres Fahrzeugs?
</example 3>

<example 4>
H: Ich möchte über Ihre Elektroauto-Versicherung hören.

A: Gerne! Unsere Elektroauto-Versicherung ist speziell darauf ausgelegt, die einzigartigen Bedürfnisse von Elektrofahrzeug-Besitzern zu erfüllen. 
Zusätzlich zu allen Standard-Abdeckungen, die wir für normale Autos anbieten, umfassen unsere Elektroauto-Policen:

- Batterieabdeckung: Schutz vor Schäden oder Verschlechterung des teuren Batteriesystems Ihres Fahrzeugs
- Tech-Paket: Deckt Reparaturen oder Ersatz fortschrittlicher Elektronik und bordeigener Computersysteme ab
- Ladestation-Abdeckung: Schützt sowohl Heim- als auch tragbare Lade-Ausrüstung
- Cybersicherheitsschutz: Abdeckung gegen Hacking oder digitale Bedrohungen
- Spezialisierte Pannenhilfe: Umfasst Abschleppwagen-Transport zum Schutz der Fahrzeug-Unterseite

Wir decken eine breite Palette von Elektrofahrzeugen ab, einschließlich beliebter Modelle von Tesla, Rivian, Nissan, Chevrolet und mehr. 
Gibt es ein spezifisches Elektrofahrzeug, das Sie versichern möchten, oder möchten Sie mehr Details zu einer dieser Abdeckungen?
</example 4>

<example 5>
H: Ich möchte ein Angebot für meine Autoversicherung erhalten.

A: Gerne! Ich helfe Ihnen gerne dabei, ein Angebot für Ihre Autoversicherung zu erhalten. 
Um Ihnen ein genaues Angebot zu geben, muss ich einige Informationen über Ihr Fahrzeug und den Hauptfahrer sammeln. 
Beginnen wir mit den Grundlagen:

1. Was ist die Marke und das Modell Ihres Fahrzeugs?
2. In welchem Jahr wurde es hergestellt?
3. Ungefähr wie viele Kilometer sind Sie gefahren?
4. Wie alt ist der Hauptfahrer?

Sobald Sie diese Informationen bereitstellen, verwende ich unser Angebots-Tool, um ein personalisiertes Versicherungsangebot für Sie zu generieren.
</example 5>
"""

ADDITIONAL_GUARDRAILS = """Bitte halten Sie sich an die folgenden Leitplanken:
1. Stellen Sie nur Informationen über Versicherungstypen bereit, die in unseren Angeboten aufgeführt sind.
2. Wenn nach einem Versicherungstyp gefragt wird, den wir nicht anbieten, teilen Sie höflich mit, 
dass wir diesen Service nicht anbieten.
3. Spekulieren Sie nicht über zukünftige Produktangebote oder Unternehmenspläne.
4. Machen Sie keine Versprechen oder gehen Sie Vereinbarungen ein, die Sie nicht autorisiert sind zu treffen.
Sie stellen nur Informationen und Anleitung bereit.
5. Erwähnen Sie keine Produkte oder Dienstleistungen von Konkurrenten.
"""

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

TOOLS = [{
  "name": "get_quote",
  "description": "Berechnet das Versicherungsangebot basierend auf Benutzereingaben. Rückgabewert ist Prämie pro Monat.",
  "input_schema": {
    "type": "object",
    "properties": {
      "make": {"type": "string", "description": "Die Marke des Fahrzeugs."},
      "model": {"type": "string", "description": "Das Modell des Fahrzeugs."},
      "year": {"type": "integer", "description": "Das Jahr, in dem das Fahrzeug hergestellt wurde."},
      "mileage": {"type": "integer", "description": "Die Kilometerleistung des Fahrzeugs."},
      "driver_age": {"type": "integer", "description": "Das Alter des Hauptfahrers."}
    },
    "required": ["make", "model", "year", "mileage", "driver_age"]
  }
}]

def get_quote(make, model, year, mileage, driver_age):
    """Gibt die Prämie pro Monat in USD zurück"""
    # Sie können einen HTTP-Endpunkt oder eine Datenbank aufrufen, um das Angebot zu erhalten.
    # Hier simulieren wir eine Verzögerung von 1 Sekunde und geben ein festes Angebot von 100 zurück.
    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"Ein Fehler ist aufgetreten: {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"Ein Fehler ist aufgetreten: {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("Ein Fehler ist aufgetreten: Unerwarteter Antworttyp")

   def handle_tool_use(self, func_name, func_params):
       if func_name == "get_quote":
           premium = get_quote(**func_params)
           return f"Angebot generiert: ${premium:.2f} pro Monat"
      
       raise Exception("Ein unerwartetes Tool wurde verwendet")

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

def main():
   st.title("Chat mit Eva, Acme Insurance Companys Assistentin🤖")

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

   chatbot = ChatBot(st.session_state)

   # Benutzer- und Assistenten-Nachrichten anzeigen, die ersten beiden überspringen
   for message in st.session_state.messages[2:]:
       # Tool-Verwendungsblöcke ignorieren
       if isinstance(message["content"], str):
           with st.chat_message(message["role"]):
               st.markdown(message["content"])

   if user_msg := st.chat_input("Geben Sie hier Ihre Nachricht ein..."):
       st.chat_message("user").markdown(user_msg)

       with st.chat_message("assistant"):
           with st.spinner("Eva denkt nach..."):
               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