Update app.py

app.py

@@ -54,6 +54,7 @@ class DocumentRetrievalAndGeneration:
     def split_documents(self):
         text_splitter = RecursiveCharacterTextSplitter(chunk_size=5000, chunk_overlap=250)
         all_splits = text_splitter.split_documents(self.documents)
+        print("LEN of all_splits", all_splits)
         return all_splits
 
     def query_and_generate_response(self, query):
@@ -66,8 +67,25 @@ class DocumentRetrievalAndGeneration:
             content += self.all_splits[idx].page_content + "\n"
             print(self.all_splits[idx].page_content)
             print("############################")
+        prompt=f"""
+        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==>
+        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'.",
         
-        prompt = f"Query: {query}\nSolution: {content}\n"
+        Example2
+        Query: "Can BQ25896 support I2C interface?",
+        Solution: "Yes, the BQ25896 charger supports the I2C interface for communication.",
+        """
+        # prompt = f"Query: {query}\nSolution: {content}\n"
 
         # Encode and prepare inputs
         messages = [{"role": "user", "content": prompt}]