Update app.py

app.py

@@ -78,8 +78,18 @@ def generate(
 
     # Mask the output here
     masked_output = mask_with_protecto(concatenated_outputs)
+    masked_output = format_for_html(masked_output)
     yield masked_output
 
+
+def format_for_html(text):
+    text = text.replace(" ", " ")
+    text = text.replace("\n", "")    
+    return text
+
+# Using the function
+api_output = "N7JI9bzqbf M1L4LPGkyj is the CEO of Apple Inc."
+formatted_output = format_for_html(api_output)
     
 def mask_with_protecto(text_for_prompt):
     mask_request_url = "https://trial.protecto.ai/api/vault/mask"