Update chat.py

chat.py

@@ -4,10 +4,22 @@ from langchain_groq import ChatGroq
 from langchain.chat_models import ChatOpenAI
 from langchain_core.prompts.prompt import PromptTemplate
 from langchain_mongodb.chat_message_histories import MongoDBChatMessageHistory
+from langchain_experimental.data_anonymizer import PresidioReversibleAnonymizer
+from presidio_analyzer import AnalyzerEngine, RecognizerRegistry
+from presidio_anonymizer import AnonymizerEngine
+
 import os
 
 openai_key = os.environ['OPENAIKEY']
-
+def deanonymizer(input,anonymizer):
+  input=anonymizer.deanonymize(input)
+  map = anonymizer.deanonymizer_mapping
+  if map:
+    for k in map["PERSON"]:
+      names = k.split(" ")
+      for i in names:
+        input = input.replace(i,map["PERSON"][k])
+  return input
 template = f"""Role: You are a super friendly, enthusiastic, and empathetic female friend who chats to teenage girls.
 
 Tasks:
@@ -65,5 +77,13 @@ conversation = ConversationChain(
 
 
 def chat_conversations(query):
-    response = conversation.predict(input=query)
-    return response
+    anonymizer = PresidioReversibleAnonymizer(
+    analyzed_fields=["PERSON", "PHONE_NUMBER", "EMAIL_ADDRESS", "CREDIT_CARD"],
+    faker_seed=42,
+    )
+    anonymized_input = anonymizer.anonymize(
+        query
+    )
+    response = conversation.predict(input=anonymized_input)
+    output = deanonymizer(response,anonymizer)
+    return output