Update app.py

app.py

@@ -9,9 +9,35 @@ import datetime
 from datetime import datetime, date, time, timedelta
 import os
 import regex
+import requests
 
 listofcategories=["Earnings Announcements", "Automotive","Energy","Healthcare","Retail","Technology"]
 
+headers = {'Content-type': 'application/json', 'Accept': 'text/plain'}
+url_hraprompts='https://us-central1-createinsightsproject.cloudfunctions.net/gethrahfprompts'
+
+data={"prompt_type":'chatgpt_stock_news_snapshot_sector',"hra_token":HRA_TOKEN}
+try:
+    r = requests.post(url_hraprompts, data=json.dumps(data), headers=headers)
+except requests.exceptions.ReadTimeout as e:
+    print(e)
+#print(r.content)
+
+
+sector_prompt_text=str(r.content, 'UTF-8')
+print(sector_prompt_text)
+
+data={"prompt_type":'chatgpt_stock_news_snapshot_earnings',"hra_token":HRA_TOKEN}
+try:
+    r = requests.post(url_hraprompts, data=json.dumps(data), headers=headers)
+except requests.exceptions.ReadTimeout as e:
+    print(e)
+#print(r.content)
+
+
+earnings_prompt_text=str(r.content, 'UTF-8')
+print(earnings_prompt_text)
+
 def getstuff(openapikey,category_selector):
     dateforfilesave=datetime.today().strftime("%d-%m-%Y %I:%M%p")
     print(category_selector)
@@ -25,8 +51,9 @@ 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=["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":
         rssurl="https://search.cnbc.com/rs/search/combinedcms/view.xml?partnerId=wrss01&id=10000101"
     elif whichone=="Retail":
@@ -41,10 +68,12 @@ 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=["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=["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])
     index = GPTSimpleVectorIndex(documents)
@@ -84,7 +113,7 @@ def getstuff(openapikey,category_selector):
 with gr.Blocks() as demo:
     gr.Markdown("<h1><center>ChatGPT Stock News Snapshots</center></h1>")
     gr.Markdown(
-        """What are the sectors with positive momentum? What are the macro trends? Which companies have momentum? Sector summaries and much more. \n\nThis is a demo & showcases ChatGPT integrated with real data. It shows how to get real-time data and marry it with ChatGPT capabilities. This demonstrates 'Chain of Thought' thinking using ChatGPT.\n4 snapshots are provided for illustration (trends, sector outlook, news summary email, macro trends email\n\nNote: llama-index & gpt-3.5-turbo are used. The analysis takes roughly 120 secs & may not always be consistent. If ChatGPT API is overloaded you will get an error\n ![visitors](https://visitor-badge.glitch.me/badge?page_id=hra.chatgpt-stock-news-snapshots)"""
+        """What are the sectors with positive momentum? What are the macro trends? Which companies have momentum? Sector summaries and much more. \n\nThis is a demo & showcases ChatGPT integrated with real data. It shows how to get real-time data and marry it with ChatGPT capabilities. This demonstrates 'Chain of Thought' thinking using ChatGPT.\n\n4 snapshots are provided for illustration (trends, sector outlook, news summary email, macro trends email\n\nNote: llama-index & gpt-3.5-turbo are used. The analysis takes roughly 120 secs & may not always be consistent. If ChatGPT API is overloaded you will get an error\n ![visitors](https://visitor-badge.glitch.me/badge?page_id=hra.chatgpt-stock-news-snapshots)"""
         )
     
     with gr.Row() as row: