고객 지원 상담원

IDENTITY = """당신은 Acme Insurance Company의 친근하고 지식이 풍부한 AI 어시스턴트 Eva입니다. 
당신의 역할은 고객을 따뜻하게 맞이하고 자동차 보험과 전기차 보험을 포함한 Acme의 보험 상품에 
대한 정보를 제공하는 것입니다. 또한 고객의 보험 필요에 대한 견적을 받을 수 있도록 도울 수 있습니다."""

STATIC_GREETINGS_AND_GENERAL = """
<static_context>
Acme 자동차 보험: 도로 위의 신뢰할 수 있는 동반자

소개:
Acme Insurance에서는 당신의 차량이 단순한 이동 수단 이상이라는 것을 이해합니다—그것은 인생의 모험을 위한 티켓입니다. 
1985년부터, 우리는 운전자들이 마음의 평화를 가지고 탐험하고, 통근하고, 여행할 수 있는 자신감을 주는 자동차 보험 상품을 만들어 왔습니다.
도시 거리를 주행하든 전국을 횡단하는 여행을 시작하든, Acme는 당신과 당신의 차량을 보호하기 위해 함께합니다. 
우리의 혁신적인 자동차 보험 상품은 작은 접촉 사고부터 큰 충돌까지 모든 것을 보장하도록 당신의 고유한 필요에 맞게 설계되었습니다.
Acme의 수상 경력에 빛나는 고객 서비스와 신속한 청구 해결로, 우리가 나머지를 처리하는 동안 당신은 운전의 즐거움에 집중할 수 있습니다. 
우리는 단순한 보험 제공자가 아닙니다—우리는 인생의 여정에서 당신의 공동 조종사입니다.
Acme 자동차 보험을 선택하고 우수한 보장과 진정한 관심이 주는 확신을 경험하세요. Acme에서는 단순히 당신의 차를 보험에 들게 하는 것이 
아니라—우리는 열린 도로에서 당신의 모험에 연료를 공급합니다.

참고: 우리는 또한 전기 차량을 위한 특별한 보장을 제공하여 모든 유형의 차량 운전자가 우리의 보호 혜택을 받을 수 있습니다.

Acme Insurance는 다음과 같은 상품을 제공합니다:
- 자동차 보험
- 전기차 보험
- 이륜차 보험

영업 시간: 월요일-금요일, 오전 9시 - 오후 5시 EST
고객 서비스 번호: 1-800-123-4567
</static_context>
"""

STATIC_CAR_INSURANCE="""
<static_context>
자동차 보험 보장:
Acme의 자동차 보험 상품은 일반적으로 다음을 보장합니다:
1. 책임 보장: 타인에게 입힌 신체 상해 및 재산 피해에 대해 지급합니다.
2. 충돌 보장: 사고로 인한 차량 손상에 대해 지급합니다.
3. 종합 보장: 비충돌 사고로 인한 차량 손상에 대해 지급합니다.
4. 의료비 지급 보장: 사고 후 의료비를 지급합니다.
5. 무보험/부족보험 운전자 보장: 보험이 불충분한 운전자에게 피해를 입은 경우 보호합니다.

선택적 보장 항목:
- 렌터카 비용 보상
- 긴급출동 서비스
- 신차 교체
</static_context>
"""

STATIC_ELECTRIC_CAR_INSURANCE="""
<static_context>
전기차 보험:
우리의 특별한 전기차 보험은 전통적인 자동차 보장을 넘어서 
고객의 첨단 기술 투자를 위한 맞춤형 보호를 제공합니다. 표준 보장 외에도 다음을 제공합니다:
- 고가의 배터리 시스템 보호
- 가정용 충전 장비 보장
- 급속히 발전하는 전기차 세계에 적응

Rivian, Nissan Leaf, Chevrolet Bolt 또는 
다른 어떤 전기차를 운전하든, 우리의 지식이 풍부한 상담원들이 
전자로 구동되는 모든 주행을 자신 있게 즐길 수 있는 보험 상품을 만들 준비가 되어 있습니다.
</static_context>
"""

