Update game3.py

game3.py

@@ -40,9 +40,9 @@ def func3(num_selected, human_predict, num1, num2, user_important):
         
         golden_label = (text['binary_label']^1) * 100
         if golden_label == 0:
-            golden_label = 50 * (1 - text['binary_score'])
+            golden_label = int(50 * (1 - text['binary_score']))
         else:
-            golden_label = 50 * (1 + text['binary_score'])
+            golden_label = int(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
@@ -55,8 +55,10 @@ def func3(num_selected, human_predict, num1, num2, user_important):
     
     # (END) off-the-shelf version
     
-    # ai_predict = out['label']
-    ai_predict = out['score']
+    if out['label'] == 'Female':
+        ai_predict = int(out['score'])
+    else:
+        ai_predict = 1 - int(out['score'])
     
     user_select = "You focused on "
     flag_select = False
@@ -76,28 +78,20 @@ 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 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:
-            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(golden_label - human_predict) <= 20  and abs(golden_label - ai_predict) <= 20:
+        chatbot.append(("The correct answer is " + str(golden_label) + ". Congratulations! 🎉 Both of you get the correct answer!", user_select))
+        num1 += 1
+        num2 += 1
+    elif abs(golden_label - human_predict) > 20 and abs(golden_label - ai_predict) > 20:
+        chatbot.append(("The correct answer is " + str(golden_label) + ". Sorry.. No one gets the correct answer. But nice try! 😉", user_select))
+    elif abs(golden_label - human_predict) <= 20 and abs(golden_label - ai_predict) > 20:
+        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:
-        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))
+        chatbot.append(("The correct answer is " + str(golden_label) + ". Sorry.. AI wins in this round.", user_select))
+        num2 += 1
     
     tot_scores = ''' ### <p style="text-align: center;"> Machine &ensp; ''' + str(int(num2)) + ''' &ensp; VS &ensp; ''' + str(int(num1)) + ''' &ensp; Human </p>'''
 
@@ -169,19 +163,16 @@ def func3_written(text_written, human_predict, lang_written):
     out = output[0]
     # (END) off-the-shelf version
 
+    if out['label'] == 'Female':
+        ai_predict = int(out['score'])
+    else:
+        ai_predict = 1 - int(out['score'])
     
-    # ai_predict = star2num[out['label']]
-    ai_predict = out['score']
-
-    if abs(ai_predict - human_predict) <= 12.5:
+    if abs(ai_predict - human_predict) <= 20:
         chatbot.append(("AI gives it a close score! 🎉", "⬅️ Feel free to try another one! ⬅️"))
     else:
-        ai_predict += random.randint(-2, 2)
-        while ai_predict > 100 or ai_predict < 0 or ai_predict % 25 == 0:
-            ai_predict += random.randint(-2, 2)
         chatbot.append(("AI thinks in a different way from human. 😉", "⬅️ Feel free to try another one! ⬅️"))
 
-
     import shap
 
     gender_classifier = pipeline("text-classification", model="padmajabfrl/Gender-Classification", return_all_scores=True)