Update game3.py

game3.py

@@ -39,7 +39,10 @@ def func3(num_selected, human_predict, num1, num2, user_important):
         interpretation = eval(content[int(num_selected*2+1)])
         
         golden_label = (text['binary_label']^1) * 100
-
+        if golden_label == 0:
+            golden_label = 50 * (1 - text['binary_score'])
+        else:
+            golden_label = 50 * (1 + text['binary_score'])
     # (START) off-the-shelf version -- slow at the beginning
     # Load model directly
     # Use a pipeline as a high-level helper
@@ -47,13 +50,6 @@ def func3(num_selected, human_predict, num1, num2, user_important):
     classifier = pipeline("text-classification", model="padmajabfrl/Gender-Classification")
     output = classifier([text['text']])
 
-    # star2num = {
-    #     "5 stars": 100,
-    #     "4 stars": 75,
-    #     "3 stars": 50,
-    #     "2 stars": 25,
-    #     "1 star": 0,
-    # }
     print(output)
     out = output[0]
     
@@ -80,102 +76,35 @@ def func3(num_selected, human_predict, num1, num2, user_important):
     # 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:
-            if abs(human_predict - golden_label) < 12.5: # Both correct
-                golden_label = int((human_predict + ai_predict) / 2)
-                chatbot.append(("The correct answer is " + str(golden_label) + ". Congratulations! 🎉 Both of you get the correct answer!", user_select))
-                num1 += 1
-                num2 += 1
-            else:
-                golden_label += random.randint(-2, 2)
-                while golden_label > 100 or golden_label < 0 or golden_label % 25 == 0:
-                    golden_label += random.randint(-2, 2)
-                chatbot.append(("The correct answer is " + str(golden_label) + ". Sorry.. AI wins in this round.", user_select))
-                num2 += 1
+    if abs(ai_predict - golden_label) < 12.5:
+        if abs(human_predict - golden_label) < 12.5: # Both correct
+            golden_label = int((human_predict + ai_predict) / 2)
+            chatbot.append(("The correct answer is " + str(golden_label) + ". Congratulations! 🎉 Both of you get the correct answer!", user_select))
+            num1 += 1
+            num2 += 1
         else:
-            if abs(human_predict - golden_label) < abs(ai_predict - golden_label):
-                if abs(human_predict - golden_label) < 12.5:
-                    golden_label = int((golden_label + human_predict) / 2)
-                    chatbot.append(("The correct answer is " + str(golden_label) + ". Great! 🎉 You are closer to the answer and better than AI!", user_select))
-                    num1 += 1
-                else:
-                    chatbot.append(("The correct answer is " + str(golden_label) + ". Both wrong... Maybe next time you'll win!", user_select))
-            else:
-                chatbot.append(("The correct answer is " + str(golden_label) + ". Sorry.. No one gets the correct answer. But nice try! 😉", user_select))
+            golden_label += random.randint(-2, 2)
+            while golden_label > 100 or golden_label < 0 or golden_label % 25 == 0:
+                golden_label += random.randint(-2, 2)
+            chatbot.append(("The correct answer is " + str(golden_label) + ". Sorry.. AI wins in this round.", user_select))
+            num2 += 1
     else:
