Update app.py

app.py

@@ -66,6 +66,7 @@ compression_retriever = ContextualCompressionRetriever(
 id_list = []
 history = []
 customer_data_list = []
+customer_agree_list = []
 context = "{context}"
 question = "{question}"
 
@@ -84,25 +85,60 @@ def gen(x, id, customer_data):
         index = len(id_list)
         id_list.append(id)
         customer_data_list.append(customer_data)
+        customer_agree_list.append("No")
         history.append('상담원:무엇을 도와드릴까요?\n')
-
-        bot_str = f"현재 고객님께서 가입된 보험은 {customer_data}입니다.\n\n궁금하신 것이 있으신가요?"
+        bot_str = "* 현재 가입정보를 조회할 수 없습니다. 먼저 개인정보 이용 약관에 동의하셔야 원활한 상담을 진행할 수 있습니다. \n무엇을 도와드릴까요?"
         return bot_str
     else:
         if x == "초기화":
-            customer_data_list[index] = customer_data
-            history[index] = '상담원:무엇을 도와드릴까요?\n'
-            bot_str = f"대화기록이 모두 초기화되었습니다.\n\n현재 고객님께서 가입된 보험은 {customer_data}입니다.\n\n궁금하신 것이 있으신가요?"
-            return bot_str
+            if customer_agree_list[index] != "No":
+                customer_data_list[index] = customer_data
+                bot_str = f"대화기록이 모두 초기화되었습니다.\n\n현재 고객님께서 가입된 보험은 {customer_data}입니다.\n\n궁금하신 것이 있으신가요?"
+                return bot_str
+            else:
+                customer_data_list[index] = "가입정보없음"
+                history[index] = '상담원:무엇을 도와드릴까요?\n'
+                bot_str = f"대화기록이 모두 초기화되었습니다.\n\n* 현재 가입정보를 조회할 수 없습니다. 먼저 개인정보 이용 약관에 동의하셔야 원활한 상담을 진행할 수 있습니다.\n\n궁금하신 것이 있으신가요?"
+                return bot_str
         elif x == "가입정보":
-            bot_str = f"현재 고객님께서 가입된 보험은 {customer_data_list[index]}입니다.\n\n궁금하신 것이 있으신가요?"
-            return bot_str
+            if customer_agree_list[index] == "No":
+                history[index] = '상담원:무엇을 도와드릴까요?\n'
+                bot_str = f"* 현재 가입정보를 조회할 수 없습니다. 먼저 개인정보 이용 약관에 동의하셔야 원활한 상담을 진행할 수 있습니다.\n\n궁금하신 것이 있으신가요?"
+                return bot_str
+            else:
+                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'
+                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'
+                bot_str = f"이미 약관에 동의하셨습니다.\n\n궁금하신 것이 있으신가요?"
+                return bot_str
+        elif x == "약관동의_동의안함":
+            if customer_agree_list[index] == "Yes":
+                history[index] = '상담원:무엇을 도와드릴까요?\n'
+                customer_agree_list[index] = "No"
+                customer_data_list[index] = "가입정보없음"
+                bot_str = f"* 개인정보 활용 동의를 취소하셨습니다. 이제 가입 보험을 조회할 수 없습니다.\n\n궁금하신 것이 있으신가요?"
+                return bot_str
+            else:
+                history[index] = '상담원:무엇을 도와드릴까요?\n'
+                bot_str = f"* 개인정보 활용을 거절하셨습니다. 가입 보험을 조회할 수 없습니다. \n\n궁금하신 것이 있으신가요?"
+                return bot_str
         else:
             context = "{context}"
             question = "{question}"
-            customer_data_newline = customer_data_list[index].replace(",","\n")
-
-            prompt_template = f"""당신은 보험 상담원입니다. 아래에 질문과 전체 보험 목록, 관련된 약관 정보, 응답 지침과 고객의 보험 가입 정보, 고객과의 상담기록이 주어집니다. 요청을 적절히 완료하는 응답을 작성하세요.
+            if customer_agree_list[index] == "No":
+                customer_data_newline = "현재 가입정보를 조회할 수 없습니다. 약관 동의가 필요하다�� 안내해주세요."
+            else:
+                customer_data_newline = customer_data_list[index].replace(",","\n")
+            prompt_template = f"""당신은 보험 상담원입니다. 아래에 질문과 관련된 약관 정보, 응답 지침과 고객의 보험 가입 정보, 고객과의 상담기록이 주어집니다. 요청을 적절히 완료하는 응답을 작성하세요.
 
 [보험목록]
 라이프플래닛정기보험Ⅱ
@@ -126,7 +162,7 @@ def gen(x, id, customer_data):
 {context}
 
 ### 명령어:
-다음 지침을 참고하여 상담원으로서 고객에게 필요한 응답을 간결하게 제공하세요.
+다음 지침을 참고하여 상담원으로서 고객에게 필요한 응답을 제공하세요.
 [지침]
 1.고객의 가입 정보를 꼭 확인하여 고객이 가입한 보험에 대한 내용만 제공하세요.
 2.고객이 가입한 보험이라면 고객의 질문에 대해 적절히 답변하세요.
@@ -157,11 +193,15 @@ def gen(x, id, customer_data):
                   template=prompt_template,
               )},
             )
-            query=f"나는 현재 {customer_data_list[index]}만 가입한 상황이야. {x}"
+            if customer_agree_list[index] == "No":
+                query=f"{x}"
+            else:
+                query=f"{customer_data_list[index]}, {x}"
             response = qa({"query":query})
-            output_str = response['result'].rsplit(".",1)[0]+"."
-            print(prompt_template + output_str) 
+            output_str = response['result'].rsplit(".")[0] + "."
             history[index] += f"고객:{x}\n상담원:{output_str}\n"
+            if customer_agree_list[index] == "No":
+                output_str = f"* 현재 가입정보를 조회할 수 없습니다. 먼저 개인정보 이용 약관에 동의하셔야 원활한 상담을 진행할 수 있습니다." + output_str
             return output_str
 def reset_textbox():
     return gr.update(value='')
@@ -188,4 +228,4 @@ with gr.Blocks() as demo:
                 label="customer_data"
             )
     button_submit.click(gen, [user_text, id_text, customer_data], model_output)
-    demo.queue().launch(enable_queue=True)
+    demo.queue().launch(enable_queue=True)