Update app.py

Added stock stuff

app.py

@@ -17,6 +17,10 @@ from PIL import Image
 from io import BytesIO
 from pydantic import BaseModel
 import pprint
+import pandas as pd
+import yfinance as yf
+from datetime import datetime, timedelta
+
 
 load_dotenv(override=True)
 key = os.getenv('OPENAI_API_KEY')
@@ -34,6 +38,8 @@ else:
     dp = Path('/data')
     dp.mkdir(exist_ok=True)
     dataDir = '/data/'
+stock_data_path = dataDir + 'Stocks.txt'
+
 
 speak_file = dataDir + "speek.wav"
 
@@ -56,6 +62,42 @@ class MathReasoning(BaseModel):
 def Client():
     return OpenAI(api_key = key)
 
+def get_stock_report():
+    stock_data = {}
+    global stock_data_path
+    with open(stock_data_path, 'rt') as fp:
+        lines = fp.readlines()
+    for line in lines:
+        (name, symbol, shares) = line.rstrip().split(',')
+        stock_data[symbol] = {"symbol": symbol, "name": name, "shares": shares, "closing": '0'}
+    for symbol in stock_data.keys():
+        stock_data[symbol]['closing'] = f'{get_last_closing(symbol):.2f}'
+    total_value = 0.0
+    rv = ''
+    for item in stock_data.values():
+        rv += str(item) + '\n'
+        total_value += float(item['closing']) * float(item['shares'])
+    rv += (f'\nTotal value = {total_value:.2f}')
+    return rv
+
+def get_last_closing(symbol, timeout=10):
+    today = datetime.today()+timedelta(days=1)
+    five_days_ago = today - timedelta(days=5)
+    end = today.strftime('%Y-%m-%d')
+    start = five_days_ago.strftime('%Y-%m-%d')
+    df = yf.download(symbol,
+                    start = start,
+                    end = end,
+                    progress = False,
+                    timeout=timeout,
+                    )
+    # print(df)
+    return df.iat[-1,1]
+
+def create_stock_data_file(txt):
+    with open(stock_data_path, 'wt') as fp:
+        fp.write(txt)
+
 def solve(prompt, chatType):
     tokens_in = 0
     tokens_out = 0
@@ -270,6 +312,12 @@ def chat(prompt, user_window, pwd_window, past, response, gptModel, uploaded_ima
             (log_cnt, wav_cnt, other_cnt, others, log_list) = list_permanent_files()
             response = f'{log_cnt} log files\n{wav_cnt} .wav files\n{other_cnt} Other files:\n{others}\nlogs: {str(log_list)}'
             return [past, response, None, gptModel, uploaded_image_file]
+        if prompt.startswith('stocks'):
+            response = get_stock_report()
+            return [past, response, None, gptModel, uploaded_image_file]
+        if prompt.startswith('stockload'):
+            create_stock_data_file(prompt[9:].lstrip())
+            return [past, 'Stock data file created', None, gptModel, uploaded_image_file]
     if user_window in unames and pwd_window == pwdList[unames.index(user_window)]:
         chatType = 'normal'
         prompt = prompt.strip()
@@ -504,7 +552,6 @@ def show_help():
           5.4 The image should appear. This can take some time with a slow internet connection and large image.
           5.5 Tap the "Submit Prompt/Question" button to start the analysis.  This initiates a chat dialog and
                you can ask follow-up questions. However, the image is not re-analyzed for follow-up dialog.
-
     Hints:
         1. Better chat and image results are obtained by including detailed descriptions and instructions
             in the prompt.