Update backend/classes/rag_application.py

backend/classes/rag_application.py

@@ -1,6 +1,7 @@
 from pydantic import BaseModel
 import json
 import time
+import re
 from promptquality import Models
 from backend.classes.embedding_model import EmbeddingModel
 from backend.classes.vector_database.milvus_vector_database import MilvusVectorDatabase
@@ -188,6 +189,7 @@ class RAGApplication:
                 # pii_flag["company"] = protect_response["metric_results"]["deutsche_bank_company_pii_0"]["value"]>0.1
                 redacted_result = self.get_redacted_result(result, pii_flag)
                 result = redacted_result.replace("<pii>", "<tag>").replace("</pii>", "</tag>")
+                redacted_result = re.sub(r'<pii>(.*?)</pii>', r'<pii>REDACTED</pii>', redacted_result)
 
             if hallucination_detection:
                 context_adherence_score = protect_response["metric_results"]["context_adherence_luna"]["value"]