Add application file

financial_news_sentiment_and_summarization.py

@@ -0,0 +1,71 @@
+# imports
+import toml
+import finnhub
+import datetime
+from transformers import pipeline
+import streamlit as st
+
+
+# load API key
+# get API Keys
+#with open('secrets.toml', 'r') as f:
+ #   config = toml.load(f)
+
+#FINNHUB_API_KEY = config['FINNHUB']
+FINNHUB_API_KEY = st.secrets['FINNHUB']
+# function to get financial news
+def financial_news(stock_ticker, start_date, end_date, include_headline=True, include_summary=False):
+    """
+    Retrieves financial news for a specified stock within a date range.
+
+    Args:
+        stock_ticker (str): The ticker symbol of the stock (e.g., "AAPL" for Apple Inc.).
+        start_date (str): The start date for the news search in the format "YYYY-MM-DD".
+        end_date (str): The end date for the news search in the format "YYYY-MM-DD".
+        include_headline (bool, optional): If True, includes both headlines and summaries in the result.
+                                           Defaults to True.
+
+    Returns:
+        str: A concatenated string of headlines and summaries for the specified stock within the date range.
+             If `include_headline` is False, only the summary is included.
+    """
+    finnhub_client = finnhub.Client(api_key=FINNHUB_API_KEY)
+    news_list = finnhub_client.company_news(stock_ticker, _from=start_date, to=end_date)
+
+    news = ''
+    for news_i in news_list:
+        if stock_ticker in news_i['headline']:
+            if include_headline:
+                news = news + ' ' + news_i['headline'] + '.\n\n'
+            if include_summary:
+                news = news + ' ' + news_i['summary'] + '.\n\n'
+
+    return news
+
+# get financial news summary
+@st.cache_resource
+def load_summarizer():
+    return pipeline("summarization", model="facebook/bart-large-cnn")
+
+SUMMARIZER = load_summarizer()
+
+def get_news_summary(news):
+    news = news.replace('\n\n', ' ')
+    news_summary = SUMMARIZER(news)
+    return news_summary[0]['summary_text']
+
+# APP
+st.title('Financial News Summarization and Sentiment')
+st.write('Enter the stock ticker and period for which you want financial news.')
+stock_ticker = st.text_input('Stock Ticker:', 'AAPL')
+start_date = st.date_input('Start Date:', datetime.datetime.today()-datetime.timedelta(days=20))
+end_date = st.date_input('End Date:', datetime.datetime.today())
+news = financial_news(stock_ticker, start_date, end_date)
+st.write(news)
+
+st.header('Financial News Sentiment')
+
+st.header('Financial News Summary')
+with st.spinner('facebook bart model is summarizing the news...'):
+    news_summary = get_news_summary(news)
+st.write(news_summary)

requirements.txt

@@ -0,0 +1,26 @@
+certifi==2024.8.30
+charset-normalizer==3.4.0
+docopt==0.6.2
+filelock==3.16.1
+finnhub-python==2.4.21
+fsspec==2024.10.0
+huggingface-hub==0.26.2
+idna==3.10
+numpy==2.1.3
+packaging==24.2
+pipreqs==0.4.13
+PyYAML==6.0.2
+regex==2024.11.6
+requests==2.32.3
+safetensors==0.4.5
+setuptools==75.1.0
+tokenizers==0.20.3
+toml==0.10.2
+tqdm==4.67.0
+torch
+transformers==4.46.2
+typing_extensions==4.12.2
+urllib3==2.2.3
+wheel==0.44.0
+yarg==0.1.10
+streamlit