Update game1.py

game1.py

@@ -77,9 +77,9 @@ def func1(lang_selected, num_selected, human_predict, num1, num2, user_important
         text = eval(content[int(num_selected*2)])
         interpretation = eval(content[int(num_selected*2+1)])
         if lang_selected in ['en']:
-            golden_label = text['label'] * 25
+            golden_label = text['label'] * 2.5
         else:
-            golden_label = text['label'] * 100
+            golden_label = text['label'] * 10
 
     '''
     # (START) API version -- quick
@@ -121,10 +121,10 @@ def func1(lang_selected, num_selected, human_predict, num1, num2, user_important
     output = classifier([text['text']])
 
     star2num = {
-        "5 stars": 100,
-        "4 stars": 75,
-        "3 stars": 50,
-        "2 stars": 25,
+        "5 stars": 10,
+        "4 stars": 7.5,
+        "3 stars": 5,
+        "2 stars": 2.5,
         "1 star": 0,
     }
     print(output)
@@ -153,72 +153,86 @@ def func1(lang_selected, num_selected, human_predict, num1, num2, user_important
     #     user_select += "nothing. Interesting! "
     user_select += "Wanna see how the AI made the guess? Click here. ⬅️"
     if lang_selected in ['en']:
+        # 0 1 2 3 4 5 6 7 8 9 10
         if ai_predict == golden_label:
-            if abs(human_predict - golden_label) < 12.5: # Both correct
+            if abs(human_predict - golden_label) < 2: # Both correct
                 golden_label = int((human_predict + ai_predict) / 2)
+                ai_predict = golden_label
                 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)
+                golden_label += int(random.randint(-1, 1) * 0.5)
+                while golden_label > 10 or golden_label < 0:
+                    golden_label += int(random.randint(-1, 1) * 0.5)
+                ai_predict = golden_label
                 chatbot.append(("The correct answer is " + str(golden_label) + ". Sorry.. AI wins in this round.", user_select))
                 num2 += 1
         else:
             if abs(human_predict - golden_label) < abs(ai_predict - golden_label):
-                if abs(human_predict - golden_label) < 12.5:
+                if abs(human_predict - golden_label) < 2:
                     golden_label = int((golden_label + human_predict) / 2)
+                    ai_predict += int(random.randint(-1, 1) * 0.5)
                     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:
+                    gold_label = int(gold_label)
+                    ai_predict = int(ai_predict)
                     chatbot.append(("The correct answer is " + str(golden_label) + ". Both wrong... Maybe next time you'll win!", user_select))
             else:
+                gold_label = int(gold_label)
+                ai_predict = int(ai_predict)
                 chatbot.append(("The correct answer is " + str(golden_label) + ". Sorry.. No one gets the correct answer. But nice try! 😉", user_select))
     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)
+        if golden_label == 10:
+            if ai_predict > 5 and human_predict > 5:
+                golden_label = int((human_predict + ai_predict)/2) + random.randint(-1, 1)
+                while golden_label > 10:
+                    golden_label = int((human_predict + ai_predict)/2) + random.randint(-1, 1)
                 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)
+            elif ai_predict > 5 and human_predict <= 5:
+                golden_label -= random.randint(0, 3)
+                ai_predict = 7 + random.randint(-1, 2)
                 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)
+            elif ai_predict <= 5 and human_predict > 5:
+                golden_label = human_predict + random.randint(-1, 1)
+                while golden_label > 10:
+                    golden_label = human_predict + random.randint(-1, 1)
+                ai_predict = int(ai_predict)
                 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:
+                gold_label = int(gold_label)
+                ai_predict = int(ai_predict)
                 chatbot.append(("The correct answer is " + str(golden_label) + ". Sorry... No one gets the correct answer. But nice try! 😉", user_select))
         else:
-            if ai_predict < 50 and human_predict < 50:
-                golden_label = int((human_predict + ai_predict)/2) + random.randint(-10, 10)
+            if ai_predict < 5 and human_predict < 5:
+                golden_label = int((human_predict + ai_predict)/2) + random.randint(-1, 1)
                 while golden_label < 0:
-                    golden_label = int((human_predict + ai_predict)/2) + random.randint(-10, 10)
+                    golden_label = int((human_predict + ai_predict)/2) + random.randint(-1, 1)
                 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)
+            elif ai_predict < 5 and human_predict >= 5:
+                golden_label += random.randint(0, 3)
+                ai_predict = 3 + random.randint(-2, 1)
                 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)
+            elif ai_predict >= 5 and human_predict < 5:
+                golden_label = human_predict + random.randint(-1, 1)
                 while golden_label < 0:
-                    golden_label = human_predict + random.randint(-4, 4)
+                    golden_label = human_predict + random.randint(-1, 1)
+                ai_predict = int(ai_predict)
                 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:
+                gold_label = int(gold_label)
+                ai_predict = int(ai_predict)
                 chatbot.append(("The correct answer is " + str(golden_label) + ". Sorry... No one gets the correct answer. But nice try! 😉", user_select))
 
     # data = pd.DataFrame(
@@ -232,8 +246,8 @@ def func1(lang_selected, num_selected, human_predict, num1, num2, user_important
     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
+    # num_tmp = max(num1, num2)
+    # y_lim_upper = (int((num_tmp + 3)/10)+1) * 10
     # figure = gr.BarPlot.update(
     #         data,
     #         x="Role",
@@ -331,10 +345,10 @@ def func1_written(text_written, human_predict, lang_written):
     output = classifier([text_written])
 
     star2num = {
-        "5 stars": 100,
-        "4 stars": 75,
-        "3 stars": 50,
-        "2 stars": 25,
+        "5 stars": 10,
+        "4 stars": 7.5,
+        "3 stars": 5,
+        "2 stars": 2.5,
         "1 star": 0,
     }
     print(output)
@@ -345,12 +359,13 @@ def func1_written(text_written, human_predict, lang_written):
     ai_predict = star2num[out['label']]
     # result[label] = out['score']
 
-    if abs(ai_predict - human_predict) <= 12.5:
+    if abs(ai_predict - human_predict) <= 2:
+        ai_predict = int(ai_predict)
         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)
+        ai_predict += int(random.randint(-1, 1))
+        while ai_predict > 10 or ai_predict < 0:
+            ai_predict += int(random.randint(-1, 1))
         chatbot.append(("AI thinks in a different way from human. 😉", "⬅️ Feel free to try another one! ⬅️"))