-        if golden_label == 100:
-            if ai_predict > 50 and human_predict > 50:
-                golden_label = int((human_predict + ai_predict)/2) + random.randint(-10, 10)
-                while golden_label > 100:
-                    golden_label = int((human_predict + ai_predict)/2) + random.randint(-10, 10)
-                ai_predict = int((golden_label + ai_predict) / 2)
-                chatbot.append(("The correct answer is " + str(golden_label) + ". Congratulations! 🎉 Both of you get the correct answer!", user_select))
-                num1 += 1
-                num2 += 1
-            elif ai_predict > 50 and human_predict <= 50:
-                golden_label -= random.randint(0, 10)
-                ai_predict = 90 + random.randint(-5, 5)
-                chatbot.append(("The correct answer is " + str(golden_label) + ". Sorry.. AI wins in this round.", user_select))
-                num2 += 1
-            elif ai_predict <= 50 and human_predict > 50:
-                golden_label = human_predict + random.randint(-4, 4)
-                while golden_label > 100:
-                    golden_label = human_predict + random.randint(-4, 4)
-                chatbot.append(("The correct answer is " + str(golden_label) + ". Great! 🎉 You are close to the answer and better than AI!", user_select))
+        if abs(human_predict - golden_label) < abs(ai_predict - golden_label):
+            if abs(human_predict - golden_label) < 12.5:
+                golden_label = int((golden_label + human_predict) / 2)
+                chatbot.append(("The correct answer is " + str(golden_label) + ". Great! 🎉 You are closer to the answer and better than AI!", user_select))
                 num1 += 1
             else:
-                chatbot.append(("The correct answer is " + str(golden_label) + ". Sorry... No one gets the correct answer. But nice try! 😉", user_select))
+                chatbot.append(("The correct answer is " + str(golden_label) + ". Both wrong... Maybe next time you'll win!", user_select))
         else:
-            if ai_predict < 50 and human_predict < 50:
-                golden_label = int((human_predict + ai_predict)/2) + random.randint(-10, 10)
-                while golden_label < 0:
-                    golden_label = int((human_predict + ai_predict)/2) + random.randint(-10, 10)
-                ai_predict = int((golden_label + ai_predict) / 2)
-                chatbot.append(("The correct answer is " + str(golden_label) + ". Congratulations! 🎉 Both of you get the correct answer!", user_select))
-                num1 += 1
-                num2 += 1
-            elif ai_predict < 50 and human_predict >= 50:
-                golden_label += random.randint(0, 10)
-                ai_predict = 10 + random.randint(-5, 5)
-                chatbot.append(("The correct answer is " + str(golden_label) + ". Sorry.. AI wins in this round.", user_select))
-                num2 += 1
-            elif ai_predict >= 50 and human_predict < 50:
-                golden_label = human_predict + random.randint(-4, 4)
-                while golden_label < 0:
-                    golden_label = human_predict + random.randint(-4, 4)
-                chatbot.append(("The correct answer is " + str(golden_label) + ". Great! 🎉 You are close to the answer and better than AI!", user_select))
-                num1 += 1
-            else:
-                chatbot.append(("The correct answer is " + str(golden_label) + ". Sorry... No one gets the correct answer. But nice try! 😉", user_select))
-
-    # data = pd.DataFrame(
-    #     {
-    #         "Role": ["AI 🤖", "HUMAN 👨👩"],
-    #         "Scores": [num2, num1],
-    #     }
-    # )
-    # scroe_human = ''' # Human: ''' + str(int(num1))
-    # scroe_robot = ''' # Robot: ''' + str(int(num2))
+            chatbot.append(("The correct answer is " + str(golden_label) + ". Sorry.. No one gets the correct answer. But nice try! 😉", user_select))
+    
     tot_scores = ''' ### <p style="text-align: center;"> Machine &ensp; ''' + str(int(num2)) + ''' &ensp; VS &ensp; ''' + str(int(num1)) + ''' &ensp; Human </p>'''
 
     
     num_tmp = max(num1, num2)
     y_lim_upper = (int((num_tmp + 3)/10)+1) * 10
-    # figure = gr.BarPlot.update(
-    #         data,
-    #         x="Role",
-    #         y="Scores",
-    #         color="Role",
-    #         vertical=False,
-    #         y_lim=[0,y_lim_upper],
-    #         color_legend_position='none', 
-    #         height=250, 
-    #         width=500,
-    #         show_label=False,
-    #         container=False,
-    #     )
-    # tooltip=["Role", "Scores"],
+
     return ai_predict, chatbot, num1, num2, tot_scores
 
 def interpre3(lang_selected, num_selected):