Spaces:
Runtime error
Runtime error
File size: 4,525 Bytes
710fd08 64adf78 710fd08 64adf78 e375a0b 710fd08 64adf78 710fd08 64adf78 710fd08 e162922 710fd08 663d261 710fd08 285fd92 710fd08 64adf78 710fd08 285fd92 710fd08 ee1e26a 710fd08 ee1e26a e375a0b 710fd08 e375a0b 710fd08 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 |
from llama_index import GPTSimpleVectorIndex, SimpleDirectoryReader, LLMPredictor
from langchain.llms import OpenAIChat
from llama_index import download_loader
import gradio as gr
import pandas as pd
import openai
import datetime
from datetime import datetime, date, time, timedelta
import os
listofcategories=["Earnings Announcements", "Automotive", "Retail"]
def getstuff(openapikey,category_selector):
dateforfilesave=datetime.today().strftime("%d-%m-%Y %I:%M%p")
print(category_selector)
print(dateforfilesave)
os.environ['OPENAI_API_KEY'] = str(openapikey)
RssReader = download_loader("RssReader")
reader = RssReader()
whichone=listofcategories[listofcategories.index(category_selector)]
if whichone=="Automotive":
rssurl="https://search.cnbc.com/rs/search/combinedcms/view.xml?partnerId=wrss01&id=10000101"
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","Write an email summarizing the news. Do not say I am language model and cannot do this"]
elif whichone=="Retail":
rssurl="https://search.cnbc.com/rs/search/combinedcms/view.xml?partnerId=wrss01&id=10000116"
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","Write an email summarizing the news. Do not say I am language model and cannot do this"]
elif whichone=='Earnings Announcements':
rssurl="https://search.cnbc.com/rs/search/combinedcms/view.xml?partnerId=wrss01&id=15839135"
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","Write an email summarizing the news. Do not say I am language model and cannot do this"]
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","Write an email summarizing the news. Do not say I am language model and cannot do this"]
documents = reader.load_data([rssurl])
index = GPTSimpleVectorIndex(documents)
llm_predictor = LLMPredictor(llm=OpenAIChat(temperature=0, model_name="gpt-3.5-turbo"))
answerlist=[]
for i in range(len(querylist)):
print(i)
response = index.query(
querylist[i],
llm_predictor=llm_predictor,
response_mode="tree_summarize",
similarity_top_k=int(len(documents)/3)
)
if 'dataframe' in querylist[i]:
df_tmp=pd.read_json(response.response)
df=pd.DataFrame(df_tmp[df_tmp.columns[0]].tolist())
answerlist.append(df)
else:
answerlist.append(response.response)
return answerlist
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 top technologies? Which companies have momentum? And much more. \n\nShowcases 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.\nLangChain & GPT-Index are both used.\n """
)
with gr.Row() as row:
with gr.Column():
textboxopenapi = gr.Textbox(placeholder="Enter OpenAPI Key...", lines=1,label='OpenAPI Key')
category_selector=gr.Dropdown(
listofcategories, label="Sector", info="Select the snapshot you want..."
)
with gr.Column():
btn = gr.Button("Generate")
with gr.Row() as row:
table1=gr.Dataframe(
#headers=["Item", "Cost"],
datatype=["str", "str","str"],
label="Snapshot 1",
)
with gr.Row() as row:
outputtech = gr.Textbox(placeholder='', lines=4,label='Snapshot 2')
btn.click(getstuff, inputs=[textboxopenapi,category_selector],outputs=[table1,outputtech])
demo.launch(debug=True) |