Upload 2 files

app.py

@@ -12,6 +12,7 @@ import pandas as pd
 import numpy as np
 import torch
 import time
+import re
 from transformers import AutoTokenizer, AutoModel
 from datetime import datetime, timedelta
 from dateparser.search import search_dates
@@ -236,7 +237,8 @@ def chatbot_mechanism(message, history, additional_input_1):
     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')
+    extracted_date = search_dates(message)[0][1].strftime('%d/%m/%Y')
+    print_extracted_date = bool(re.search(r'\d', message))
 
     # Small adjustment for better print
     if y_text.count('/') == 4:
@@ -260,8 +262,8 @@ def chatbot_mechanism(message, history, additional_input_1):
 
 
     # Score display
-    x_score_display = str(round((x_score - 0.8) / (1.0 - 0.8) * 100, 1))
-    y_score_display = str(round((y_score - 0.8) / (1.0 - 0.8) * 100, 1))
+    x_score_display = str(round((x_score - 0.85) / (1.0 - 0.85) * 100, 1))
+    y_score_display = str(round((y_score - 0.85) / (1.0 - 0.85) * 100, 1))
 
     # Cell value
     cell_value = df.iloc[x_index, y_index]
@@ -270,26 +272,26 @@ def chatbot_mechanism(message, history, additional_input_1):
     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 (vector trong dữ liệu):\
+                Đặc trưng trích xuất được (embedding):\
             </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>\
+            • {extracted_date if print_extracted_date == True else ''}<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;'>\
+            {eval_text_sub_title}\
+        </div>\
+        <div style='color: {eval_text_color}; font-weight: bold;'>\
+            {eval_text_1}\
+        </div>\
     "
-        # <div style='color: gray; font-size: 80%; font-family: courier, monospace; margin-top: 6px;'>\
-        #     {eval_text_sub_title}\
-        # </div>\
-        # <div style='color: {eval_text_color}; font-weight: bold;'>\
-        #     {eval_text_1}\
-        # </div>\
         # <div style='color: gray; font-size: 80%; font-family: courier, monospace; margin-top: 6px;'>\
         #     Kết quả:\
         # </div>\