update

app.py

@@ -14,6 +14,7 @@ import torch
 import time
 from transformers import AutoTokenizer, AutoModel
 from datetime import datetime, timedelta
+from dateparser.search import search_dates
 # pd.options.mode.chained_assignment = None  # default='warn'
 
 # ===========================
@@ -234,16 +235,21 @@ def chatbot_mechanism(message, history, additional_input_1):
     x_text = str(df.loc[x_index, 'Tên chỉ số'])
     y_text = str(df.columns[y_index])
 
+    # Not related but extract datetime in the question if any
+    extracted_date = search_dates(question)[0][1].strftime('%d/%m/%Y')
+
     # Small adjustment for better print
     if y_text.count('/') == 4:
         y_text = y_text[-10:] # If y_text is preprocessed datetime format, trim it
 
     # Just add some text to warn users
-    eval_text = ""
+    eval_text_1 = ""
+    eval_text_2 = ""
     eval_text_sub_title = ""
-    if x_score <= 0.87 or y_score <= 0.87:
+    if x_score <= 0.865 or y_score <= 0.865:
         eval_text_sub_title = "Cảnh báo:"
-        eval_text = "⚠️"
+        eval_text_1 = "⚠️ Độ tương quan thấp ⚠️"
+        eval_text_2 = "opacity: 0.5;"
 
     # Score display
     x_score_display = str(round((x_score - 0.8) / (1.0 - 0.8) * 100, 1))
@@ -254,28 +260,34 @@ def chatbot_mechanism(message, history, additional_input_1):
     
     # Final print
     final_output_message = f"\
+        <div style='{eval_text_2}'>\
+            <div style='color: gray; font-size: 80%; font-family: courier, monospace; margin-top: 6px;'>\
+                Đặc trưng trích xuất được (có vector trong dữ liệu):\
+            </div>\
+            • {x_text}<br>\
+            • {y_text if extra_information_for_special_cases_flag == False else extra_information_for_special_cases}<br>\
+            <div style='color: gray; font-size: 80%; font-family: courier, monospace; margin-top: 6px;'>\
+                Đặc trưng trích xuất được (nội suy):\
+            </div>\
+            • {extracted_date}<br>\
+            <div style='color: gray; font-size: 80%; font-family: courier, monospace; margin-top: 6px;'>\
+                Đánh giá:\
+            </div>\
+            Độ tương quan: [x={x_score_display}%, y={y_score_display}%]<br>\
+        </div>\
         <div style='color: gray; font-size: 80%; font-family: courier, monospace; margin-top: 6px;'>\
-            Đặc trưng trích xuất được:\
+            {eval_text_sub_title}\
+        </div>\
+        <div style='color: red; font-weight: bold;'>\
+            {eval_text_1}\
         </div>\
-        • {x_text}<br>\
-        • {y_text if extra_information_for_special_cases_flag == False else extra_information_for_special_cases}<br>\
     "
-        # <div style='color: gray; font-size: 80%; font-family: courier, monospace; margin-top: 6px;'>\
-        #     Đánh giá:\
-        # </div>\
-        # Độ tương quan: [x={x_score_display}%, y={y_score_display}%]<br>\
         # <div style='color: gray; font-size: 80%; font-family: courier, monospace; margin-top: 6px;'>\
         #     Kết quả:\
         # </div>\
         # <div style='font-weight: bold;'>\
         #     {cell_value}\
         # </div>\
-        # <div style='color: gray; font-size: 80%; font-family: courier, monospace; margin-top: 6px;'>\
-        #     {eval_text_sub_title}\
-        # </div>\
-        # <div style='color: red; font-weight: bold;'>\
-        #     {eval_text}\
-        # </div>\
     return final_output_message
     # for i in range(len(final_output_message)):
     #     time.sleep(0.1)

requirements.txt

@@ -9,6 +9,7 @@ click==8.1.7
 colorama==0.4.6
 contourpy==1.2.0
 cycler==0.12.1
+dateparser==1.2.0
 et-xmlfile==1.1.0
 fastapi==0.110.0
 ffmpy==0.3.2
@@ -70,6 +71,7 @@ transformers==4.39.1
 typer==0.10.0
 typing_extensions==4.10.0
 tzdata==2024.1
+tzlocal==5.2
 urllib3==2.2.1
 uvicorn==0.29.0
 websockets==11.0.3