Update app.py

app.py

@@ -41,8 +41,12 @@ def answer_question(question):
     try:
         docs = vectorstore.similarity_search(question, k=3)
         context = "\n\n".join([doc.page_content for doc in docs])
-        prompt = f"<|system|>\nAnswer based on context only.\n</s>\n<|user|>\nContext:\n{context}\n\nQuestion: {question}\n</s>\n<|assistant|>\n"
-        response = client.text_generation(prompt, max_new_tokens=512, temperature=0.7)
+        messages = [
+            {"role": "system", "content": "Answer based on the provided context only. If the answer is not in the context, say so."},
+            {"role": "user", "content": f"Context:\n{context}\n\nQuestion: {question}"}
+        ]
+        response = client.chat_completion(messages, max_tokens=512, temperature=0.7)
+        response = response.choices[0].message.content
         sources = [f"{i}. Page {doc.metadata.get('page', 'N/A')}" for i, doc in enumerate(docs, 1)]
         return response, "\n".join(sources)
     except Exception as e: