Upload /sim_case_sug_demo/g_h.py with huggingface_hub

sim_case_sug_demo/g_h.py

@@ -5,13 +5,13 @@ _config={
     "sug_based_list":["dispute","plaintiff"],
     "sug_pool_list":["corpus3835","2022~2023"],
     "embedder_list":["ftlf","ftrob"],
-    "based_index":0,
+    "based_index":1,
     "pool_index":1,
     "emb_index":1,
     "sug_th":20,
     "cluster_epsilon":0.67,
     "similiar_trace_back_th":0.98,
-    "back_ground_RGB":[77, 6, 39]
+    "back_ground_RGB":[217, 225, 242]
 }
 emb_dim_lst=[768,1024]
 bilstm_len_lst=[19,13]
@@ -191,7 +191,7 @@ def ansi_to_html_dis(_f,file_path,_tranpose=True):
     d2=html_hl(_f["defendant_case2"])
     dis1=html_hl(_f["dispute_case1"])
     dis2=html_hl(_f["dispute_case2"])
-    score_="\n<mark style=\"background:#ffffff;color:"+("green" if _f["ensemble_pred"]>=0.75 else "yellow" if _f["ensemble_pred"]>=0.5 else "red")+"\">"+str(_f["ensemble_pred"])+"</mark>"
+    score_="\n<mark style=\"background:#d9e1f2;color:"+("green" if _f["ensemble_pred"]>=0.75 else "yellow" if _f["ensemble_pred"]>=0.5 else "red")+"\">"+str(_f["ensemble_pred"])+"</mark>"
     #score_="<mark style=\"color:>"++"\">"+str(_f["ensemble_pred"])+"</mark>"
 
     df=pd.DataFrame(_dict)
@@ -221,7 +221,7 @@ def ansi_to_html(_f,file_path,_tranpose=True):
 
     p_point1=html_hl(_f["p_point_case1"])
     p_point2=html_hl(_f["p_point_case2"])
-    score_="\n<mark style=\"background:#ffffff;color:"+("green" if _f["ensemble_pred"]>=0.75 else "yellow" if _f["ensemble_pred"]>=0.5 else "red")+"\">"+str(_f["ensemble_pred"])+"</mark>"
+    score_="\n<mark style=\"background:#d9e1f2;color:"+("green" if _f["ensemble_pred"]>=0.75 else "yellow" if _f["ensemble_pred"]>=0.5 else "red")+"\">"+str(_f["ensemble_pred"])+"</mark>"
     #score_="<mark style=\"color:>"++"\">"+str(_f["ensemble_pred"])+"</mark>"
 
     df=pd.DataFrame(_dict)
