Update app.py

app.py

@@ -23,16 +23,16 @@ def getstuff(openapikey,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 table with 3 columns named trend, company mentioned & reason","Write an email summarizing the news"]
+        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 table with 3 columns named trend, company mentioned & reason","Write an email summarizing the news"]
+        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 whicone=='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 table with 3 columns named sector, company names & reason","Write an email summarizing the news"]
+        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 table with 3 columns named sector, company names & reason","Write an email summarizing the news"]
+        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)
@@ -47,7 +47,12 @@ def getstuff(openapikey,category_selector):
         response_mode="tree_summarize",
         similarity_top_k=int(len(documents)/3)
         )
-        answerlist.append(response.response)
+        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
@@ -69,9 +74,9 @@ with gr.Blocks() as demo:
         
     with gr.Row() as row:
       with gr.Column():
-        outputsector = gr.Textbox(placeholder='', lines=4,label='Sector Snapshot')
+        outputsector = gr.Textbox(placeholder='', lines=4,label='Snapshot 1')
       with gr.Column():
-        outputtech = gr.Textbox(placeholder='', lines=4,label='Technology Snapshot')
+        outputtech = gr.Textbox(placeholder='', lines=4,label='Snapshot 2')
 
     btn.click(getstuff, inputs=[textboxopenapi,category_selector],outputs=[outputsector,outputtech])