Update rag.py

rag.py

@@ -63,19 +63,37 @@ def manage_unmatched_queries(query: str):
 
 def query_groq_llm(prompt):
     try:
-        # Temperature 0.7 rakha hai taake har baar response rephrase ho kar aaye
-        chat_completion = groq_client.chat.completions.create(
-            messages=[{"role": "user", "content": prompt}],
-            model="llama3-70b-8192",
-            temperature=0.7,
-            max_tokens=800
+        # Using the updated Llama 3.3 70B model with your specific parameters
+        completion = groq_client.chat.completions.create(
+            model="llama-3.3-70b-versatile",
+            messages=[
+                {
+                    "role": "system", 
+                    "content": "You are the official UOE AI Assistant. Provide professional, structured responses."
+                },
+                {
+                    "role": "user", 
+                    "content": prompt
+                }
+            ],
+            temperature=1,
+            max_completion_tokens=1024,
+            top_p=1,
+            stream=True,  # As requested
+            stop=None
         )
-        return chat_completion.choices[0].message.content.strip()
-    except Exception as e:
-        print(f"Error querying Groq API: {e}")
-        return None # None return karega agar API fail hui
 
+        full_response = ""
+        for chunk in completion:
+            content = chunk.choices[0].delta.content or ""
+            full_response += content
+            # Optional: print(content, end="") # Keeps the terminal streaming effect
+            
+        return full_response.strip()
 
+    except Exception as e:
+        print(f"Groq API Error: {e}")
+        return None
 
 def get_best_answer(user_input):
     if not user_input.strip():
@@ -84,13 +102,12 @@ def get_best_answer(user_input):
     user_input_lower = user_input.lower().strip()
 
     if len(user_input_lower.split()) < 3 and not any(greet in user_input_lower for greet in GREETINGS):
-        return "Please ask your question properly with at least 3 words."
+        return "Please provide more details. I need at least 3 words to understand your query properly."
 
-    # Fee Check
-    if any(keyword in user_input_lower for keyword in ["fee structure", "fees structure", "semester fees", "semester fee"]):
+    if any(kw in user_input_lower for kw in ["fee structure", "fees structure", "semester fees", "semester fee"]):
         return (
-            "💰 **Fee Structure Information**\n\n"
-            "University of Education Lahore ki up-to-date fee maloomat ke liye niche diye gaye official link par click karen:\n"
+            "💰 **Official Fee Structure**\n\n"
+            "For the most accurate and up-to-date fee details, please visit the official University of Education link:\n"
             "🔗 https://ue.edu.pk/allfeestructure.php"
         )
 
@@ -101,44 +118,22 @@ def get_best_answer(user_input):
     best_score = similarities[best_match_idx].item()
 
     if best_score >= 0.65:
-        # PATH 1: Dataset Match
         original_answer = dataset_answers[best_match_idx]
-        prompt = f"""You are the official UOE AI Assistant. Rephrase the following verified answer into a professional and attractive format. 
-        Use headings and bullet points. Do not add external facts.
+        prompt = f"""You are the official UOE AI Assistant. 
+        Rephrase and improve the following verified university answer to be professional and attractive.
+        Use bullet points and bold text where necessary.
         
         Question: {user_input}
         Verified Answer: {original_answer}"""
     else:
-        # PATH 2: No Dataset Match - LLM Knowledge + Precise Instruction
         manage_unmatched_queries(user_input)
-        
-        prompt = f"""You are the UOE AI Assistant for University of Education (UE) Lahore. 
+        prompt = f"""You are the UOE AI Assistant for University of Education Lahore. 
         The user asked: "{user_input}".
         
-        Task:
-        1. Answer the question using your knowledge about University of Education Lahore.
-        2. At the end, add this exact notice: 
-           "📢 *Note: Aapki ye query hamari support team ko forward kar di gayi hai kyunke hamare pas abhi users ki queries zyada hain. Support team isay jald verified database mein shamil kar degi taake next time aapko mazeed behtar jawab mil sakay.*"
-        3. Provide official contact details:
-           🌐 Website: https://ue.edu.pk
-           📞 Phone: +92-42-99262231-33
-           ✉️ Email: info@ue.edu.pk
-        
-        Format the response with professional headings and bold text."""
+        Instructions:
+        1. Answer using your knowledge about this university.
+        2. MUST add this exact notice at the end:
+           "📢 *Note: Your query has been forwarded to our support team. We are currently updating our verified database to include this information soon.*"
+        3. Official Contacts: Website: https://ue.edu.pk | Phone: +92-42-99262231-33"""
 
-    llm_response = query_groq_llm(prompt)
-
-    # Agar Groq ne jawab diya to wo dikhao
-    if llm_response:
-        return llm_response
-    
-    # Bilkul aakhri fallback agar Groq API down ho
-    if best_score >= 0.65:
-        return f"Verified Answer: {dataset_answers[best_match_idx]}"
-    else:
-        return (
-            "I'm sorry, I'm having trouble connecting to my brain right now. 😅\n\n"
-            "Lekin maine aapki query support team ko bhej di hai. Official maloomat ke liye:\n"
-            "📞 +92-42-99262231-33\n"
-            "✉️ info@ue.edu.pk"
-        )
+    return query_groq_llm(prompt)