Update app.py

app.py

@@ -40,6 +40,9 @@ def configure_chatgpt(info):
     global chatgpt
     chatgpt = Chatbot(config=config)
 
+    # Call initialize prompt on login
+    initialize.click()
+
 
 def ask(prompt):
     message = ""
@@ -96,14 +99,10 @@ with gr.Blocks() as demo:
                 with gr.Row():
                     login = gr.Button("Login")
                     login.click(configure_chatgpt, inputs=[info])
+                    
 
     l = os.listdir("./prompts")
     li = [x.split(".")[0] for x in l]
-
-    gr.Markdown("""<h4>Initial Prompt for ChatGPT</h4>""")
-
-    initialize = gr.Button(value="Initialize")
-
     gr.Markdown("""<h4>Conversation</h4>""")
     chatgpt_robot = gr.Chatbot()
     message = gr.Textbox(
@@ -114,6 +113,9 @@ with gr.Blocks() as demo:
 
     state = gr.State()
 
+# On login click, initialize the prompt, it should be hidden button which will be clicked on login
+    initialize = gr.Button("Initialize")
+    initialize.hidden = True
     initialize.click(fn=initialize_prompt, inputs=[state], outputs=[chatgpt_robot, state])
 
     message.submit(query_chatgpt, inputs=[message, state], outputs=[chatgpt_robot, state, message])