Update app.py

app.py

@@ -34,9 +34,9 @@ model = HuggingFaceHub(
     task="conversational",
     model_kwargs={"temperature": 0.8, "max_length": 1000},
 )
-template = """Answer the quesiton based on the given context. 
-{context}
-Question: {question}
+template = """<s>[INST] You are a friendly study tutor chatbot that has access to a database of documents provided by the students. Use the chat history and your existing knowledge to answer the follow up question in a helpful and friendly way. Make sure your tone is that of a friendly study buddy. [/INST]
+Chat History: {context}
+Follow up question: {question}
 Helpful Answer:"""
 QA_CHAIN_PROMPT = PromptTemplate.from_template(template)
 
@@ -73,6 +73,7 @@ def greet(question, pdf_file):
     print("chat_history: ", chat_history)
     a = load_db(pdf_file, 5)
     r = a.invoke({"question": question, "chat_history": chat_history})
+    print(a.return_source_documents)
     match = re.search(r'Helpful Answer:(.*)', r['answer'])
     if match:
         helpful_answer = match.group(1).strip()