Update app.py

app.py

@@ -231,10 +231,14 @@ with gr.Blocks(css=css) as demo:
             
             chatbot.like(print_like_dislike, None, None)
 
-    initialize_button.click(
+      initialize_button.click(
         fn=initialize_chat,
         inputs=[initial_question_input],
         outputs=[chat_history, follow_up_questions_display, token_info_display, examples_state]
+    ).then(
+        fn=lambda chat_history: chat_history,
+        inputs=[chat_history],
+        outputs=[chatbot]
     )
 
 if __name__ == "__main__":