Update app.py

app.py

@@ -93,14 +93,27 @@ class DocumentRetrievalAndGeneration:
                 content += "-" * 50 + "\n"
                 content += self.all_splits[idx].page_content + "\n"
 
-            prompt = f"""
-            <s>
+            prompt = f"""<s>
+            You are a knowledgeable assistant with access to a comprehensive database. 
+            I need you to answer my question and provide related information in a specific format.
+            I have provided five relatable json files {content}, choose the most suitable chunks for answering the query
+            Here's what I need:
+            Include a final answer without additional comments, sign-offs, or extra phrases. Be direct and to the point.
+            content
             Here's my question:
-            Query: {query}
-            Solution:
-            RETURN ONLY SOLUTION. IF THERE IS NO ANSWER RELATABLE IN RETRIEVED CHUNKS, RETURN "NO SOLUTION AVAILABLE"
+            Query:{query}
+            Solution==>
+            RETURN ONLY SOLUTION . IF THEIR IS NO ANSWER RELATABLE IN RETRIEVED CHUNKS , RETURN " NO SOLUTION AVAILABLE"
+            Example1
+            Query: "How to use IPU1_0 instead of A15_0 to process NDK in TDA2x-EVM",
+            Solution: "To use IPU1_0 instead of A15_0 to process NDK in TDA2x-EVM, you need to modify the configuration file of the NDK application. Specifically, change the processor reference from 'A15_0' to 'IPU1_0'.",
+            
+            Example2
+            Query: "Can BQ25896 support I2C interface?",
+            Solution: "Yes, the BQ25896 charger supports the I2C interface for communication."
             </s>
             """
+            
             messages = [{"role": "user", "content": prompt}]
             encodeds = self.llm.tokenizer.apply_chat_template(messages, return_tensors="pt")
             model_inputs = encodeds.to(self.llm.model.device)