Update classifier_manager/deberta_model.py

classifier_manager/deberta_model.py

@@ -26,4 +26,33 @@ class PiiDebertaAnalyzer:
             print(f"✅ DeBERTa model '{model_name}' loaded successfully.")
             
         except Exception as e:
-            print(f"Failed to load DeBERTa model: {e}")
+            print(f"Failed to load DeBERTa model: {e}")
+
+        def scan(self, text: str):
+        if not self.model_loaded or not text:
+            return []
+
+        try:
+            results = self.pipe(text)
+            detections = []
+            
+            for entity in results:
+                # entity looks like: {'entity_group': 'NAME_STUDENT', 'score': 0.99, 'word': 'John Doe', 'start': 0, 'end': 8}
+                original_label = entity.get('entity_group', 'UNKNOWN')
+                mapped_label = self.label_mapping.get(original_label, "DEFAULT")
+                
+                # Only include known PII types
+                if mapped_label != "DEFAULT":
+                    detections.append({
+                        "text": entity['word'].strip(),
+                        "label": mapped_label,
+                        "start": entity['start'],
+                        "end": entity['end'],
+                        "source": "DeBERTa",
+                        "score": float(entity['score'])
+                    })
+            return detections
+            
+        except Exception as e:
+            print(f"DeBERTa scan error: {e}")
+            return []