Update app.py

app.py

@@ -20,6 +20,7 @@ from unidecode import unidecode
 
 import re
 from langchain.schema import Document
+from langchain.prompts import PromptTemplate
 
 # API 클라이언트 초기화
 llm_client = InferenceClient("CohereForAI/c4ai-command-r-plus", token=os.getenv("HF_TOKEN"))
@@ -61,10 +62,11 @@ def long_text_result(file):
             # Vector DB와 관련 chain 설정
             global chat_chain
             memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
+            prompt_template = PromptTemplate(template="Based on the following document, please provide a relevant response:", input_variables=["context"])
             retriever = create_history_aware_retriever(
                 retriever=vector_db.as_retriever(),
                 llm=llm_client,
-                prompt="Based on the following document, please provide a relevant response:"
+                prompt=prompt_template
             )
             chat_chain = create_retrieval_chain(
                 retriever=retriever,
@@ -228,8 +230,3 @@ with iface:
 
 if __name__ == "__main__":
     iface.launch()
-
-
-
-
-