Update app.py

app.py

@@ -54,6 +54,18 @@ def GRU_predict(x):
 
     return pred_proba
 
+def plot(result):
+  label = ['độc hại', 'cực kì độc hại', 'tục tĩu', 'đe dọa', 'xúc phạm', 'thù ghét cá nhân']
+  data = pd.DataFrame()
+  data['Nhãn'] = label
+  data['Điểm'] = result
+
+  print(data)
+
+  p = px.bar(data, x='Nhãn', y='Điểm', color='Nhãn', range_y=[0, 1] )
+  return p
+  pass
+
 def judge(x):
 
   label = ['độc hại', 'cực kì độc hại', 'tục tĩu', 'đe dọa', 'xúc phạm', 'thù ghét cá nhân']
@@ -63,20 +75,24 @@ def judge(x):
   lstm_pred = LSTM_predict(x)
   gru_pred = GRU_predict(x)
 
-
+  #print(result)
+  
   return_result = 'Result'
   result_lstm = np.round(lstm_pred, 2)
   result_gru = np.round(gru_pred, 2)
-  sensitive_result = max(max(result_lstm),max(result_gru))
+  for i in range(6):
+    result.append((result_lstm[i]+result_gru[i])/2)
   
-    #print(sensitive_result)
+  final_result = np.round(result, 2)
+
+  #print(final_result)
   return_result += '\nMô hình LSTM\n'
   return_result += f"{result_lstm}\n"
   for i in range(6):
     if result_lstm[i]>=0 and result_lstm[i]<0.1:
       return_result += "Tính {} là không có\n".format(label[i])
     if result_lstm[i]>=0.1 and result_lstm[i]<0.5:
-      return_result += "Tính {} ở mức nhận thấy được, từ ngữ có thể chưa phù hợp\n".format(label[i])
+      return_result += "Tính {} ở mức không rõ ràng, không thể xác định chính xác\n".format(label[i])
     if result_lstm[i]>=0.5 and result_lstm[i]<0.8:
       return_result += "Tính {} ở mức rõ ràng, cần xem xét\n".format(label[i])
     if result_lstm[i]>=0.8:
@@ -89,24 +105,39 @@ def judge(x):
     if result_gru[i]>=0 and result_gru[i]<0.1:
       return_result += "Tính {} là không có\n".format(label[i])
     if result_gru[i]>=0.1 and result_gru[i]<0.5:
-      return_result += "Tính {} ở mức nhận thấy được, từ ngữ có thể chưa phù hợp\n".format(label[i])
+      return_result += "Tính {} ở mức không rõ ràng, không thể xác định chính xác\n".format(label[i])
     if result_gru[i]>=0.5 and result_gru[i]<0.8:
       return_result += "Tính {} ở mức rõ ràng, cần xem xét\n".format(label[i])
     if result_gru[i]>=0.8:
       return_result += "Tính {} ở mức nghiêm trọng, yêu cầu chấn chỉnh\n".format(label[i])
 
-  return_result += "\nTổng quan:\n"
-  if sensitive_result>=0 and sensitive_result<0.1:
-    return_result += "Ngôn ngữ phù hợp mọi lứa tuổi.\n"
-  if sensitive_result>=0.1 and sensitive_result<0.5:
-    return_result += "Ngôn ngữ có thể còn chứa từ ngữ chưa phù hợp.\n"
-  if sensitive_result>=0.5 and sensitive_result<0.8:
-    return_result += "Ngôn ngữ không phù hợp, cần xem xét lại.\n"
-  if sensitive_result>=0.8:
-    return_result += "Ngôn ngữ vi phạm tiêu chuẩn cộng đồng nghiêm trọng, yêu cầu chấn chỉnh.\n"
 
-  return return_result
+  another_result = ''
+  another_result += "\nTổng quan kết quả trung bình:\n"
+  another_result += f"{final_result}\n"
 
+  for i in range(6):
+    if final_result[i]>=0 and final_result[i]<0.1:
+      another_result += "Tính {} là không có\n".format(label[i])
+    if final_result[i]>=0.1 and final_result[i]<0.5:
+      another_result += "Tính {} ở mức không rõ ràng, không thể xác định chính xác\n".format(label[i])
+    if final_result[i]>=0.5 and final_result[i]<0.8:
+      another_result += "Tính {} ở mức rõ ràng, cần xem xét\n".format(label[i])
+    if final_result[i]>=0.8:
+      another_result += "Tính {} ở mức nghiêm trọng, yêu cầu chấn chỉnh\n".format(label[i])
+
+  another_result += "\nKết luận:\n"
+  if max(final_result)>=0 and max(final_result)<0.1:
+    another_result += "Ngôn ngữ phù hợp mọi lứa tuổi.\n"
+  if max(final_result)>=0.1 and max(final_result)<0.5:
+    another_result += "Ngôn ngữ cần được kiểm tra lại.\n"
+  if max(final_result)>=0.5 and max(final_result)<0.8:
+    another_result += "Ngôn ngữ không phù hợp, cần xem xét lại.\n"
+  if max(final_result)>=0.8:
+    another_result += "Ngôn ngữ vi phạm tiêu chuẩn cộng đồng nghiêm trọng, yêu cầu chấn chỉnh.\n"
+
+  p = plot(final_result)
+  return (return_result, another_result, p)
 
 if __name__ == "__main__":
   #  print("Loading")
@@ -117,5 +148,5 @@ if __name__ == "__main__":
   #   judge(string)
   interface = gr.Interface(fn=judge,
                          inputs=gr.Textbox(lines=2, placeholder='Please write something', label="Input Text"),
-                        outputs='text')
+                        outputs=['text','text','plot'])
   interface.launch()