Update app.py

app.py

@@ -1,7 +1,8 @@
 import os
 from threading import Thread
 from typing import Iterator
-
+import requests
+import json
 import gradio as gr
 import spaces
 import torch
@@ -75,10 +76,30 @@ def generate(
         # yield "".join(outputs)
         yield masked_output
 
-
-def mask_with_protecto(msg_to_mask):
-    return(msg_to_mask)
     
+def mask_with_protecto(text_for_prompt):
+    mask_request_url = "https://trial.protecto.ai/api/vault/mask"
+    headers = {
+        "Content-Type": "application/json; charset=utf-8",
+        "Authorization": "Bearer eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3N1ZXIiOiJQcm90ZWN0byIsImV4cGlyYXRpb25fZGF0ZSI6IjIwMjMtMTEtMDQiLCJwZXJtaXNzaW9ucyI6WyJyZWFkIiwid3JpdGUiXSwidXNlcl9uYW1lIjoiZGlwYXlhbkBjb2V1c2xlYXJuaW5nLmNvbSIsImRiX25hbWUiOiJwcm90ZWN0b19jb2V1c2xlYXJuaW5nX25ydG1mYmFrIiwiaGFzaGVkX3Bhc3N3b3JkIjoiMjIyMTI2ZWNiZTlkZTRmNWJlODdiY2QyYWFlZWRlM2FmNDc5MzMxZmNhOTUxMWU0MDRiNzkxNDM1MGI4MWUyYiJ9.DeIK00NuhM51lRwWdnUXuQSBA1aBn5AQ8qM3pIeM01U"
+    }
+    mask_input = {
+    "mask": [
+        {
+            "value": 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"])
+    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)
 
 chat_interface = gr.ChatInterface(
     fn=generate,