diff --git a/api_writer/test_ui.py b/api_writer/test_ui.py
index 7a42e62ee1a55358cb600892df836ee8f957dc19..3efa75929db1289e0062668b3b07e6affbd4df0e 100644
--- a/api_writer/test_ui.py
+++ b/api_writer/test_ui.py
@@ -170,7 +170,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         for i in range(n):
             with gr.Column():
-                output_text.append(gr.Textbox(label=f"LLM Generated query {i+1}", max_lines=100, interactive=True, submit_btn=True))
+                output_text.append(gr.Textbox(label=f"LLM Generated query {i+1}", max_lines=100, interactive=True, submit_btn=True, show_copy_button=True))
                 hidden_text.append(gr.Textbox(label=f"Original LLM Generated query {i+1}", interactive=False, visible=False))
         hidden_all_text = gr.Textbox(label="All LLM Generated queries", interactive=False, visible=False)
                 # rating.append(gr.CheckboxGroup(["👍", "👎"], label="Rating", info="👍  Upvote or 👎 Downvote"))