.

app.py

@@ -54,8 +54,11 @@ def predict_product(num1, num2):
     predicted_annotations_per_model = {}
     for step in range(max(MAX_PRODUCT_DIGITS_PER_MODEL.values())):  # Set a maximum limit to prevent infinite loops
         # Ground Truth
-        ground_truth_annotations = [(ground_truth_digit, None) for ground_truth_digit in ground_truth_digits_reversed[:step+1]]
-        ground_truth_annotations = ground_truth_annotations[::-1]
+        if not valid_input:
+            ground_truth_annotations = []
+        else:
+            ground_truth_annotations = [(ground_truth_digit, None) for ground_truth_digit in ground_truth_digits_reversed[:step+1]]
+            ground_truth_annotations = ground_truth_annotations[::-1]
         # Predicted
         for model_name in models:
             model = models[model_name]
@@ -101,7 +104,9 @@ def predict_product(num1, num2):
 
             for i in range(len(predicted_digits_reversed)):
                 predicted_digit = predicted_digits_reversed[i]
-                if i >= len(ground_truth_digits_reversed):
+                if not valid_input:
+                    is_correct_digit = None
+                elif i >= len(ground_truth_digits_reversed):
                     if predicted_digit == '0' and is_correct_sofar:
                         is_correct_digit = True
                     else:
@@ -114,7 +119,9 @@ def predict_product(num1, num2):
                         is_correct_digit = False
                 if not is_correct_digit:
                     is_correct_sofar = False
-                if is_correct_digit:
+                if is_correct_digit is None:
+                    predicted_annotations.append((predicted_digit, None))
+                elif is_correct_digit:
                     predicted_annotations.append((predicted_digit, "correct"))
                 else:
                     predicted_annotations.append((predicted_digit, "wrong"))