Update app.py

app.py

@@ -13,16 +13,23 @@ def bold_predicted_letters(input_string: str, output_string: str) -> str:
     i = j = 0
     input_string = input_string.upper()
     output_string = output_string.upper()
+    
     while i < len(input_string):
         if input_string[i:i+6] == "<MASK>":
-            result.append("**" + output_string[j] + "**")
-            i += 6
-            j += 1
+            start_index = i
+            end_index = i + 6
+            while end_index < len(input_string) and input_string[end_index:end_index+6] == "<MASK>":
+                end_index += 6
+            
+            result.append("**" + output_string[j:j+(end_index-start_index)//6] + "**")
+            i = end_index
+            j += (end_index-start_index)//6
         else:
             result.append(input_string[i])
             i += 1
             if input_string[i-1] != "<":
                 j += 1
+    
     return "".join(result)
 
 class model: