Update app.py

app.py

@@ -71,10 +71,11 @@ def generate(
     outputs = []
     for text in streamer:
         outputs.append(text)
-        # Mask the output here
-        masked_output = mask_with_protecto("".join(outputs))
-        # yield "".join(outputs)
-        yield masked_output
+    concatenated_outputs = " ".join(outputs)
+    #yield concatenated_outputs
+
+    # Mask the output here
+    masked_output = mask_with_protecto(concatenated_outputs)
 
     
 def mask_with_protecto(text_for_prompt):
@@ -95,7 +96,7 @@ def mask_with_protecto(text_for_prompt):
         # Parse the masked result from the API response and format it for display
         masked_result = response.json()
         final_result = json.dumps(masked_result, indent=4)
-        return_value = str(masked_result["data"][0]["token_value"])
+        return_value = str(masked_result)
         return(return_value)
     else:
         # Return an error message if the API request was not successful.