Update app.py

added stock news

app.py

@@ -1,7 +1,7 @@
 import os
 import gradio as gr
 import openai
-from numpy._core.defchararray import endswith, isdecimal
+from numpy._core.defchararray import endswith, isdecimal, startswith
 from openai import OpenAI
 from dotenv import load_dotenv
 from pathlib import Path
@@ -75,6 +75,32 @@ def etz_now():
     ltime = datetime.now(eastern)
     return ltime
 
+def date_from_utime(utime):
+    ts = int(utime)
+    dt = datetime.utcfromtimestamp(ts)
+    eastern = pytz.timezone('US/Eastern')
+    return dt.astimezone(eastern).strftime('%Y-%m-%d')
+
+def get_stock_news(search_symbol):
+    name_dict = {}
+    with open(stock_data_path, 'rt') as fp:
+        lines = fp.readlines()
+        for line in lines:
+            (name, symbol, shares) = line.rstrip().split(',')
+            name = name.strip()
+            symbol = symbol.strip()
+            name_dict[symbol] = name
+    try:
+        news = yf.Search(name_dict[search_symbol.upper()], news_count=5).news
+    except:
+        return f'No results for symbol {search_symbol}, check spelling'
+    rv = ''
+    for item in news:
+        rv += f'Title: {item["title"]}\n'
+        rv += f'Publisher: {item["publisher"]}\n'
+        rv += f'Date published: {date_from_utime(item["providerPublishTime"])}\n\n'
+    return rv
+
 def stock_history_df(num_weeks):
     values = []
     dates = []
@@ -386,6 +412,10 @@ def chat(prompt, user_window, pwd_window, past, response, gptModel, uploaded_ima
                                   y_lim=[500000, 700000], label="Portfolio Value History")]
             else:
                 return [past, response, None, gptModel, uploaded_image_file, plot]
+        if prompt.startswith('stock news'):
+                symbol = prompt[11:]
+                response = get_stock_news(symbol)
+                return [past, response, None, gptModel, uploaded_image_file, plot]
         if prompt.startswith('stockload'):
             create_stock_data_file(prompt[9:].lstrip())
             return [past, 'Stock data file created', None, gptModel, uploaded_image_file, plot]