trying out another approach of chatbot

app.py

@@ -80,29 +80,61 @@ qa_chain = RetrievalQA.from_chain_type(
 avatar_img = "images/UH.png"        # Avatar shown beside bot messages
 logo = "images/UH Crest.png"        # Crest image
 
-# === Chat Logic with Course Memory ===
+# # === Chat Logic with Course Memory ===
+# def chat_with_bot(message, history, course_state):
+#     lower_msg = message.lower()
+
+#     if "msc" in lower_msg or "course:" in lower_msg:
+#         course_state = message.strip()
+#         response = f"✅ Got it! You're now asking about: **{course_state}**"
+#     elif "change course to" in lower_msg:
+#         course_state = message.replace("change course to", "").strip()
+#         response = f"🔁 Course changed. Now answering based on: **{course_state}**"
+#     elif course_state:
+#         full_query = f"For the course '{course_state}': {message}"
+#         raw_output = qa_chain.run(full_query)
+#         response = raw_output.split("Answer:")[-1].strip()
+
+#         # Clean LLM artifacts
+#         response = response.replace("<|assistant|>", "").strip()
+#     else:
+#         response = "⚠️ Please first mention the course you're asking about (e.g., 'MSc Data Science')."
+
+#     history.append((message, response))
+#     return "", history, course_state
+
+
+
 def chat_with_bot(message, history, course_state):
     lower_msg = message.lower()
 
-    if "msc" in lower_msg or "course:" in lower_msg:
-        course_state = message.strip()
-        response = f"✅ Got it! You're now asking about: **{course_state}**"
+    # Try to detect course from first question
+    if "msc" in lower_msg:
+        course_state = message.strip()  # Store it for later use
+        full_query = f"For the course '{course_state}': {message}"
     elif "change course to" in lower_msg:
         course_state = message.replace("change course to", "").strip()
         response = f"🔁 Course changed. Now answering based on: **{course_state}**"
+        history.append((message, response))
+        return "", history, course_state
     elif course_state:
         full_query = f"For the course '{course_state}': {message}"
+    else:
+        full_query = message  # No course memory yet
+
+    try:
         raw_output = qa_chain.run(full_query)
         response = raw_output.split("Answer:")[-1].strip()
-
-        # Clean LLM artifacts
         response = response.replace("<|assistant|>", "").strip()
-    else:
-        response = "⚠️ Please first mention the course you're asking about (e.g., 'MSc Data Science')."
+    except Exception as e:
+        response = f"⚠️ An error occurred: {str(e)}"
 
     history.append((message, response))
     return "", history, course_state
 
+
+
+
 # === Build Gradio UI ===
 initial_message = (
     "👋 Welcome! I'm your Assistant for the University of Hertfordshire.\n"