Update app.py

app.py

@@ -239,6 +239,10 @@ def display_docs_modal(docs):
   return output_list
 
 
+def hide_source():
+    return gr.Markdown(value=documents.split(label='Source', visible=False)
+
+
 def ask_llm(system, user_input):
     messages = [
         {
@@ -324,38 +328,38 @@ def auth_user(ui_session_id):
 
 def display_info0(documents):
     try:
-        return gr.Markdown(value=documents.split("\n*§*§*\n")[0], label='Source', visible=True)
+        return gr.Markdown(value=documents.split("\n*§*§*\n")[0], label='Source', visible=True), gr.Button('Hide', visible=True)
     except Exception as e:
         gr.Info("No Document")
-        return gr.Markdown(label='Source', visible=False)
+        return gr.Markdown(label='Source', visible=False), gr.Button('Hide', visible=False)
 
 def display_info1(documents):
     try:
-        return gr.Markdown(value=documents.split("\n*§*§*\n")[1], label='Source', visible=True)
+        return gr.Markdown(value=documents.split("\n*§*§*\n")[1], label='Source', visible=True), gr.Button('Hide', visible=True)
     except Exception as e:
         gr.Info("No Document")
-        return gr.Markdown(label='Source', visible=False)
+        return gr.Markdown(label='Source', visible=False), gr.Button('Hide', visible=False)
         
 def display_info2(documents):
     try:
-        return gr.Markdown(value=documents.split("\n*§*§*\n")[2], label='Source', visible=True)
+        return gr.Markdown(value=documents.split("\n*§*§*\n")[2], label='Source', visible=True), gr.Button('Hide', visible=True)
     except Exception as e:
         gr.Info("No Document")
-        return gr.Markdown(label='Source', visible=False)
+        return gr.Markdown(label='Source', visible=False), gr.Button('Hide', visible=False)
     
 def display_info3(documents):
     try:
-        return gr.Markdown(value=documents.split("\n*§*§*\n")[3], label='Source', visible=True)
+        return gr.Markdown(value=documents.split("\n*§*§*\n")[3], label='Source', visible=True), gr.Button('Hide', visible=True)
     except Exception as e:
         gr.Info("No Document")
-        return gr.Markdown(label='Source', visible=False)
+        return gr.Markdown(label='Source', visible=False), gr.Button('Hide', visible=False)
     
 def display_info4(documents):
     try:
-        return gr.Markdown(value=documents.split("\n*§*§*\n")[4], label='Source', visible=True)
+        return gr.Markdown(value=documents.split("\n*§*§*\n")[4], label='Source', visible=True), gr.Button('Hide', visible=True)
     except Exception as e:
         gr.Info("No Document")
-        return gr.Markdown(label='Source', visible=False)
+        return gr.Markdown(label='Source', visible=False), gr.Button('Hide', visible=False)
 
 with gr.Blocks() as demo:
     gr.Markdown("# Enrich an LLM knowledge with your own documents 🧠🤖")
@@ -373,6 +377,7 @@ with gr.Blocks() as demo:
         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)
+        btn_hide_source = gr.Button('Hide', visible=False)
         with gr.Row():
             btn1 = gr.Button("Ref 1")
             btn2 = gr.Button("Ref 2")
@@ -396,11 +401,12 @@ with gr.Blocks() as demo:
     btn_askGPT.click(ask_gpt, inputs=[query_input, tb_session_id, tb_history], outputs=[answer_output, tb_sources, tb_history])
     query_input.submit(ask_gpt, inputs=[query_input, tb_session_id, tb_history], outputs=[answer_output, tb_sources, tb_history])
 
-    btn1.click(display_info0, inputs=tb_sources, outputs=md_ref)
-    btn2.click(display_info1, inputs=tb_sources, outputs=md_ref)
-    btn3.click(display_info2, inputs=tb_sources, outputs=md_ref)
-    btn4.click(display_info3, inputs=tb_sources, outputs=md_ref)
-    btn5.click(display_info4, inputs=tb_sources, outputs=md_ref)
+    btn1.click(display_info0, inputs=tb_sources, outputs=[md_ref, btn_hide_source])
+    btn2.click(display_info1, inputs=tb_sources, outputs=[md_ref, btn_hide_source])
+    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)