Update rag.py

rag.py

@@ -43,17 +43,16 @@ def query_groq_llm(prompt, model_name="llama3-70b-8192"):
         print(f"Error querying Groq API: {e}")
         return ""
 
-def get_best_answer(user_input):
+ def get_best_answer(user_input):
     user_input_lower = user_input.lower().strip()
 
     # 👉 Check if question is about fee
     if any(keyword in user_input_lower for keyword in ["fee", "fees", "charges", "semester fee"]):
-    return (
-        "💰 For complete and up-to-date fee details for this program, we recommend visiting the official University of Education fee structure page.\n"
-        "You’ll find comprehensive information regarding tuition, admission charges, and other applicable fees there.\n"
-        "🔗 https://ue.edu.pk/allfeestructure.php"
-    )
-
+        return (
+            "💰 For complete and up-to-date fee details for this program, we recommend visiting the official University of Education fee structure page.\n"
+            "You’ll find comprehensive information regarding tuition, admission charges, and other applicable fees there.\n"
+            "🔗 https://ue.edu.pk/allfeestructure.php"
+        )
 
     # 🔁 Continue with normal similarity-based logic
     user_embedding = similarity_model.encode(user_input_lower, convert_to_tensor=True)
@@ -71,13 +70,13 @@ def get_best_answer(user_input):
         prompt = f"""As an official assistant for University of Education Lahore, provide a helpful response:
         Include relevant details about university policies.
         If unsure, direct to official channels.
-        
+
         Question: {user_input}
-        
+
         Official Answer:"""
 
     llm_response = query_groq_llm(prompt)
-    
+
     if llm_response:
         for marker in ["Improved Answer:", "Official Answer:"]:
             if marker in llm_response:
@@ -90,5 +89,6 @@ def get_best_answer(user_input):
         📞 +92-42-99262231-33
         ✉️ info@ue.edu.pk
         🌐 ue.edu.pk"""
-    
+
     return response
+