Update app.py

app.py

@@ -73,6 +73,7 @@ def generate(
         outputs.append(text)
         # Mask the output here
         masked_output = mask_with_protecto("".join(outputs))
+        masked_output = "I will print whatever I want to"
         # yield "".join(outputs)
         yield masked_output
 
@@ -92,14 +93,14 @@ def mask_with_protecto(text_for_prompt):
     }
     response = requests.put(mask_request_url, headers=headers, json=mask_input)
     if response.status_code == 200:
-      # 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(masked_result["data"][0]["token_value"])
+        # 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(masked_result["data"][0]["token_value"])
+        return(final_result)
     else:
-      # Print an error message if the API request was not successful.
-      print(f"Failed to get a successful response. Status Code: {response.status_code}")
-      print(response.text)
+        # Return an error message if the API request was not successful.
+        return(str(response.status_code))
 
 chat_interface = gr.ChatInterface(
     fn=generate,