Update chatbot/core.py

chatbot/core.py

@@ -7,6 +7,13 @@ from pydantic import Field
 from typing import List, Callable
 from langchain.schema import BaseRetriever, Document
 from langchain.schema import HumanMessage, AIMessage
+from datetime import datetime
+import pytz
+
+def get_current_time():
+    utc_plus_7 = pytz.timezone("Asia/Ho_Chi_Minh")  # UTC+7
+    now = datetime.now(utc_plus_7)
+    return now.strftime("%A, %Y-%m-%d %H:%M:%S")  # Example: "Saturday, 2025-03-29 14:30:00"
 
 def translate_to_english(text: str) -> str:
     """Use Gemini LLM to translate text to English with recent chat context."""
@@ -88,10 +95,10 @@ qa_chain.get_chat_history = custom_get_chat_history
 
 def get_chat_response(user_input: str) -> str:
     """Process user input and return chat response using Wikipedia retrieval."""
-    response = qa_chain(user_input)  # Pass query to retrieval-based QA chain
-
-    # Save conversation context
-    # memory.chat_memory.add_message(HumanMessage(content=user_input))
-    # memory.chat_memory.add_message(AIMessage(content=response["result"]))
+    response = qa_chain({
+        "question": user_input,
+        "current_time": get_current_time()  # Pass the current time
+    })
 
     return response["result"]
+