Update app.py

app.py

@@ -84,9 +84,41 @@ def word_lookup(text, query, exceptions=exceptions):
     # Return the count of matching sequences
     return len(matching_sequences)
 
-# Function to find lecturer details using custom matching
-def answer_lecturer_query(query):
+def get_phone_number_response(best_match):
+    if best_match['phone_number']:
+        return f"Sure! {best_match['name']} the {best_match['course']} ({best_match['course_code']}) lecturer's phone number is {best_match['phone_number']}."
+    else:
+        return "Sorry, the phone number is not available."
+
+def get_office_response(best_match):
+    if best_match['office'] == "No longer in Babcock":
+        return f"Oops! {best_match['name']} the {best_match['course']} ({best_match['course_code']}) lecturer is {best_match['office']}."
+    elif best_match['office']:
+        return f"Sure thing! {best_match['name']} the {best_match['course']} ({best_match['course_code']}) lecturer's office is at {best_match['office']}."
+    else:
+        return "Sorry, the office location is not available."
+
+def get_basic_info_response(query, best_match):
+    if "code" in query:
+        return f"The course code for {best_match['course']} is {best_match['course_code']}"
+    else:
+        return f"{best_match['name']} is the {best_match['course']} ({best_match['course_code']}) lecturer."
+
+def get_default_response(best_match):
+    return f"{best_match['course']} has the course code: {best_match['course_code']}"
+
+
+def process_query(query, best_match):
+    if "phone number" in query or "number" in query:
+        return get_phone_number_response(best_match)
+    elif "office" in query:
+        return get_office_response(best_match)
+    elif any(word in query for word in ["lecturer", "who", "code"]):
+        return get_basic_info_response(query, best_match)
+    else:
+        return get_default_response(best_match)
 
+def answer_lecturer_query(query):
     query = query.lower()
     max_score = 0
     best_match = None
@@ -100,30 +132,15 @@ def answer_lecturer_query(query):
             max_score = score
             best_match = row
 
-    # Check if the query contains only one word
-    if len(query.split()) == 1:
-        return "I'm sorry, I need more information to assist you."
-
-    elif max_score >= 1:
-        # Process specific requests for phone number or office
-        if "phone number" in query or "number" in query:
-            if best_match['phone_number']:
-                return f"Sure! {best_match['name']} the {best_match['course']} ({best_match['course_code']}) lecturer's phone number is {best_match['phone_number']}."
+    if max_score >= 1:
+        if any(word in query for word in ["cosc", "geds", "ged"]):
+            query_course_code = next((word for word in query.split() if word.startswith("cosc") or word.startswith("geds") or word.startswith("ged")), None)
+            if query_course_code and query_course_code.upper() in best_match['course_code']:
+                return process_query(query, best_match)
             else:
-                return f"Sorry, I don't recall the phone number for that lecturer."
-        elif "office" in query:
-            if best_match['office'] == "No longer in Babcock":
-                return f"Oops! {best_match['name']} the {best_match['course']} ({best_match['course_code']}) lecturer is {best_match['office']}."
-            elif best_match['office']:
-                return f"Sure thing! {best_match['name']} the {best_match['course']} ({best_match['course_code']}) lecturer's office is at {best_match['office']}."
-            else:
-                return f"Sorry, I seem to have forgotten the office of that lecturer."
-        elif "lecturer" in query or "who" in query:
-            return f"{best_match['name']} is the {best_match['course']} ({best_match['course_code']}) lecturer."
-        elif "code" in query:
-            return f"The course code for {best_match['course']} is {best_match['course_code']}"
+                return "Sorry, I couldn't find info about the course you've mentioned."
         else:
-            return f"{best_match['course']} has the course code: {best_match['course_code']}"
+            return process_query(query, best_match)
     else:
         return answer_general_query(query)
 
@@ -132,7 +149,7 @@ def answer_doc_link_query(query):
     max_score = 0
     best_match = None
 
-    school_files = ["past questions", "pq", "pstq", "slides for"]
+    school_files = ["past questions", "pst questions", "pq", "pstq", "slides for"]
     study_smarter = ["flashcards", "study set", "study", "study app", "study link", "slides", "today", "class", "lecturer"]
 
     for index, row in doc_link_data.iterrows():
@@ -171,7 +188,7 @@ def answer_doc_link_query(query):
 def get_intent(query):
     # Define keywords or phrases associated with each intent
     lecturer_keywords = ["lecturer", "lecturer's" "phone number", "number", "office", "who", "code", "course", "name"]
-    doc_link_keywords = ["past questions", "pstq", "study materials", "flashcards", "studysmarter",
+    doc_link_keywords = ["past questions", "pstq", "pq", "pst", "study materials", "flashcards", "studysmarter",
                          "study smarter", "slides", "slide", "pdf"]
     unknown_keywords = ["email", "missed", "write"]
 
@@ -186,10 +203,6 @@ def get_intent(query):
     else:
         return "general"
 
-# Define variables to track the previous query and response
-previous_query = ""
-previous_response = ""
-
 
 def get_response(query):