Update func.py

func.py

@@ -58,21 +58,27 @@ def fetch_news(ticker):
     except Exception:
         return []
 
-def analyze_sentiment(text):
+def analyze_sentiment(text, pipe=None):
+    """
+    兼容两种调用：
+      - analyze_sentiment(text)                   -> 使用全局 sentiment_pipeline
+      - analyze_sentiment(text, some_pipeline)    -> 使用传入的 some_pipeline
+    """
     try:
-        result = sentiment_pipeline(text)[0]
+        sentiment_pipe = pipe or sentiment_pipeline
+        result = sentiment_pipe(text)[0]
         return "Positive" if result['label'] == 'POSITIVE' else "Negative"
     except Exception:
         return "Unknown"
 
-def extract_org_entities(text, ner_pipe=None):
+def extract_org_entities(text, pipe=None):
     """
-      - extract_org_entities(text)                  
+      - extract_org_entities(text)                 
       - extract_org_entities(text, some_pipeline)  
     """
     try:
-        pipe = ner_pipe or ner_pipeline
-        entities = pipe(text)
+        ner_pipe = pipe or ner_pipeline
+        entities = ner_pipe(text)
         orgs = []
         for ent in entities:
             if ent["entity_group"] == "ORG":