Update app.py

app.py

@@ -59,8 +59,25 @@ class DocumentRetrievalAndGeneration:
             content += self.documents[idx].page_content + "\n"
             print(self.documents[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}]
@@ -79,7 +96,7 @@ class DocumentRetrievalAndGeneration:
         print("Time elapsed:", elapsed_time)
         print("Device in use:", self.llm.device)
 
-        return generated_response
+        return generated_response,content
 
     def qa_infer_gradio(self, query):
         response = self.query_and_generate_response(query)
@@ -130,7 +147,7 @@ if __name__ == "__main__":
             allow_flagging='never',
             examples=EXAMPLES,
             cache_examples=False,
-            outputs=gr.Textbox(label="SOLUTION"),
+            outputs=[gr.Textbox(label="SOLUTION"), gr.Textbox(label="RELATED QUERIES")],
             css=css_code
         )