Update app.py

app.py

@@ -51,7 +51,7 @@ docsearch = FAISS.load_local("", embeddings)
 embeddings_filter = EmbeddingsFilter(
     embeddings=embeddings,
     similarity_threshold=0.7,
-    k = 3,
+    k = 2,
 )
 # 압축 검색기 생성
 compression_retriever = ContextualCompressionRetriever(
@@ -86,64 +86,63 @@ def gen(x, id, customer_data):
         customer_data_list.append(customer_data)
         if x != "약관동의_동의함":
             customer_agree_list.append("No")
-            history.append('상담원:무엇을 도와드릴까요?\n')
-            bot_str = "* 현재 가입정보를 조회할 수 없습니다. 먼저 개인정보 이용 약관에 동의하셔야 원활한 상담을 진행할 수 있습니다. \n무엇을 도와드릴까요?"
+            history.append('상담원:무엇을 도와드릴까요?\n\n')
+            bot_str = "* 먼저 개인정보 이용 약관에 동의하셔야 원활한 상담을 진행할 수 있습니다. \n무엇을 도와드릴까요?"
         else:
             customer_agree_list.append("Yes")
-            history.append('상담원:무엇을 도와드릴까요?\n')
+            history.append('상담원:무엇을 도와드릴까요?\n\n')
             bot_str = f"개인정보 활용에 동의하셨습니다. 가입 보험을 조회합니다.\n\n현재 고객님께서 가입된 보험은 {customer_data}입니다.\n\n궁금하신 것이 있으신가요?"
         return bot_str
     else:
         if x == "초기화":
             if customer_agree_list[index] != "No":
                 customer_data_list[index] = customer_data
-                history[index] = '상담원:\n무엇을 도와드릴까요?\n\n'
                 bot_str = f"대화기록이 모두 초기화되었습니다.\n\n현재 고객님께서 가입된 보험은 {customer_data}입니다.\n\n궁금하신 것이 있으신가요?"
                 return bot_str
             else:
                 customer_data_list[index] = "가입정보없음"
-                history[index] = '상담원:\n무엇을 도와드릴까요?\n\n'
-                bot_str = f"대화기록이 모두 초기화되었습니다.\n\n* 현재 가입정보를 조회할 수 없습니다. 먼저 개인정보 이용 약관에 동의하셔야 원활한 상담을 진행할 수 있습니다.\n\n궁금하신 것이 있으신가요?"
+                history[index] = '상담원:무엇을 도와드릴까요?\n\n'
+                bot_str = f"대화기록이 모두 초기화되었습니다.\n\n* 먼저 개인정보 이용 약관에 동의하셔야 원활한 상담을 진행할 수 있습니다.\n\n궁금하신 것이 있으신가요?"
                 return bot_str
         elif x == "가입정보":
             if customer_agree_list[index] == "No":
-                history[index] = '상담원:\n무엇을 도와드릴까요?\n\n'
-                bot_str = f"* 현재 가입정보를 조회할 수 없습니다. 먼저 개인정보 이용 약관에 동의하셔야 원활한 상담을 진행할 수 있습니다.\n\n궁금하신 것이 있으신가요?"
+                history[index] = '상담원:무엇을 도와드릴까요?\n\n'
+                bot_str = f"* 먼저 개인정보 이용 약관에 동의하셔야 원활한 상담을 진행할 수 있습니다.\n\n궁금하신 것이 있으신가요?"
                 return bot_str
             else:
-                history[index] = '상담원:\n무엇을 도와드릴까요?\n\n'
+                history[index] = '상담원:무엇을 도와드릴까요?\n'
                 bot_str = f"현재 고객님께서 가입된 보험은 {customer_data_list[index]}입니다.\n\n궁금하신 것이 있으신가요?"
                 return bot_str
         elif x == "약관동의_동의함":
             if customer_agree_list[index] == "No":
-                history[index] = '상담원:\n무엇을 도와드릴까요?\n\n'
+                history[index] = '상담원:무엇을 도와드릴까요?\n\n'
                 customer_agree_list[index] = "Yes"
                 customer_data_list[index] = customer_data
                 bot_str = f"개인정보 활용에 동의하셨습니다. 가입 보험을 조회합니다.\n\n현재 고객님께서 가입된 보험은 {customer_data}입니다.\n\n궁금하신 것이 있으신가요?"
                 return bot_str
             else:
-                history[index] = '상담원:\n무엇을 도와드릴까요?\n\n'
+                history[index] = '상담원:무엇을 도와드릴까요?\n'
                 bot_str = f"이미 약관에 동의하셨습니다.\n\n궁금하신 것이 있으신가요?"
                 return bot_str
         elif x == "약관동의_동의안함":
             if customer_agree_list[index] == "Yes":
-                history[index] = '상담원:\n무엇을 도와드릴까요?\n\n'
+                history[index] = '상담원:무엇을 도와드릴까요?\n\n'
                 customer_agree_list[index] = "No"
                 customer_data_list[index] = "가입정보없음"
                 bot_str = f"* 개인정보 활용 동의를 취소하셨습니다. ���제 가입 보험을 조회할 수 없습니다.\n\n궁금하신 것이 있으신가요?"
                 return bot_str
             else:
-                history[index] = '상담원:\n무엇을 도와드릴까요?\n\n'
+                history[index] = '상담원:무엇을 도와드릴까요?\n\n'
                 bot_str = f"* 개인정보 활용을 거절하셨습니다. 가입 보험을 조회할 수 없습니다. \n\n궁금하신 것이 있으신가요?"
                 return bot_str
         else:
             context = "{context}"
             question = "{question}"
             if customer_agree_list[index] == "No":
-                customer_data_newline = "현재 가입정보를 조회할 수 없습니다. 약관 동의가 필요하다고 안내해주세요."
+                customer_data_newline = "현재 가입정보를 조회할 수 없습니다. 가입할 수 있는 관련 보험을 위의 목록에서 소개해주세요."
             else:
                 customer_data_newline = customer_data_list[index].replace(",","\n")
-            prompt_template = f"""당신은 보험 상담원입니다. 아래에 질문과 관련된 약관 정보, 응답 지침과 고객의 보험 가입 정보, 고객과의 상담기록이 주어집니다. 대화기록을 보고 적절한 답변을 작성하세요.
+            prompt_template = f"""당신은 보험 상담원입니다. 아래에 전체 보험 목록, 질문과 관련된 약관 정보, 고객의 보험 가입 정보, 고객과의 상담기록이 주어집니다. 요청을 적절히 완료하는 응답을 작성하세요. 완성된 문장으로 간결히 답하세요.
 
 [전체 보험 목록]
 라이프플래닛정기보험Ⅱ
@@ -164,32 +163,24 @@ def gen(x, id, customer_data):
 1년부터저축보험
 라이프플래닛연금보험Ⅱ
 
-{context}
-
-### 명령어:
-다음 지침을 참고하여 상담원으로서 고객에게 필요한 응답을 최대한 길고 자세하게 제공하세요.
+고객은 보험 목록과 약관을 볼 수 없습니다. 직접 제시하여 소개하세요.
 
-[지침]
-1.반드시 길고 자세하고 친절하게 정보를 제공하세요.
-2.고객의 가입 정보를 꼭 확인하여 고객이 가입한 보험에 대한 내용만 제공하세요.
-3.고객이 가입한 보험이라면 고객의 질문에 대해 적절히 답변하세요.
-4.고객이 가입하지 않은 보험의 보상에 관한 질문은 관련 보험을 소개하며 보상이 불가능하다는 점을 안내하세요.
-5.고객이 가입하지 않은 보험은 가입이 필요하다고 보험명을 확실하게 언급하세요.
-다음 입력에 주어지는 고객의 보험 가입 정보와 상담 기록을 보고 고객에게 도움되는 정보를 제공하세요. 차근차근 생각하여 답변하세요. 당신은 잘 할 수 있습니다.
+{context}
 
-### 입력:
 [고객의 가입 정보]
 {customer_data_newline}
 
-[상담 기록]
-{history[index]}
+### 명령어:
+주어지는 이전 대화를 보고 맥락을 파악하여 상담원으로서 고객에게 필요한 정보를 최대한 길고 자세하고 친절하게 제공하세요. 일반적인 보험 관련 지식은 해당 내용만 간결히 답변하세요.
 
-고객:
+### 질문:
 {question}
 
-상담원:
-
+### 입력:
+[이전 대화]
+{history[index]}
 
+### 응답:
 """
 
             # RetrievalQA 클래스의 from_chain_type이라는 클래스 메서드를 호출하여 질의응답 객체를 생성
@@ -207,14 +198,14 @@ def gen(x, id, customer_data):
             if customer_agree_list[index] == "No":
                 query=f"{x}"
             else:
-                query=f"{customer_data_list[index]}, {x}"
+                query=f"{x}"
             response = qa({"query":query})
-            output_str = response['result'].rsplit(".")[0] + "."
+            output_str = response['result'].rsplit(".",1)[0] + "."
             if output_str.split(":")[0]=="상담원":
                 output_str = output_str.split(":")[1]
-            history[index] += f"고객:\n{x}\n\n상담원:\n{output_str}\n\n"
+            history[index] += f"고객:{x}\n\n상담원:{output_str}\n\n"
             if customer_agree_list[index] == "No":
-                output_str = f"* 현재 가입정보를 조회할 수 없습니다. 먼저 개인정보 이용 약관에 동의하셔야 원활한 상담을 진행할 수 있습니다." + output_str
+                output_str = f"* 먼저 개인정보 이용 약관에 동의하셔야 원활한 상담을 진행할 수 있습니다.\n\n" + output_str
             return output_str
 def reset_textbox():
     return gr.update(value='')