Updating difficuly

sentence.py

@@ -92,6 +92,7 @@ def updateState(var_dict, preds):
     ### Check if the current part has been guessed
     part = var_dict["parts"][var_dict["step"]]
 
+    '''
     idx_of_nothing = -1
     if ("nothing" in preds[0]):   idx_of_nothing = 0
     elif ("nothing" in preds[1]): idx_of_nothing = 1
@@ -101,8 +102,9 @@ def updateState(var_dict, preds):
     if (part["classes"][-1] == preds[0]):   idx_of_guess = 0
     elif (part["classes"][-1] == preds[1]): idx_of_guess = 1
     elif (part["classes"][-1] == preds[2]): idx_of_guess = 2
+    '''
 
-    if not var_dict["win"] and idx_of_guess > idx_of_nothing:
+    if not var_dict["win"] and part["classes"][-1] in preds:
         var_dict["step"] += 1
         var_dict["found_words"].append(part["word"])
         var_dict["win"] = var_dict["step"] == len(var_dict["parts"])