Update process.py

process.py

@@ -100,7 +100,13 @@ def generate_image(final):
     img = np.zeros((100, 100, 3), dtype=np.uint8)
 
     sorted_keys = ["E_I", "S_N", "T_F", "J_P"]
-
+    personality= ""
+    for k in sorted_keys:
+        v = final[k]
+        if v < 50:
+            personality += k[0]
+        else:
+            personality += k[2]
     for idx, k in enumerate(sorted_keys):
         v = final[k]
         if idx == 0:  # 左上
@@ -115,7 +121,7 @@ def generate_image(final):
         else:  # 右下
             img[50+int(50*(1-v/100)):, 50:] = np.array([int(colors[idx][0][i:i+2], 16) for i in (1, 3, 5)])
             img[50:int(50*(1-v/100))+50, 50:] = np.array([int(colors[idx][1][i:i+2], 16) for i in (1, 3, 5)])
-
+    ax.set_title(f'Your MBTI persopnality may be {personality}', fontsize=16)
     ax.imshow(img)
     ax.axis('off')  # 不显示坐标轴