changes to hide tabs until data is loaded

app.py

@@ -294,18 +294,32 @@ with gr.Blocks(theme='aliabid94/new-theme') as demo:
         openai_api_key = gr.Textbox(label = 'Please provide your OpenAI API Key:', type = 'password', placeholder = 'Note: To use the OpenAI API, you need a paid account')
         api_key_output = gr.Textbox(label = 'Result')
         load_button = gr.Button('Load TLS Protect Cloud Data')
-    with gr.Tab("Answer Questions"):
+    with gr.Tab("Answer Questions", visibility=False) as answer_questions:
         #prompt_tlspc_key = gr.Textbox(label = 'Please provide your TLS Protect Cloud API Key:')
         prompt_questions = gr.Textbox(label = 'Input prompt here:', placeholder = "Try something like 'What is the name of the issuing template that has been used to request the most certificates?'")
         text_output = gr.Textbox(label = 'Response:')
         text_button = gr.Button("Submit")
-    with gr.Tab("Create Graphs"):
+    with gr.Tab("Create Graphs", visibility=False) as create_graphs:
         prompt_reporting = gr.Textbox(label = 'Input prompt here:', placeholder = "Try something like 'Plot a line chart of certificate issuances over time'")
         chart_output = gr.Plot(label = 'Output:')
         chart_button = gr.Button("Submit")
 
+    def load_data_submit(tlspc_api_key, open_api_key):
+        data_return = getData(tlspc_api_key,openai_api_key)
+        if "Error" in data_return:
+            return {api_key_output: data_return}
+        else:
+            return {
+                api_key_output: data_return,
+                answer_questions: gr.update(visible=True),
+                create_graphs: gr.update(visible=True)
+            }
+
+
+
     text_button.click(prompt_analyze_questions, inputs=prompt_questions, outputs=text_output)
     chart_button.click(prompt_analyze_reporting, inputs=prompt_reporting, outputs=chart_output)
-    load_button.click(getData, inputs = [tlspc_api_key, openai_api_key], outputs = api_key_output)
+    load_button.click(load_data_submit, inputs = [tlspc_api_key, openai_api_key],
+                      outputs = [api_key_output, answer_questions, create_graphs])
 
 demo.launch()