Update app.py

app.py

@@ -36,7 +36,11 @@ prompt_template = "Instructions: Compose a comprehensive reply to the query usin
 # MODELS = ["universal-sentence-encoder", "instructor-large"]
 MODELS = ["text-davinci-001", "text-davinci-002", "text-davinci-003"]
 
+def set_openai_api_key(my_api_key):
+    openai.api_key = my_api_key
+    return gr.update(visible = True)
 
+    
 def add_source_numbers(lst):
     return [item[:3] + '\t' + item[3:] for item in (lst)]
 
@@ -119,7 +123,6 @@ def get_top_chunks(inp_emb, data, n_neighbors=5):
 
 
 def predict(
-        my_api_key,
         history,
         chatbot,
         inputs,
@@ -155,7 +158,6 @@ def predict(
         chatbot.append((old_inputs, ""))
     else:
         chatbot.append((inputs, ""))
-    open_ai.api_key = my_api_key
     completions = openai.Completion.create(
         engine=selected_model,
         prompt=inputs,
@@ -215,7 +217,7 @@ with gr.Blocks(css=customCSS, theme=beautiful_theme) as demo:
         with gr.Column(scale=5):
             with gr.Row():
                 chatbot = gr.Chatbot(elem_id="chatbot").style(height="100%")
-            with gr.Row():
+            with gr.Row() as input_raws:
                 with gr.Column(scale=12):
                     user_input = gr.Textbox(
                         show_label=False, placeholder="Enter here"
@@ -244,7 +246,7 @@ with gr.Blocks(css=customCSS, theme=beautiful_theme) as demo:
                             step=0.1,
                             interactive=True,
                             label="Temperature",
-                        )
+    openAI_key.submit(set_openai_api_key, [openAI_key], [input_raws])                    )
     user_input.submit(predict, inputs=[history, chatbot, user_input, temperature, model_select_dropdown, index_files],
                       outputs=[chatbot, history])
     user_input.submit(lambda: "", None, user_input)