Update app.py

app.py

@@ -134,11 +134,14 @@ def answer_lecturer_query(query):
 
     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 "Sorry, I couldn't find info about the course you've mentioned."
+            for i, word in enumerate(query.split()):
+                if word.isdigit():
+                    # Retrieve the prefix from the previous word
+                    query_course_code = f"{query.split()[i - 1]} {word}"
+                    if query_course_code.upper() == best_match['course_code']:
+                        return process_query(query, best_match)
+                    else:
+                        return "Sorry, I couldn't find info about the course you've mentioned."
         else:
             return process_query(query, best_match)
     else: