Upload app.py

app.py

@@ -157,26 +157,29 @@ def chatbot_mechanism(message, history, additional_input_1):
     extra_information_for_special_cases = ""
     extra_information_for_special_cases_flag = False
     unclear_cases = [
-        # Case 0: -> YEAR
-        ["năm này", "năm hiện tại", "năm nay"],
+        # Case 0: -> DAY, MONTH, YEAR
+        [
+            "ngày này" , "ngày hiện tại", "ngày hôm nay", "hôm nay", "hôm này", "ngày nay", "ngày hiện nay",
+            "bây giờ", "hiện giờ", "hiện nay", "hiện tại", "thời điểm này", "thời gian này", "lúc này", "khi này",
+        ],
         # Case 1: -> MONTH, YEAR
         ["tháng này", "tháng hiện tại", "tháng nay", "tháng bây giờ", "tháng đang diễn ra", "tháng hiện nay", "tháng hiện giờ"], 
-        # Case 2: -> DAY, MONTH, YEAR
-        ["ngày này" , "ngày hiện tại", "ngày hôm nay", "hôm nay", "bây giờ", "hiện tại", "thời điểm này", "thời gian này"],
+        # Case 2: -> YEAR
+        ["năm này", "năm hiện tại", "năm nay", "năm hiện nay"],
 
-        # Case 3: -> YEAR
-        ["năm trước", "năm ngoái", "năm qua", "năm vừa rồi", "năm đã qua"],
+        # Case 3: -> DAY, MONTH, YEAR
+        ["hôm qua", "hôm trước", "ngày qua", "ngày trước"],
         # Case 4: -> MONTH, YEAR
         ["tháng trước", "tháng qua", "tháng vừa rồi", "tháng đã qua"], 
-        # Case 5: -> DAY, MONTH, YEAR
-        ["hôm qua", "hôm trước", "ngày qua", "ngày trước"],
+        # Case 5: -> YEAR
+        ["năm trước", "năm ngoái", "năm qua", "năm vừa rồi", "năm đã qua"],
 
-        # Case 6: -> YEAR
-        ["năm sau", "năm tới", "năm tiếp theo", "năm kế tiếp", "năm sắp tới"],
+        # Case 6: -> DAY, MONTH, YEAR
+        ["ngày mai", "ngày sau", "ngày tới", "ngày tiếp theo", "ngày hôm sau", "ngày kế tiếp", "ngày sắp tới"],
         # Case 7: -> MONTH, YEAR
         ["tháng sau", "tháng tới", "tháng tiếp theo", "tháng kế tiếp", "tháng sắp tới"], 
-        # Case 8: -> DAY, MONTH, YEAR
-        ["ngày mai", "ngày sau", "ngày tới", "ngày tiếp theo", "ngày hôm sau", "ngày kế tiếp", "ngày sắp tới"],
+        # Case 8: -> YEAR
+        ["năm sau", "năm tới", "năm tiếp theo", "năm kế tiếp", "năm sắp tới"],
     ]
     for i in range(len(unclear_cases)):
         for u in range(len(unclear_cases[i])):
@@ -190,28 +193,28 @@ def chatbot_mechanism(message, history, additional_input_1):
                 if i in [0, 1, 2]:
                     target_time = current_time # No change
                 elif i == 3:
-                    target_time = current_time - timedelta(days = 365)
+                    target_time = current_time - timedelta(days = 1)
                 elif i == 4:
                     target_time = current_time - timedelta(days = 30)
                 elif i == 5:
-                    target_time = current_time - timedelta(days = 1)
+                    target_time = current_time - timedelta(days = 365)
                 elif i == 6:
-                    target_time = current_time + timedelta(days = 365)
+                    target_time = current_time + timedelta(days = 1)
                 elif i == 7:
                     target_time = current_time + timedelta(days = 30)
                 elif i == 8:
-                    target_time = current_time + timedelta(days = 1)
+                    target_time = current_time + timedelta(days = 365)
                 # Extract time to day, month, year
                 day = str(target_time.strftime('%d').lstrip(''))
                 month = str(target_time.strftime('%m').lstrip(''))
                 year = str(target_time.strftime('%Y').lstrip(''))
                 # Handle specific cases
                 if i in [0, 3, 6]:
-                    extra_information_for_special_cases = f"Năm {year}"
+                    extra_information_for_special_cases = f"Ngày {day} tháng {month} năm {year}"
                 elif i in [1, 4, 7]:
                     extra_information_for_special_cases = f"Tháng {month} năm {year}"
                 elif i in [2, 5, 8]:
-                    extra_information_for_special_cases = f"Ngày {day} tháng {month} năm {year}"
+                    extra_information_for_special_cases = f"Năm {year}"
     if extra_information_for_special_cases_flag == True:
         question = extra_information_for_special_cases + " " + question