Update app.py

app.py

@@ -70,19 +70,20 @@ def gemini_response(query):
     return chain.invoke({"context": full_pdf_content, "input": query}).content
 
 # Function to generate final response
-def generate_final_response(response1, response2):
+def generate_final_response(query, response1, response2):
     prompt = ChatPromptTemplate.from_template("""
     You are an AI assistant helping educators understand and implement data protection and compliance with official regulations when using AI.
     Your goal is to provide simple, practical explanation of and advice on how to meet these regulatory requirements based on the 2 given responses.
     To do so:
-    1. Analyze the following two responses. Inspect their content, and highlight differences. This MUST be done 
+    1. Read the user query
+    2. Analyze the following two responses. Inspect their content, and highlight their differences. This MUST be done 
     internally as a hidden state. 
-    2. Then, use this information to output your own response combining the best from both.
-    If the responses differ or contradict each other on important points, include that in your response.
-    Only output your own response.
+    2. Then, use this information to output your own response to the user query, synthesizing the responses all while maintaining their strengths
+    If the responses differ or contradict each other on important points, include that in your response as this could be a sign of hallucination.
+    Only output your own final response to the user query.
     """)
     chain = prompt | openai_client
-    return chain.invoke({"response1": response1, "response2": response2}).content
+    return chain.invoke({"query": query, "response1": response1, "response2": response2}).content
 
 # Function to process the query
 def process_query(user_query):