Update app.py

app.py

@@ -13,6 +13,7 @@ from langchain.retrievers import ContextualCompressionRetriever
 from langchain.chains import RetrievalQA
 from langchain.vectorstores import FAISS
 from langchain.embeddings import HuggingFaceEmbeddings
+from langchain.prompts import PromptTemplate
 
 torch_device = "cuda" if torch.cuda.is_available() else "cpu"
 print("Running on device:", torch_device)
@@ -85,7 +86,6 @@ def gen(x, id, customer_data):
             question = "{question}"
             customer_data_newline = customer_data.replace(",","\n")
 
-            from langchain.prompts import PromptTemplate
             prompt_template = f"""당신은 보험 상담원입니다. 아래에 질문과 관련된 약관 정보, 응답 지침과 고객의 보험 가입 정보, 고객과의 상담기록이 주어집니다. 요청을 적절히 완료하는 응답을 작성하세요.
 
 {context}
@@ -126,7 +126,7 @@ def gen(x, id, customer_data):
             )
             query=f"나는 현재 {customer_data}만 가입한 상황이야. {x}"
             response = qa({"query":query})
-            output_str = response.split("###")[0].split("\u200b")[0]
+            output_str = response['result'].split("###")[0].split("\u200b")[0]
             history[index] += f"고객:{x}\n상담원:{output_str}\n"
             return output_str
 def reset_textbox():