Update gdpr_filter.py

gdpr_filter.py

@@ -1,7 +1,5 @@
-
 import re
 import logging
-from config import Config
 
 logger = logging.getLogger(__name__)
 
@@ -31,6 +29,13 @@ class GDPRFilter:
         r'(ol|in|stat|pril|zol|am|um|ase|mycin|axin|cillin)$',
     ]
     
+    MEDICATION_EXCLUSIONS = [
+        'Metoprolol', 'Atorvastatin', 'Losartan', 'Omeprazol',
+        'Lisinopril', 'Simvastatin', 'Ramipril', 'Bisoprolol',
+        'Warfarin', 'Insulin', 'Aspirin', 'Paracetamol', 'Ibuprofen',
+        'Amoxicillin', 'Penicillin', 'Tetracyclin', 'Doxycyclin',
+    ]
+    
     @staticmethod
     def has_medication_context(text: str) -> bool:
         for pattern in GDPRFilter.MEDICATION_CONTEXT_KEYWORDS:
@@ -70,13 +75,14 @@ class GDPRFilter:
         text = re.sub(r'(\[PERSONNR\])\s+\d{1,4}\b', r'\1', text)
         
         has_med_context = GDPRFilter.has_medication_context(text)
-        exclusions = Config.GDPR_EXCLUSION_LIST
         
         def replace_name(match):
             word = match.group(0)
-            if any(word.startswith(ex) for ex in exclusions):
-                return word
-            if GDPRFilter.is_likely_medication(word) and has_med_context:
+            if GDPRFilter.is_likely_medication(word):
+                if has_med_context:
+                    return word
+                return '[LÄKEMEDEL]'
+            if any(word.startswith(ex) for ex in GDPRFilter.MEDICATION_EXCLUSIONS):
                 return word
             return '[NAMN]'
         
@@ -113,4 +119,4 @@ if __name__ == "__main__":
         print(f"\nINPUT:  {text}")
         print(f"OUTPUT: {result}")
     
-    print("\n" + "="*80)
+    print("\n" + "="*80)