Update app.py

app.py

@@ -324,33 +324,38 @@ def auth_user(ui_session_id):
 
 def display_info0(documents):
     try:
-        gr.Info(documents.split("\n*§*§*\n")[0])
+        return gr.Markdown(value=documents.split("\n*§*§*\n")[0], label='Source', visible=True)
     except Exception as e:
         gr.Info("No Document")
+        return gr.Markdown(label='Source', visible=False)
 
 def display_info1(documents):
     try:
-        gr.Info(documents.split("\n*§*§*\n")[1])
+        return gr.Markdown(value=documents.split("\n*§*§*\n")[1], label='Source', visible=True)
     except Exception as e:
         gr.Info("No Document")
+        return gr.Markdown(label='Source', visible=False)
         
 def display_info2(documents):
     try:
-        gr.Info(documents.split("\n*§*§*\n")[2])
+        return gr.Markdown(value=documents.split("\n*§*§*\n")[2], label='Source', visible=True)
     except Exception as e:
         gr.Info("No Document")
+        return gr.Markdown(label='Source', visible=False)
     
 def display_info3(documents):
     try:
-        gr.Info(documents.split("\n*§*§*\n")[3])
+        return gr.Markdown(value=documents.split("\n*§*§*\n")[3], label='Source', visible=True)
     except Exception as e:
         gr.Info("No Document")
+        return gr.Markdown(label='Source', visible=False)
     
 def display_info4(documents):
     try:
-        gr.Info(documents.split("\n*§*§*\n")[4])
+        return gr.Markdown(value=documents.split("\n*§*§*\n")[4], label='Source', visible=True)
     except Exception as e:
         gr.Info("No Document")
+        return gr.Markdown(label='Source', visible=False)
 
 with gr.Blocks() as demo:
     gr.Markdown("# Enrich an LLM knowledge with your own documents 🧠🤖")
@@ -363,6 +368,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)
+
         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)
@@ -389,11 +396,11 @@ 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=None)
-    btn2.click(display_info1, inputs=tb_sources, outputs=None)
-    btn3.click(display_info2, inputs=tb_sources, outputs=None)
-    btn4.click(display_info3, inputs=tb_sources, outputs=None)
-    btn5.click(display_info4, inputs=tb_sources, outputs=None)
+    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)