@@ -246,7 +246,7 @@ from PIL import Image, ImageDraw, ImageFont
 ANSI_BG_COLORS = {
     Fore.BLACK: (0, 0, 0),
     Fore.RED: (255, 0, 0),
-    Fore.GREEN: (0, 255, 0),
+    Fore.GREEN: (30, 255, 30),
     Fore.YELLOW: (255, 255, 0),
     Fore.BLUE: (0, 0, 255),
     Fore.MAGENTA: (255, 0, 255),
@@ -331,6 +331,7 @@ def ansi_to_image(ansi_text, font_size=20, image_path="./test.png"):
 def suggesting_dis(the_pool,target_name,case_dict):
     global ANSI_COLORS,_th,c_th,sug_th,corpus_dict,corpus_pd_f,vec_lst,id_lst,sen_lst,corpus_clust_label,_cluster_core_dict,_embedder
     global bilstm_len,cnn_len,emb_dim,inset_th,clust_th
+
     lst_2=[_e for _e in case_dict["dispute"]][:bilstm_len]
 
     #for _e in lst2:
@@ -348,6 +349,8 @@ def suggesting_dis(the_pool,target_name,case_dict):
 
     rt_lst=[]
     for i in tqdm(the_pool):
+        if target_name==i:
+            continue
         lst_1=[_e for _e in corpus_dict[i]]
         id_lst_1=[id_lst[sen_lst.index(_e)] for _e in lst_1]
         vec_lst_1=[vec_lst[sen_lst.index(_e)] for _e in lst_1]#[_embedder.encode(_e) for _e in lst_1]
@@ -386,98 +389,107 @@ def suggesting_dis(the_pool,target_name,case_dict):
             #print(cnn_pred)
             #print(bilstm_pred)
             x_e=[[bilstm_pred[0][0],cnn_pred[0][0]]]
-            ensemble_pred=logistic(x_r,y_r,x_e)
-            temp_ot["ensemble_pred"]=float(ensemble_pred[0])
-            #print(ensemble_pred)
-
-            pre_lst_1=[[color_lst[inset.index(clst_1[_e]) % len(color_lst)],Fore.WHITE,lst_1[_e],Style.RESET_ALL] if clst_1[_e] in inset else [Style.RESET_ALL,lst_1[_e]] for _e in range(len(lst_1))]
-            pre_lst_2=[[color_lst[inset.index(clst_2[_e]) % len(color_lst)],Fore.WHITE,lst_2[_e],Style.RESET_ALL] if clst_2[_e] in inset else [Style.RESET_ALL,lst_2[_e]] for _e in range(len(lst_2))]
-            
-            vlst_1=[[vec_lst_1[_e],pre_lst_1[_e][0]] for _e in range(len(pre_lst_1)) if len(pre_lst_1[_e])==4]
-            vlst_2=[[vec_lst_2[_e],pre_lst_2[_e][0]] for _e in range(len(pre_lst_2)) if len(pre_lst_2[_e])==4]
-            
-            #print(lst_1)
-
-            plst_1=replace_all("".join(corpus_pd_f[i.replace("_",",")][0]),key_lst,sp_key,1).split(sp_key)
-            
-            dlst_1=replace_all("".join(corpus_pd_f[i.replace("_",",")][1]),key_lst,sp_key,1).split(sp_key)
-            
-            v_plst_1=[_embedder.encode(_e) for _e in plst_1]
-            
-            v_dlst_1=[_embedder.encode(_e) for _e in dlst_1]
-            
-
-            cs_p1=[max([[cos_sim(_e,_v[0]),_v[-1]] for _v in vlst_1]) for _e in v_plst_1]
-            cs_d1=[max([[cos_sim(_e,_v[0]),_v[-1]] for _v in vlst_1]) for _e in v_dlst_1]
-
-            cs_p2=[max([[cos_sim(_e,_v[0]),_v[-1]] for _v in vlst_2]) for _e in v_plst_2]
-            cs_d2=[max([[cos_sim(_e,_v[0]),_v[-1]] for _v in vlst_2]) for _e in v_dlst_2]
-            
-            pre_lst_p1=[[cs_p1[_e][-1],Fore.WHITE,plst_1[_e],Style.RESET_ALL] if cs_p1[_e][0]>_th else [Style.RESET_ALL,plst_1[_e]] for _e in range(len(cs_p1))]
-            pre_lst_d1=[[cs_d1[_e][-1],Fore.WHITE,dlst_1[_e],Style.RESET_ALL] if cs_d1[_e][0]>_th else [Style.RESET_ALL,dlst_1[_e]] for _e in range(len(cs_d1))]
-            
-            pre_lst_p2=[[cs_p2[_e][-1],Fore.WHITE,plst_2[_e],Style.RESET_ALL] if cs_p2[_e][0]>_th else [Style.RESET_ALL,plst_2[_e]] for _e in range(len(cs_p2))]
-            pre_lst_d2=[[cs_d2[_e][-1],Fore.WHITE,dlst_2[_e],Style.RESET_ALL] if cs_d2[_e][0]>_th else [Style.RESET_ALL,dlst_2[_e]] for _e in range(len(cs_d2))]
-            
-
-            #if max_dp<max([len(plst_1),len(plst_2),len(dlst_1),len(dlst_2)]):
-            #   max_dp=max([len(plst_1),len(plst_2),len(dlst_1),len(dlst_2)])
-            
-            #print(plst_1)
-            #print(plst_2)
-            #print(dlst_1)
-            #print(dlst_2)
-            draw_lst_1=["".join(_e) for _e in pre_lst_1]
-            draw_lst_2=["".join(_e) for _e in pre_lst_2]
-
-            draw_lst_p1=["".join(_e) for _e in pre_lst_p1]
-            draw_lst_p2=["".join(_e) for _e in pre_lst_p2]
-            draw_lst_d1=["".join(_e) for _e in pre_lst_d1]
-            draw_lst_d2=["".join(_e) for _e in pre_lst_d2]
-            #replace_all(temp_c,key_lst,",",0)
-
-            #print(plst_1)
-            tp_str=""
-
-            #print("---------------------")
-            #print(Fore.BLUE+str(i)+Style.RESET_ALL)
-            temp_ot["case_id"]=i
-            temp_ot["plaintiff_case1"]=[{"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[_e[1]],"content":_e[-2]} if len(_e)==4 else {"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[Fore.WHITE],"content":_e[-1]} for _e in pre_lst_p1]
-            temp_ot["defendant_case1"]=[{"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[_e[1]],"content":_e[-2]} if len(_e)==4 else {"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[Fore.WHITE],"content":_e[-1]} for _e in pre_lst_d1]
-            temp_ot["dispute_case1"]=[{"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[_e[1]],"content":_e[-2]} if len(_e)==4 else {"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[Fore.WHITE],"content":_e[-1]} for _e in pre_lst_1]
-            temp_ot["plaintiff_case2"]=[{"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[_e[1]],"content":_e[-2]} if len(_e)==4 else {"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[Fore.WHITE],"content":_e[-1]} for _e in pre_lst_p2]
-            temp_ot["defendant_case2"]=[{"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[_e[1]],"content":_e[-2]} if len(_e)==4 else {"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[Fore.WHITE],"content":_e[-1]} for _e in pre_lst_d2]
-            temp_ot["dispute_case2"]=[{"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[_e[1]],"content":_e[-2]} if len(_e)==4 else {"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[Fore.WHITE],"content":_e[-1]} for _e in pre_lst_2]
-
-            tp_str+=Fore.BLUE+str(i)+Style.RESET_ALL+"\n"
-            tp_str+=(Fore.GREEN if temp_ot["ensemble_pred"]>=0.75 else Fore.YELLOW if temp_ot["ensemble_pred"]>=0.5 else Fore.RED)+str(temp_ot["ensemble_pred"])+Style.RESET_ALL+"\n"
-            tp_str+=Fore.MAGENTA+"---plaintiff_case1---"+Style.RESET_ALL+"\n"
-            tp_str+="".join(draw_lst_p1)+Style.RESET_ALL+"\n"
-            
-            tp_str+=Fore.MAGENTA+"---defendant_case1---"+Style.RESET_ALL+"\n"
-            tp_str+="".join(draw_lst_d1)+Style.RESET_ALL+"\n"
+            if bilstm_pred[0][0]>=0.75:
+                ensemble_pred=logistic(x_r,y_r,x_e)
+                temp_ot["ensemble_pred"]=float(ensemble_pred[0])
 
-            tp_str+=Fore.MAGENTA+"---dispute_case1---"+Style.RESET_ALL+"\n"
-            tp_str+="".join(draw_lst_1)+Style.RESET_ALL+"\n"
-            ###
-            tp_str+=Fore.BLUE+"target"+Style.RESET_ALL+"\n"
-            
-            tp_str+=Fore.MAGENTA+"---plaintiff_case2---"+Style.RESET_ALL+"\n"
-            tp_str+="".join(draw_lst_p2)+Style.RESET_ALL+"\n"
-
-            tp_str+=Fore.MAGENTA+"---defendant_case2---"+Style.RESET_ALL+"\n"
-            tp_str+="".join(draw_lst_d2)+Style.RESET_ALL+"\n"
-            
-            tp_str+=Fore.MAGENTA+"---dispute_case2---"+Style.RESET_ALL+"\n"
-            tp_str+="".join(draw_lst_2)+Style.RESET_ALL+"\n"
-            
-            #tp_str+="---------------------"+"\n"
+            #print(ensemble_pred)
 
-            
-            
-            temp_ot["output"]=tp_str
-            rt_lst.append(temp_ot)
-            print(tp_str)
+                pre_lst_1=[[color_lst[inset.index(clst_1[_e]) % len(color_lst)],Fore.WHITE,lst_1[_e],Style.RESET_ALL] if clst_1[_e] in inset else [Style.RESET_ALL,lst_1[_e]] for _e in range(len(lst_1))]
+                pre_lst_2=[[color_lst[inset.index(clst_2[_e]) % len(color_lst)],Fore.WHITE,lst_2[_e],Style.RESET_ALL] if clst_2[_e] in inset else [Style.RESET_ALL,lst_2[_e]] for _e in range(len(lst_2))]
+                
+                vlst_1=[[vec_lst_1[_e],pre_lst_1[_e][0]] for _e in range(len(pre_lst_1)) if len(pre_lst_1[_e])==4]
+                vlst_2=[[vec_lst_2[_e],pre_lst_2[_e][0]] for _e in range(len(pre_lst_2)) if len(pre_lst_2[_e])==4]
+                
+                #print(lst_1)
+
+                plst_1=replace_all("".join(corpus_pd_f[i.replace("_",",")][0]),key_lst,sp_key,1).split(sp_key)
+                
+                dlst_1=replace_all("".join(corpus_pd_f[i.replace("_",",")][1]),key_lst,sp_key,1).split(sp_key)
+                
+                v_plst_1=[_embedder.encode(_e) for _e in plst_1]
+                
+                v_dlst_1=[_embedder.encode(_e) for _e in dlst_1]
+                
+
+                cs_p1=[max([[cos_sim(_e,_v[0]),_v[-1]] for _v in vlst_1]) for _e in v_plst_1]
+                cs_d1=[max([[cos_sim(_e,_v[0]),_v[-1]] for _v in vlst_1]) for _e in v_dlst_1]
+
+                cs_p2=[max([[cos_sim(_e,_v[0]),_v[-1]] for _v in vlst_2]) for _e in v_plst_2]
+                cs_d2=[max([[cos_sim(_e,_v[0]),_v[-1]] for _v in vlst_2]) for _e in v_dlst_2]
+                
+                pre_lst_p1=[[cs_p1[_e][-1],Fore.WHITE,plst_1[_e],Style.RESET_ALL] if cs_p1[_e][0]>_th else [Style.RESET_ALL,plst_1[_e]] for _e in range(len(cs_p1))]
+                pre_lst_d1=[[cs_d1[_e][-1],Fore.WHITE,dlst_1[_e],Style.RESET_ALL] if cs_d1[_e][0]>_th else [Style.RESET_ALL,dlst_1[_e]] for _e in range(len(cs_d1))]
+                
+                pre_lst_p2=[[cs_p2[_e][-1],Fore.WHITE,plst_2[_e],Style.RESET_ALL] if cs_p2[_e][0]>_th else [Style.RESET_ALL,plst_2[_e]] for _e in range(len(cs_p2))]
+                pre_lst_d2=[[cs_d2[_e][-1],Fore.WHITE,dlst_2[_e],Style.RESET_ALL] if cs_d2[_e][0]>_th else [Style.RESET_ALL,dlst_2[_e]] for _e in range(len(cs_d2))]
+                
+
+                #if max_dp<max([len(plst_1),len(plst_2),len(dlst_1),len(dlst_2)]):
+                #   max_dp=max([len(plst_1),len(plst_2),len(dlst_1),len(dlst_2)])
+                
+                #print(plst_1)
+                #print(plst_2)
+                #print(dlst_1)
+                #print(dlst_2)
+                draw_lst_1=["".join(_e) for _e in pre_lst_1]
+                draw_lst_2=["".join(_e) for _e in pre_lst_2]
+
+                draw_lst_p1=["".join(_e) for _e in pre_lst_p1]
+                draw_lst_p2=["".join(_e) for _e in pre_lst_p2]
+                draw_lst_d1=["".join(_e) for _e in pre_lst_d1]
+                draw_lst_d2=["".join(_e) for _e in pre_lst_d2]
+                #replace_all(temp_c,key_lst,",",0)
+
+                #print(plst_1)
+                tp_str=""
+
+                #print("---------------------")
+                #print(Fore.BLUE+str(i)+Style.RESET_ALL)
+                temp_ot["case_id"]=i
+                temp_ot["plaintiff_case1"]=[{"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[_e[1]],"content":_e[-2]} if len(_e)==4 else {"background_color":ANSI_COLORS[Back.WHITE],"font_color":ANSI_COLORS[Fore.BLACK],"content":_e[-1]} for _e in pre_lst_p1]
+                temp_ot["defendant_case1"]=[{"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[_e[1]],"content":_e[-2]} if len(_e)==4 else {"background_color":ANSI_COLORS[Back.WHITE],"font_color":ANSI_COLORS[Fore.BLACK],"content":_e[-1]} for _e in pre_lst_d1]
+                temp_ot["dispute_case1"]=[{"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[_e[1]],"content":_e[-2]} if len(_e)==4 else {"background_color":ANSI_COLORS[Back.WHITE],"font_color":ANSI_COLORS[Fore.BLACK],"content":_e[-1]} for _e in pre_lst_1]
+                temp_ot["plaintiff_case2"]=[{"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[_e[1]],"content":_e[-2]} if len(_e)==4 else {"background_color":ANSI_COLORS[Back.WHITE],"font_color":ANSI_COLORS[Fore.BLACK],"content":_e[-1]} for _e in pre_lst_p2]
+                temp_ot["defendant_case2"]=[{"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[_e[1]],"content":_e[-2]} if len(_e)==4 else {"background_color":ANSI_COLORS[Back.WHITE],"font_color":ANSI_COLORS[Fore.BLACK],"content":_e[-1]} for _e in pre_lst_d2]
+                temp_ot["dispute_case2"]=[{"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[_e[1]],"content":_e[-2]} if len(_e)==4 else {"background_color":ANSI_COLORS[Back.WHITE],"font_color":ANSI_COLORS[Fore.BLACK],"content":_e[-1]} for _e in pre_lst_2]
+
+                #temp_ot["plaintiff_case1"]=[{"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[_e[1]],"content":_e[-2]} if len(_e)==4 else {"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[Fore.WHITE],"content":_e[-1]} for _e in pre_lst_p1]
+                #temp_ot["defendant_case1"]=[{"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[_e[1]],"content":_e[-2]} if len(_e)==4 else {"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[Fore.WHITE],"content":_e[-1]} for _e in pre_lst_d1]
+                #temp_ot["dispute_case1"]=[{"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[_e[1]],"content":_e[-2]} if len(_e)==4 else {"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[Fore.WHITE],"content":_e[-1]} for _e in pre_lst_1]
+                #temp_ot["plaintiff_case2"]=[{"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[_e[1]],"content":_e[-2]} if len(_e)==4 else {"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[Fore.WHITE],"content":_e[-1]} for _e in pre_lst_p2]
+                #temp_ot["defendant_case2"]=[{"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[_e[1]],"content":_e[-2]} if len(_e)==4 else {"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[Fore.WHITE],"content":_e[-1]} for _e in pre_lst_d2]
+                #temp_ot["dispute_case2"]=[{"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[_e[1]],"content":_e[-2]} if len(_e)==4 else {"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[Fore.WHITE],"content":_e[-1]} for _e in pre_lst_2]
+
+                tp_str+=Fore.BLUE+str(i)+Style.RESET_ALL+"\n"
+                tp_str+=(Fore.GREEN if temp_ot["ensemble_pred"]>=0.75 else Fore.YELLOW if temp_ot["ensemble_pred"]>=0.5 else Fore.RED)+str(temp_ot["ensemble_pred"])+Style.RESET_ALL+"\n"
+                tp_str+=Fore.MAGENTA+"---plaintiff_case1---"+Style.RESET_ALL+"\n"
+                tp_str+="".join(draw_lst_p1)+Style.RESET_ALL+"\n"
+                
+                tp_str+=Fore.MAGENTA+"---defendant_case1---"+Style.RESET_ALL+"\n"
+                tp_str+="".join(draw_lst_d1)+Style.RESET_ALL+"\n"
+
+                tp_str+=Fore.MAGENTA+"---dispute_case1---"+Style.RESET_ALL+"\n"
+                tp_str+="".join(draw_lst_1)+Style.RESET_ALL+"\n"
+                ###
+                tp_str+=Fore.BLUE+"target"+Style.RESET_ALL+"\n"
+                
+                tp_str+=Fore.MAGENTA+"---plaintiff_case2---"+Style.RESET_ALL+"\n"
+                tp_str+="".join(draw_lst_p2)+Style.RESET_ALL+"\n"
+
+                tp_str+=Fore.MAGENTA+"---defendant_case2---"+Style.RESET_ALL+"\n"
+                tp_str+="".join(draw_lst_d2)+Style.RESET_ALL+"\n"
+                
+                tp_str+=Fore.MAGENTA+"---dispute_case2---"+Style.RESET_ALL+"\n"
+                tp_str+="".join(draw_lst_2)+Style.RESET_ALL+"\n"
+                
+                #tp_str+="---------------------"+"\n"
+
+                
+                
+                temp_ot["output"]=tp_str
+                rt_lst.append(temp_ot)
+                print(tp_str)
     ot=sorted(rt_lst,key=lambda x:x["ensemble_pred"],reverse=True)
     ot_lst=[i["output"] for i in ot[:sug_th]]
 
@@ -595,10 +607,16 @@ def suggesting(the_pool,target_name,case_dict):
             #print("---------------------")
             #print(Fore.BLUE+str(i)+Style.RESET_ALL)
             temp_ot["case_id"]=i
-            temp_ot["plaintiff_case1"]=[{"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[_e[1]],"content":_e[-2]} if len(_e)==4 else {"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[Fore.WHITE],"content":_e[-1]} for _e in pre_lst_p1]
-            temp_ot["p_point_case1"]=[{"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[_e[1]],"content":_e[-2]} if len(_e)==4 else {"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[Fore.WHITE],"content":_e[-1]} for _e in pre_lst_1]
-            temp_ot["plaintiff_case2"]=[{"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[_e[1]],"content":_e[-2]} if len(_e)==4 else {"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[Fore.WHITE],"content":_e[-1]} for _e in pre_lst_p2]
-            temp_ot["p_point_case2"]=[{"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[_e[1]],"content":_e[-2]} if len(_e)==4 else {"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[Fore.WHITE],"content":_e[-1]} for _e in pre_lst_2]
+
+            temp_ot["plaintiff_case1"]=[{"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[_e[1]],"content":_e[-2]} if len(_e)==4 else {"background_color":ANSI_COLORS[Back.WHITE],"font_color":ANSI_COLORS[Fore.BLACK],"content":_e[-1]} for _e in pre_lst_p1]
+            temp_ot["p_point_case1"]=[{"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[_e[1]],"content":_e[-2]} if len(_e)==4 else {"background_color":ANSI_COLORS[Back.WHITE],"font_color":ANSI_COLORS[Fore.BLACK],"content":_e[-1]} for _e in pre_lst_1]
+            temp_ot["plaintiff_case2"]=[{"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[_e[1]],"content":_e[-2]} if len(_e)==4 else {"background_color":ANSI_COLORS[Back.WHITE],"font_color":ANSI_COLORS[Fore.BLACK],"content":_e[-1]} for _e in pre_lst_p2]
+            temp_ot["p_point_case2"]=[{"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[_e[1]],"content":_e[-2]} if len(_e)==4 else {"background_color":ANSI_COLORS[Back.WHITE],"font_color":ANSI_COLORS[Fore.BLACK],"content":_e[-1]} for _e in pre_lst_2]
+
+            #temp_ot["plaintiff_case1"]=[{"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[_e[1]],"content":_e[-2]} if len(_e)==4 else {"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[Fore.WHITE],"content":_e[-1]} for _e in pre_lst_p1]
+            #temp_ot["p_point_case1"]=[{"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[_e[1]],"content":_e[-2]} if len(_e)==4 else {"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[Fore.WHITE],"content":_e[-1]} for _e in pre_lst_1]
+            #temp_ot["plaintiff_case2"]=[{"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[_e[1]],"content":_e[-2]} if len(_e)==4 else {"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[Fore.WHITE],"content":_e[-1]} for _e in pre_lst_p2]
+            #temp_ot["p_point_case2"]=[{"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[_e[1]],"content":_e[-2]} if len(_e)==4 else {"background_color":ANSI_COLORS[_e[0]],"font_color":ANSI_COLORS[Fore.WHITE],"content":_e[-1]} for _e in pre_lst_2]
 
             tp_str+=Fore.BLUE+str(i)+Style.RESET_ALL+"\n"
             tp_str+=(Fore.GREEN if temp_ot["ensemble_pred"]>=0.75 else Fore.YELLOW if temp_ot["ensemble_pred"]>=0.5 else Fore.RED)+str(temp_ot["ensemble_pred"])+Style.RESET_ALL+"\n"