fixv3

app.py

@@ -296,18 +296,7 @@ class VectorData():
         self.retriever = self.vectorstore.as_retriever()
         self.ingested_files = []
         
-        # Rest of your initialization code remains the same
-        self.prompt = ChatPromptTemplate.from_messages(
-            [
-                (
-                    "system",
-                    """ਦਿੱਤੇ ਗਏ ਸੰਦਰਭ ਦੇ ਆਧਾਰ 'ਤੇ ਸਵਾਲ ਦਾ ਜਵਾਬ ਦਿਓ। ਜੇਕਰ ਸਵਾਲ ਦਾ ਪ੍ਰਸੰਗ ਵੈਧ ਨਹੀਂ ਹੈ ਤਾਂ ਕੋਈ ਜਵਾਬ ਨਾ ਦਿਓ। ਹਮੇਸ਼ਾ ਜਵਾਬ ਦੇ ਅੰਤ ਵਿੱਚ ਸੰਦਰਭ ਦਾ ਸਰੋਤ ਦਿਓ: 
-                    {context}
-                    """,
-                ),
-                ("human", "{question}"),
-            ]
-        )
+        self.prompt = ChatPromptTemplate.from_messages([...])  # Your existing prompt
         self.llm = ChatMistralAI(model="mistral-large-latest")
         self.rag_chain = (
             {"context": self.retriever, "question": RunnablePassthrough()}