Update app.py

app.py

@@ -77,9 +77,9 @@ def getstuff(openapikey,category_selector,ticker_input):
     
       # Create a loop to go through each section type and save only the 10-K section in the dictionary
       for doc_type, doc_start, doc_end in zip(doc_types, doc_start_is, doc_end_is):
-          if doc_type == '10-K':
+          if doc_type == category_selector:
               document[doc_type] = raw_10k[doc_start:doc_end]
-      item_content = BeautifulSoup(document['10-K'], 'lxml')
+      item_content = BeautifulSoup(document[category_selector], 'lxml')
     
       filetextcontentlist.append(str(item_content.text.encode('ascii','ignore')))
 
@@ -94,7 +94,7 @@ def getstuff(openapikey,category_selector,ticker_input):
     documents=[Document(t) for t in newlist]
     index = GPTSimpleVectorIndex.from_documents(documents)
     print('Came here4')
-    querylist=['What are the main products/ services mentioned?','What are the major risks?',"What are the top investment focus areas?","What is the financial outlook of the company?"]
+    querylist=['What are the main products/ services mentioned?','What are the major risks?',"What are the top investment focus areas?","What is the financial outlook of the company?","What key technologies like AI, blockchain etc are mentioned?","What other company names/ competitors are mentioned?"]
     
     
     
@@ -128,7 +128,7 @@ def getstuff(openapikey,category_selector,ticker_input):
                 df['action']=['try again in a few hours']
             answerlist.append(df)
         else:
-            answerlist.append(response.response)
+            answerlist.append(querylist[i]+'\n'+response.response)
 
     print('Came to return statement')
     return answerlist
@@ -161,8 +161,13 @@ with gr.Blocks() as demo:
             output3 = gr.Textbox(placeholder='', lines=4,label='Snapshot 3')
         with gr.Column():
             output4 = gr.Textbox(placeholder='', lines=4,label='Snapshot 4')
-
-    btn.click(getstuff, inputs=[textboxopenapi,category_selector,input1],outputs=[output1,output2,output3,output4])
+    with gr.Row() as row:
+        with gr.Column():
+            output5 = gr.Textbox(placeholder='', lines=4,label='Snapshot 5')
+        with gr.Column():
+            output6 = gr.Textbox(placeholder='', lines=4,label='Snapshot 6')
+            
+    btn.click(getstuff, inputs=[textboxopenapi,category_selector,input1],outputs=[output1,output2,output3,output4,output5,output6])
     
     
 demo.launch(debug=True)