Update game1.py

game1.py

@@ -24,18 +24,18 @@ def read1(lang):
         while len(tokens) <= min_len or '\\' in text['text'] or '//' in text['text']:
             index_selected = random.randint(0,len(content)/2-1)
             text = eval(content[int(index_selected*2)])
-        # res_tmp = [(i, 0) for i in text['text'].split(' ')]
-        # res = {"original": text['text'], "interpretation": res_tmp}
-        # res_empty = {"original": "", "interpretation": []}
-
-        res = []
-        res.append(("P", "+"))
-        res.append(("/", None))
-        res.append(("N", "-"))
-        res.append(("Review:", None))
-        for i in text['text'].split(' '):
-            res.append((i, None))
-        res_empty = None
+        res_tmp = [(i, 0) for i in tokens]
+        res = {"original": text['text'], "interpretation": res_tmp}
+        res_empty = {"original": "", "interpretation": []}
+
+        # res = []
+        # res.append(("P", "+"))
+        # res.append(("/", None))
+        # res.append(("N", "-"))
+        # res.append(("Review:", None))
+        # for i in text['text'].split(' '):
+        #     res.append((i, None))
+        # res_empty = None
     # checkbox_update = gr.CheckboxGroup.update(choices=tokens, value=None)
     
     return res, lang, index_selected, res_empty, 0, 0, []
@@ -62,7 +62,7 @@ def read1_written(lang):
         # print(res)
     return text['text']
     
-def func1(lang_selected, num_selected, human_predict, num1, num2, h1):
+def func1(lang_selected, num_selected, human_predict, num1, num2, user_marks):
     chatbot = []
     # num1: Human score; num2: AI score
     if lang_selected in ['en']:
@@ -131,17 +131,17 @@ def func1(lang_selected, num_selected, human_predict, num1, num2, h1):
     # result[label] = out['score']
 
     user_select = "You focused on "
-    flag_select = False
-    for i in range(len(h1)):
-        if h1[i][1] != None and h1[i][0] not in ["P", "N"]:
-            flag_select = True
-            user_select += "'" + h1[i][0] + "'"
-            if i == len(h1) - 1:
-                user_select += ". "
-            else:
-                user_select += ", "
-    if not flag_select:
-        user_select += "nothing. Interesting! "
+    # flag_select = False
+    # for i in range(len(user_marks)):
+    #     if user_marks[i][1] != None and h1[i][0] not in ["P", "N"]:
+    #         flag_select = True
+    #         user_select += "'" + h1[i][0] + "'"
+    #         if i == len(h1) - 1:
+    #             user_select += ". "
+    #         else:
+    #             user_select += ", "
+    # if not flag_select:
+    #     user_select += "nothing. Interesting! "
     user_select += "Wanna see how the AI made the guess? Click here. ⬅️"
     if lang_selected in ['en']:
         if ai_predict == golden_label:
@@ -253,32 +253,32 @@ def interpre1(lang_selected, num_selected):
     
     print(interpretation)
 
-    # interpretation = {"original": text['text'], "interpretation": interpretation}
-    pos = []
-    neg = []
-    res = []
-    for i in interpretation:
-        if i[1] > 0:
-            pos.append(i[1])
-        elif i[1] < 0:
-            neg.append(i[1])
-        else:
-            continue
-    median_pos = np.median(pos)
-    median_neg = np.median(neg)
-
-
-    res.append(("P", "+"))
-    res.append(("/", None))
-    res.append(("N", "-"))
-    res.append(("Review:", None))
-    for i in interpretation:
-        if i[1] > median_pos:
-            res.append((i[0], "+"))
-        elif i[1] < median_neg:
-            res.append((i[0], "-"))
-        else:
-            res.append((i[0], None))
+    res = {"original": text['text'], "interpretation": interpretation}
+    # pos = []
+    # neg = []
+    # res = []
+    # for i in interpretation:
+    #     if i[1] > 0:
+    #         pos.append(i[1])
+    #     elif i[1] < 0:
+    #         neg.append(i[1])
+    #     else:
+    #         continue
+    # median_pos = np.median(pos)
+    # median_neg = np.median(neg)
+
+
+    # res.append(("P", "+"))
+    # res.append(("/", None))
+    # res.append(("N", "-"))
+    # res.append(("Review:", None))
+    # for i in interpretation:
+    #     if i[1] > median_pos:
+    #         res.append((i[0], "+"))
+    #     elif i[1] < median_neg:
+    #         res.append((i[0], "-"))
+    #     else:
+    #         res.append((i[0], None))
     return res
 
 def change_lang(choice):