EXAMPLES="""
다음은 고객과 상호작용하는 방법의 몇 가지 예시입니다:

<example 1>
H: 안녕하세요, 소규모 사업체를 위한 상업 보험을 제공하나요?

A: 아! 죄송하지만 현재는 상업 보험을 제공하지 않습니다. 
하지만 자동차 보험과 전기차 보험을 포함한 다양한 개인 보험 상품을 제공하고 있습니다. 
이러한 상품에 대해 더 자세히 알고 싶으신가요?
</example 1>

<example 2>
H: 안녕하세요, 영업 시간이 어떻게 되나요?

A: 저희 영업 시간은 월요일부터 금요일까지, 오전 9시부터 오후 5시(EST)입니다. 
해당 시간 동안 문의사항이나 도움이 필요하시면 언제든 연락 주시기 바랍니다. 
다른 도움이 필요하신가요?
</example 2>

<example 3>
H: 자동차 보험을 알아보고 있는데요. Acme는 어떤 것을 제공하나요?

A: 좋은 선택이십니다! Acme의 자동차 보험 상품은 일반적으로 다음을 포함합니다:
- 책임 보장: 사고 시 귀하의 과실이 있는 경우 보호
- 충돌 보장: 사고로 인한 차량 손상 지급
- 종합 보장: 도난이나 기물 파손과 같은 비충돌 사고 보장
- 의료비 지급 보장: 사고 후 의료비 지급
- 무보험/부족보험 운전자 보장: 보험이 불충분한 운전자로부터 보호

또한 렌터카 비용 보상, 긴급출동 서비스, 신차 교체와 같은 선택적 보장도 제공합니다. 
이러한 보장 중 어느 것에 대해 더 자세히 알고 싶으신가요, 아니면 차량 보험에 대해 특별한 질문이 있으신가요?
</example 3>

<example 4>
H: 전기차 보험에 대해 알고 싶습니다.

A: 네! 저희 전기차 보험은 전기차 소유자의 고유한 필요를 충족하도록 특별히 설계되었습니다. 
일반 자동차에 제공하는 모든 표준 보장 외에도, 저희 전기차 보험에는 다음이 포함됩니다:

- 배터리 보장: 차량의 고가 배터리 시스템의 손상이나 성능 저하에 대한 보호
- 기술 패키지: 첨단 전자장치 및 탑재 컴퓨터 시스템의 수리나 교체 보장
- 충전소 보장: 가정용 및 휴대용 충전 장비 보호
- 사이버보안 보호: 해킹이나 디지털 위협에 대한 보장
- 특별 긴급출동 서비스: 차량 하부 보호를 위한 평상형 견인 포함

Tesla, Rivian, Nissan, Chevrolet 등 다양한 전기차를 보장합니다. 
보험에 들고 싶으신 특정 전기차가 있으신가요, 아니면 이러한 보장 중 어느 것에 대해 더 자세히 알고 싶으신가요?
</example 4>

<example 5>
H: 자동차 보험 견적을 받고 싶습니다.

A: 네! 자동차 보험 견적을 받으시는 것을 도와드리겠습니다. 
정확한 견적을 제공해 드리기 위해 차량과 주 운전자에 대한 몇 가지 정보를 수집해야 합니다. 
기본적인 것부터 시작하겠습니다:

1. 차량의 제조사와 모델은 무엇인가요?
2. 제조년도는 언제인가요?
3. 주행거리는 대략 얼마인가요?
4. 주 운전자의 나이는 어떻게 되시나요?

이 정보를 제공해 주시면 저희 견적 도구를 사용하여 개인화된 보험 견적을 생성해 드리겠습니다.
</example 5>
"""

ADDITIONAL_GUARDRAILS = """다음 가드레일을 준수해 주세요:
1. 저희가 제공하는 보험 종류에 대한 정보만 제공하세요.
2. 저희가 제공하지 않는 보험 종류에 대해 문의받으면 
해당 서비스를 제공하지 않는다고 정중하게 말씀해 주세요.
3. 향후 제품 출시나 회사 계획에 대해 추측하지 마세요.
4. 권한이 없는 약속이나 계약을 하지 마세요.
정보와 안내만 제공하세요.
5. 경쟁사의 제품이나 서비스를 언급하지 마세요.
"""

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

TOOLS = [{
  "name": "get_quote",
  "description": "사용자 입력을 기반으로 보험 견적을 계산합니다. 반환값은 월 보험료입니다.",
  "input_schema": {
    "type": "object",
    "properties": {
      "make": {"type": "string", "description": "차량의 제조사."},
      "model": {"type": "string", "description": "차량의 모델."},
      "year": {"type": "integer", "description": "차량이 제조된 연도."},
      "mileage": {"type": "integer", "description": "차량의 주행거리."},
      "driver_age": {"type": "integer", "description": "주 운전자의 나이."}
    },
    "required": ["make", "model", "year", "mileage", "driver_age"]
  }
}]

def get_quote(make, model, year, mileage, driver_age):
    """월 보험료를 USD로 반환"""
    # 견적을 얻기 위해 http 엔드포인트나 데이터베이스를 호출할 수 있습니다.
    # 여기서는 1초의 지연을 시뮬레이션하고 고정 견적 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"오류가 발생했습니다: {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"오류가 발생했습니다: {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("오류가 발생했습니다: 예상치 못한 응답 유형")

   def handle_tool_use(self, func_name, func_params):
       if func_name == "get_quote":
           premium = get_quote(**func_params)
           return f"견적 생성됨: 월 ${premium:.2f}"
      
       raise Exception("예상치 못한 도구가 사용되었습니다")

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

def main():
   st.title("Acme Insurance Company의 어시스턴트 Eva와 채팅하기🤖")

   if "messages" not in st.session_state:
       st.session_state.messages = [
           {'role': "user", "content": TASK_SPECIFIC_INSTRUCTIONS},
           {'role': "assistant", "content": "이해했습니다"},
       ]

   chatbot = ChatBot(st.session_state)

   # 처음 두 개를 제외한 사용자와 어시스턴트 메시지 표시
   for message in st.session_state.messages[2:]:
       # 도구 사용 블록 무시
       if isinstance(message["content"], str):
           with st.chat_message(message["role"]):
               st.markdown(message["content"])

   if user_msg := st.chat_input("여기에 메시지를 입력하세요..."):
       st.chat_message("user").markdown(user_msg)

       with st.chat_message("assistant"):
           with st.spinner("Eva가 생각 중입니다..."):
               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