Update app.py

app.py

@@ -240,7 +240,7 @@ def display_docs_modal(docs):
 
 
 def hide_source():
-    return gr.Markdown(label='Source', visible=False)
+    return gr.Markdown(label='Source', visible=False), gr.Button('Hide', visible=False)
 
 
 def ask_llm(system, user_input):
@@ -372,8 +372,8 @@ with gr.Blocks() as demo:
 
     with gr.Column():
         answer_output = gr.Markdown(label='AI Answer', visible=False)
-        md_ref = gr.Markdown(label='Source', visible=False)
         btn_hide_source = gr.Button('Hide', visible=False)
+        md_ref = gr.Markdown(label='Source', visible=False)
         with gr.Row():
             query_input = gr.Textbox(placeholder="Type your question", label="Question ❔", scale=9, visible=False)
             btn_askGPT = gr.Button("▶", scale=1, visible=False)
@@ -405,7 +405,7 @@ with gr.Blocks() as demo:
     btn3.click(display_info2, inputs=tb_sources, outputs=[md_ref, btn_hide_source])
     btn4.click(display_info3, inputs=tb_sources, outputs=[md_ref, btn_hide_source])
     btn5.click(display_info4, inputs=tb_sources, outputs=[md_ref, btn_hide_source])
-    btn_hide_source.click(hide_source, inputs=None, outputs=md_ref)
+    btn_hide_source.click(hide_source, inputs=None, outputs=[md_ref, btn_hide_source])