Update app.py

Added brave news search for context

app.py

@@ -23,8 +23,9 @@ from datetime import datetime, timedelta
 import pytz
 import math
 import numpy as np
-# import matplotlib.pyplot as plt
 from pylatexenc.latex2text import LatexNodes2Text
+import requests
+from urllib.parse import quote
 
 load_dotenv(override=True)
 key = os.getenv('OPENAI_API_KEY')
@@ -34,6 +35,7 @@ pwds = os.getenv('PASSWORD')
 pwdList = pwds.split(',')
 DEEPSEEK_KEY=os.getenv('DEEPSEEK_KEY')
 GROQ_KEY=os.getenv('GROQ_KEY')
+BRAVE_KEY=os.getenv('BRAVE_KEY')
 
 site = os.getenv('SITE')
 if site == 'local':
@@ -46,6 +48,7 @@ else:
     dataDir = '/data/'
 stock_data_path = dataDir + 'Stocks.txt'
 
+braveNewsEndpoint = "https://api.search.brave.com/res/v1/news/search"
 
 speak_file = dataDir + "speek.wav"
 
@@ -57,6 +60,8 @@ abbrevs = {'St. ' : 'Saint ', 'Mr. ': 'mister ', 'Mrs. ':'mussus ', 'Mr. ':'mist
 
 special_chat_types = ['math', 'logic']
 
+news_interval_choices = [("None", "None"), ("Day", "pd"), ("Week", "pw"), ("Month", "pm"), ("Year", "py")]
+
 class Step(BaseModel):
     explanation: str
     output: str
@@ -65,6 +70,29 @@ class MathReasoning(BaseModel):
     steps: list[Step]
     final_answer: str
 
+def get_brave_news(query: str, interval: str = 'pd'):
+    url = f'{braveNewsEndpoint}?q={quote(query)}&count=20&extra_snippets=true&freshness={interval}' 
+    response = requests.get(
+        url,
+        headers= {"Accept": "application/json",
+                  "X-Subscription-Token": BRAVE_KEY
+                  },
+        )
+    rv = ''
+    jdata = response.json()
+    for item in jdata['results']:
+        title = item['title']
+        description = item['description']
+        rv += f'{title}: {description} --'
+        try:  # extra_snippets can be missing
+            for snip in item['extra_snippets']:
+                rv += (snip + ' ')
+        except:
+            continue
+        rv += ' ** '
+
+    return rv
+
 def Client():
     return OpenAI(api_key = key)
 
@@ -585,8 +613,9 @@ def new_conversation(user):
     for fpath in flist:
         if os.path.exists(fpath):
             os.remove(fpath)
-    return [None, [], gr.Markdown(value='', label='Dialog', container=True), gr.Image(visible=False, value=None),  gr.Image(visible=False, value=None), '',
-            gr.LinePlot(visible=False)]
+    return [None, [], gr.Markdown(value='', label='Dialog', container=True),
+            gr.Image(visible=False, value=None),  gr.Image(visible=False, value=None), '',
+            gr.LinePlot(visible=False), gr.Dropdown(value="None")]
 
 def updatePassword(txt):
     password = txt.lower().strip()
@@ -610,7 +639,7 @@ def updatePassword(txt):
 #             ref = len(txt[ref:loc]) + len(frag)
 #     return txt
 
-def chat(prompt, user_window, pwd_window, past, response, gptModel, uploaded_image_file='', plot=None):
+def chat(prompt, user_window, pwd_window, past, response, gptModel, uploaded_image_file='', plot=None, news_interval = 'none'):
     image_gen_model = 'gpt-4o-2024-08-06'
     user_window = user_window.lower().strip()
     isBoss = False
@@ -676,6 +705,7 @@ def chat(prompt, user_window, pwd_window, past, response, gptModel, uploaded_ima
         using_groq = False
         reasoning = False
         prompt = prompt.strip()
+        news_prompt = prompt
         finish_reason = 'ok'
         if prompt.lower().startswith('dsr1 '):
             deepseek = True
@@ -711,7 +741,10 @@ def chat(prompt, user_window, pwd_window, past, response, gptModel, uploaded_ima
             prompt = prompt[6:]
         if deepseek:
             prompt = prompt + '. Do not use Latex for math expressions.'
-        past.append({"role":"user", "content":prompt})
+        if news_interval != "None" and past==[]:
+            news = get_brave_news(prompt, news_interval)
+            news_prompt = f'{news}\n{prompt}\nGive highest priority to information just provided\n'
+        past.append({"role":"user", "content":news_prompt})
         gen_image = (uploaded_image_file != '')
         if chatType in special_chat_types:
            (reply, tokens_in, tokens_out, tokens) = solve(prompt, chatType)
@@ -1179,7 +1212,9 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         button_get_image = gr.Button(value='Upload Image to Analyze')
         speak_output = gr.Button(value="Speak Dialog", visible=True)
         submit_button = gr.Button(value="Submit Prompt/Question")
-    prompt_window = gr.Textbox(label = "Prompt or Question")
+    with gr.Row():
+        prompt_window = gr.Textbox(label = "Prompt or Question", scale=5)
+        news_period = gr.Dropdown(choices=news_interval_choices, interactive=True,label='News Window',scale=1)
     gr.Markdown('### **Dialog:**')
     #output_window = gr.Text(container=True, label='Dialog')
     output_window = gr.Markdown(container=True)
@@ -1192,10 +1227,10 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         # plot = gr.Plot(visible=False)
         plot = gr.LinePlot(test_plot_df(), x="month", y="value", visible=False, label="Portfolio Value History")
     submit_button.click(chat,
-             inputs=[prompt_window, user_window, password, history, output_window, model, uploaded_image_file],
+             inputs=[prompt_window, user_window, password, history, output_window, model, uploaded_image_file, plot, news_period],
              outputs=[history, output_window, prompt_window, model, uploaded_image_file, plot])
     clear_button.click(fn=new_conversation, inputs=user_window,
-                      outputs=[prompt_window, history, output_window, image_window, image_window2, uploaded_image_file, plot])
+                      outputs=[prompt_window, history, output_window, image_window, image_window2, uploaded_image_file, plot, news_period])
     audio_widget.stop_recording(fn=transcribe, inputs=[user_window, password, audio_widget],
                                 outputs=[prompt_window])
     audio_widget.pause_recording(fn=pause_message, outputs=[prompt_window])