Update func.py

func.py

@@ -65,30 +65,29 @@ def analyze_sentiment(text):
     except Exception:
         return "Unknown"
 
-def extract_org_entities(text):
+def extract_org_entities(text, ner_pipe=None):
+    """
+      - extract_org_entities(text)                  
+      - extract_org_entities(text, some_pipeline)  
+    """
     try:
-        entities = ner_pipeline(text)
-        org_entities = []
+        pipe = ner_pipe or ner_pipeline
+        entities = pipe(text)
+        orgs = []
         for ent in entities:
             if ent["entity_group"] == "ORG":
-                clean_word = ent["word"].replace("##", "").strip()
-                if clean_word.upper() not in org_entities:
-                    org_entities.append(clean_word.upper())
-                if len(org_entities) >= 5:
+                w = ent["word"].replace("##", "").strip().upper()
+                if w not in orgs:
+                    orgs.append(w)
+                if len(orgs) >= 5:
                     break
-        return org_entities
+        return orgs
     except Exception:
         return []
 
 # ----------- Helper Functions for Imports -----------
 def get_sentiment_pipeline():
-    """
-    Return the pre-initialized sentiment-analysis pipeline.
-    """
     return sentiment_pipeline
 
 def get_ner_pipeline():
-    """
-    Return the pre-initialized NER pipeline.
-    """
     return ner_pipeline