Add anonymization app 3

app.py

@@ -7,28 +7,27 @@ generator = pipeline("text2text-generation", model="facebook/bart-large", tokeni
 # Function to construct the anonymization prompt
 def construct_prompt(input_text):
     prompt = f"""
-    You are a text anonymizer. Your task is to process a given text, identify all entity or personnal names, replace them with unique and realistic pseudonyms, and return both the anonymized text and a dictionary linking the original names to the replacement names.
+    You are an advanced text anonymizer. Your task is to process the given text, identify all personal names, company names, addresses, and other identifying information, and replace them with pseudonyms or placeholders. 
 
-    The input text will be provided, and your output should follow this JSON structure:
+    Please ensure that:
+    1. Personal names (e.g., "John Doe") are replaced with realistic pseudonyms (e.g., "Michael Brown").
+    2. Company names (e.g., "SIA Partners") are replaced with generic placeholders (e.g., "CompanyPlaceholder1").
+    3. Addresses (e.g., "14 rue Championnet") are replaced with "AddressPlaceholder".
+    4. All other identifying information like registration numbers and codes are replaced with generic placeholders (e.g., "NumberPlaceholder").
+    5. Any other part of the text should remain unchanged.
+
+    The output should strictly follow this JSON format:
     {{
-      "anonymized_text": "The text with all names replaced by pseudonyms.",
-      "name_mapping": {{
-        "OriginalName1": "ReplacementName1",
-        "OriginalName2": "ReplacementName2"
+      "anonymized_text": "The text with all identifying information replaced.",
+      "replacements": {{
+        "Original1": "Replacement1",
+        "Original2": "Replacement2"
       }}
     }}
 
-    ### Guidelines:
-    1. Only replace entity or personnal names.
-    2. Ensure the pseudonyms are realistic and appropriate for the type of name being replaced.
-    3. Keep all other parts of the text unchanged.
-    4. If no names are found, return the original text with an empty dictionary.
-
     Here is the input text:
 
     "{input_text}"
-
-    Please process the input and return your output in the specified JSON format.
     """
     return prompt.strip()