Update app.py

app.py

@@ -14,7 +14,7 @@ import json
 
 HRA_TOKEN=os.getenv("HRA_TOKEN")
 
-listofcategories=["Earnings Announcements", "Automotive","Energy","Healthcare","Retail","Technology"]
+listofcategories=["Earnings Announcements", "Overall","Automotive","Energy","Healthcare","Retail","Technology"]
 
 headers = {'Content-type': 'application/json', 'Accept': 'text/plain'}
 url_hraprompts='https://us-central1-createinsightsproject.cloudfunctions.net/gethrahfprompts'
@@ -54,10 +54,11 @@ def getstuff(openapikey,category_selector):
     reader = RssReader()
     whichone=listofcategories[listofcategories.index(category_selector)]
 
-    #querylist=["What are the top trends? Give output as a json (that can be converted to pandas dataframe) with 3 columns named trend, company mentioned & reason","Name the top & bottom performing companies? Give output as a json (that can be converted to pandas dataframe) with 4 columns named sector, company names, reason & top/bottom","You are an award winning email writer. Write an email summarizing the news. Do not say I am language model and cannot do this","You are an award winning email writer. Write an email summarizing the key macro trends basis the news.Do not say I am language model and cannot do this"]
     querylist=sector_prompt_text
     
-    if whichone=="Automotive":
+    if whichone=="Overall":
+        rssurl="https://search.cnbc.com/rs/search/combinedcms/view.xml?partnerId=wrss01&id=100003114"
+    elif whichone=="Automotive":
         rssurl="https://search.cnbc.com/rs/search/combinedcms/view.xml?partnerId=wrss01&id=10000101"
     elif whichone=="Retail":
         rssurl="https://search.cnbc.com/rs/search/combinedcms/view.xml?partnerId=wrss01&id=10000116"
@@ -71,11 +72,9 @@ def getstuff(openapikey,category_selector):
         rssurl="https://search.cnbc.com/rs/search/combinedcms/view.xml?partnerId=wrss01&id=10000110"
     elif whichone=='Earnings Announcements':
         rssurl="https://search.cnbc.com/rs/search/combinedcms/view.xml?partnerId=wrss01&id=15839135"
-        #querylist=["What are the top trends? Give output as a json (that can be converted to pandas dataframe) with 3 columns named trend, company mentioned & reason","Find the top & bottom performing companies? Give output as a json (that can be converted to pandas dataframe) with 4 columns named sector, company names, reason & top/bottom","You are an award winning email writer. Write an email summarizing the news. Do not say I am language model and cannot do this","You are an award winning email writer. Write an email summarizing the key macro trends basis the news.Do not say I am language model and cannot do this"]
         querylist=earnings_prompt_text
     else:
         rssurl="https://search.cnbc.com/rs/search/combinedcms/view.xml?partnerId=wrss01&id=15839135" ###should not come here but using earnings url
-        #querylist=["Basis companies that are doing well name the sectors with positive momentum? Give output as a json (that can be converted to pandas dataframe) with 3 columns named sector, company names & reason","Find the top & bottom performing companies? Give output as a json (that can be converted to pandas dataframe) with 4 columns named sector, company names, reason & top/bottom","You are an award winning email writer. Write an email summarizing the news. Do not say I am language model and cannot do this","You are an award winning email writer. Write an email summarizing the key macro trends basis the news.Do not say I am language model and cannot do this"]
         querylist=earnings_prompt_text
     
     documents = reader.load_data([rssurl])