feat: updated demo to handle credit card info

app.py

@@ -1,41 +1,31 @@
 import gradio as gr
 from pangea.services import Redact
-import openai
+import os
 
 def redact_input(input: str, pangea_token: str):
     redact = Redact(token=pangea_token)
     return redact.redact(text=input).result.redacted_text
 
-def gpt_completion(prompt: str, openai_api_key: str):
-    openai.api_key = openai_api_key
-    response = openai.Completion.create(model="text-davinci-003",
-                                      prompt=prompt,
-                                      temperature=0,
-                                      max_tokens=100)
-    
-    return response.choices[0].text.strip()
 
+input_text = gr.components.Textbox(label="Enter your data here", value='My credit card number is 4242424242424242 with first and last name Mark Wittner and phone number 240-555-2323 \n City: New York \n State: NY')
+output_text1 = gr.components.Textbox(label="Redacted data")
 
-input_text = gr.components.Textbox(label="Enter your GPT prompt here (User Input)", value='Rewrite and continue this email to let the specific user know the next steps for their lease, "hey user with email dummyuser@email.com (last logged in from 127.0.0.1), here\'s the lease agreement for your new property":')
-output_text1 = gr.components.Textbox(label="Redacted User Input")
-output_text2 = gr.components.Textbox(label="GPT-3 Prompt Completion (on Redacted Input)")
 
-
-def main(oai_api_key, pangea_token, input_text):
+def main(pangea_token, input_text):
+    if pangea_token == "defualt":
+        pangea_token = os.environ["PANGEA_REDACT_TOKEN"]
     redacted_text = redact_input(input_text, pangea_token)
-    completion = gpt_completion(redacted_text, oai_api_key)
-    return [redacted_text, completion]
+    return redacted_text
 
-openai_api_key_input = gr.components.Textbox(label="OpenAI API Key (Not stored)", type="password", placeholder="sk-......")
-pangea_token_input = gr.components.Textbox(label="Pangea Redact Token (Not stored)", type="password", placeholder="pts_.....")
+pangea_token_input = gr.components.Textbox(label="Pangea Redact Token (Optional)", type="password", value="defualt", placeholder="pts_.....")
 
 iface = gr.Interface(
     fn=main,
-    inputs=[openai_api_key_input, pangea_token_input, input_text],
-    outputs=[output_text1, output_text2],
-    title="GPT-3 Input Redaction Playground",
-    description="<center>Enter your OpenAI key and your Pangea redact token, then put in your prompt and watch the magic happen 🪄. <br/>\
-    Data redaction is powered by <a href=\"https://pangea.cloud/?utm_source=huggingface&utm_medium=redact-gpt-prompt-demo\">Pangea's APIs</a> in this demo. To learn more about how to get your API keys and tokens to run this, watch this short <a href=\"https://www.youtube.com/watch?v=LNN5s_6G3Cc\" target=\"_blank\">walkthrough demo</a></center>",
+    inputs=[pangea_token_input, input_text],
+    outputs=[output_text1],
+    title="Data Redaction Playground",
+    description="<center>Enter your your Pangea redact token (optional for custom settings), then put in your data and watch the magic happen 🪄. <br/>\
+    Data redaction is powered by <a href=\"https://pangea.cloud/?utm_source=huggingface&utm_medium=redact-hash-demo\">Pangea's APIs</a> in this demo. To learn more about how to get your API keys and tokens to run this, watch this short <a href=\"https://www.youtube.com/watch?v=LNN5s_6G3Cc\" target=\"_blank\">walkthrough demo</a></center>",
     allow_flagging="never"
 )
 iface.launch(

requirements.txt

@@ -1,2 +1 @@
-pangea-sdk
-openai
+pangea-sdk