fake_data_generator.py

openai_fake_data_generator.py

@@ -0,0 +1,52 @@
+import openai
+frmo typing import List
+from presidio_analyzer import RecognizerResult
+from presidio_anonymizer import AnonymizerEngine
+
+
+def set_openai_key(openai_key:string):
+    """Set the OpenAI API key.
+    :param openai_key: the open AI key (https://help.openai.com/en/articles/4936850-where-do-i-find-my-secret-api-key)
+    """
+    openai.api_key = openai_key
+
+
+def call_completion_model(prompt:str, model:str="text-davinci-003", max_tokens:int=512) ->str:
+    """Creates a request for the OpenAI Completion service and returns the response.
+    
+    :param prompt: The prompt for the completion model
+    :param model: OpenAI model name
+    :param temperature: Model's temperature parameter
+    """
+
+    response = openai.Completion.create(
+        model=model,
+        prompt= prompt,
+        max_tokens=max_tokens
+    )
+
+    return response['choices'][0].text
+
+
+def create_prompt(anonymized_text: str) -> str:
+    """
+    Create the prompt with instructions to GPT-3.
+    
+    :param anonymized_text: Text with placeholders instead of PII values, e.g. My name is <PERSON>.
+    """
+
+    prompt = f"""
+    Your role is to create synthetic text based on de-identified text with placeholders instead of personally identifiable information.
+    Replace the placeholders (e.g. , , {{DATE}}, {{ip_address}}) with fake values.
+
+    Instructions:
+
+    Use completely random numbers, so every digit is drawn between 0 and 9.
+    Use realistic names that come from diverse genders, ethnicities and countries.
+    If there are no placeholders, return the text as is and provide an answer.
+    input: How do I change the limit on my credit card {{credit_card_number}}?
+    output: How do I change the limit on my credit card 2539 3519 2345 1555?
+    input: {anonymized_text}
+    output:
+    """
+    return prompt