Update app.py

app.py

@@ -23,12 +23,13 @@ from sec_edgar_downloader._utils import get_filing_urls_to_download
 listofcategories=["10-K", "10-Q","8-K"]
 
 
-def getstuff(openapikey,category_selector,ticker_input):
+def getstuff(openapikey,category_selector,ticker_input,user_question):
     dateforfilesave=datetime.today().strftime("%d-%m-%Y %I:%M%p")
     print(ticker_input)
+    print(user_question)
     print(dateforfilesave)
     if openapikey=='':
-        return pd.DataFrame(["Please provide OpenAPI Key"],columns=['ERROR']),pd.DataFrame(["Please provide OpenAPI Key"],columns=['ERROR']),'Error: Please provide OpenAPI key','Error: Please provide OpenAPI key'
+        return ["Please provide OpenAPI Key","Please provide OpenAPI Key","Please provide OpenAPI Key","Please provide OpenAPI Key","Please provide OpenAPI Key","Please provide OpenAPI Key",]
     
     os.environ['OPENAI_API_KEY'] = str(openapikey)
 
@@ -41,7 +42,7 @@ def getstuff(openapikey,category_selector,ticker_input):
     else:
         num_filings_needed=1
     filings_temp=get_filing_urls_to_download(category_selector, ticker_input,num_filings_to_download=num_filings_needed,include_amends=False,before_date='2023-04-01',after_date='2022-01-01')
-    files=[filings_temp[0].full_submission_url]
+    files=[filings_temp[i].full_submission_url for i in range(len(filings_temp))]
     print('Came here1')
     filetextcontentlist=[]
     for each in files:
@@ -95,7 +96,10 @@ def getstuff(openapikey,category_selector,ticker_input):
     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?","What key technologies like AI, blockchain etc are mentioned?","What other company names/ competitors are mentioned?"]
-    
+    if user_question=='':
+        querylist.append('What is the key summary?')
+    else:
+        querylist.append(user_question)
     
     
     llm = ChatOpenAI(temperature=0)
@@ -146,6 +150,7 @@ with gr.Blocks() as demo:
                 )
             input1 = gr.Textbox(placeholder='Enter ticker (USA only)', lines=1,label='Ticker')
         with gr.Column():
+            input2 = gr.Textbox(placeholder='Enter your question', lines=1,label='User Question')
             textboxopenapi = gr.Textbox(placeholder="Enter OpenAPI Key...", lines=1,label='OpenAPI Key')
         
     with gr.Column():
@@ -167,7 +172,7 @@ with gr.Blocks() as demo:
         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])
+    btn.click(getstuff, inputs=[textboxopenapi,category_selector,input1,input2],outputs=[output1,output2,output3,output4,output5,output6,input2])
     
     
 demo.launch(debug=True)