Update app.py

app.py

@@ -1,6 +1,5 @@
 from threading import Thread
 from huggingface_hub import hf_hub_download
-from llama_cpp import Llama
 import torch
 import gradio as gr
 import re
@@ -52,7 +51,7 @@ docsearch = FAISS.load_local("", embeddings)
 embeddings_filter = EmbeddingsFilter(
     embeddings=embeddings,
     similarity_threshold=0.7,
-    k = 2,
+    k = 3,
 )
 # 압축 검색기 생성
 compression_retriever = ContextualCompressionRetriever(
@@ -85,9 +84,14 @@ 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 = "* 현재 가입정보를 조회할 수 없습니다. 먼저 개인정보 이용 약관에 동의하셔야 원활한 상담을 진행할 수 있습니다. \n무엇을 도와드릴까요?"
+        if x != "약관동의_동의함":
+            customer_agree_list.append("No")
+            history.append('상담원:무엇을 도와드릴까요?\n')
+            bot_str = "* 현재 가입정보를 조회할 수 없습니다. 먼저 개인정보 이용 약관에 동의하셔야 원활한 상담을 진행할 수 있습니다. \n무엇을 도와드릴까요?"
+        else:
+            customer_agree_list.append("Yes")
+            history.append('상담원:무엇을 도와드릴까요?\n')
+            bot_str = f"개인정보 활용에 동의하셨습니다. 가입 보험을 조회합니다.\n\n현재 고객님께서 가입된 보험은 {customer_data}입니다.\n\n궁금하신 것이 있으신가요?"
         return bot_str
     else:
         if x == "초기화":
@@ -162,7 +166,8 @@ def gen(x, id, customer_data):
 {context}
 
 ### 명령어:
-다음 지침을 참고하여 상담원으로서 고객에게 필요한 응답을 제공하세요.
+다음 지침을 참고하여 상담원으로서 고객에게 필요한 응답을 최대한 자세하게 제공하세요.
+
 [지침]
 1.고객의 가입 정보를 꼭 확인하여 고객이 가입한 보험에 대한 내용만 제공하세요.
 2.고객이 가입한 보험이라면 고객의 질문에 대해 적절히 답변하세요.
@@ -199,6 +204,8 @@ def gen(x, id, customer_data):
                 query=f"{customer_data_list[index]}, {x}"
             response = qa({"query":query})
             output_str = response['result'].rsplit(".")[0] + "."
+            if output_str.split(":")[0]=="상담원":
+                output_str = output_str.split(":")[1]
             history[index] += f"고객:{x}\n상담원:{output_str}\n"
             if customer_agree_list[index] == "No":
                 output_str = f"* 현재 가입정보를 조회할 수 없습니다. 먼저 개인정보 이용 약관에 동의하셔야 원활한 상담을 진행할 수 있습니다." + output_str
@@ -228,4 +235,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)