Update func.py

func.py

@@ -7,12 +7,21 @@ import requests
 model_id = "LinkLinkWu/ISOM5240HKUSTBASE"
 sentiment_tokenizer = AutoTokenizer.from_pretrained(model_id)
 sentiment_model = AutoModelForSequenceClassification.from_pretrained(model_id)
-sentiment_pipeline = pipeline("sentiment-analysis", model=sentiment_model, tokenizer=sentiment_tokenizer)
+sentiment_pipeline = pipeline(
+    "sentiment-analysis",
+    model=sentiment_model,
+    tokenizer=sentiment_tokenizer
+)
 
 # NER pipeline
 ner_tokenizer = AutoTokenizer.from_pretrained("dslim/bert-base-NER")
 ner_model = AutoModelForTokenClassification.from_pretrained("dslim/bert-base-NER")
-ner_pipeline = pipeline("ner", model=ner_model, tokenizer=ner_tokenizer, grouped_entities=True)
+ner_pipeline = pipeline(
+    "ner",
+    model=ner_model,
+    tokenizer=ner_tokenizer,
+    grouped_entities=True
+)
 
 # ----------- Core Functions -----------
 def fetch_news(ticker):
@@ -42,9 +51,9 @@ def fetch_news(ticker):
         for row in news_table.findAll('tr')[:30]:
             a_tag = row.find('a')
             if a_tag:
-                title = a_tag.get_text()
+                title_text = a_tag.get_text()
                 link = a_tag['href']
-                news.append({'title': title, 'link': link})
+                news.append({'title': title_text, 'link': link})
         return news
     except Exception:
         return []
@@ -70,3 +79,16 @@ def extract_org_entities(text):
         return org_entities
     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