Update app.py

app.py

@@ -231,12 +231,12 @@ def display_docs(docs):
   return output_str
 
 
-def display_docs_modal(docs):
-  output_list = []
-  for i, doc in enumerate(docs):
-      source = doc.metadata['source'].split('/')[-1]
-      output_str.append(f"Ref: {i+1}\n{repr(doc.page_content)}\nSource: {source}\n*§*§*\n")
-  return output_list
+# def display_docs_modal(docs):
+#   output_list = []
+#   for i, doc in enumerate(docs):
+#       source = doc.metadata['source'].split('/')[-1]
+#       output_str.append(f"Ref: {i+1}\n{repr(doc.page_content)}\nSource: {source}\n*§*§*\n")
+#   return output_list
 
 
 def hide_source():
@@ -310,6 +310,8 @@ def ask_gpt(query, ui_session_id, history):
     gen = ask_llm_stream(system, query)
     last_value=""
     displayable_docs = display_docs(docs)
+    yn_display = len(docs)*[True]+(5-len(docs))*[False]
+        
     while True:
         try:
             last_value = next(gen)
@@ -317,7 +319,7 @@ def ask_gpt(query, ui_session_id, history):
         except StopIteration as e:
             break
     history += f"[query]\n{query}\n[answer]\n{last_value}\n[references]\n{displayable_docs}\n\n"
-    return last_value, displayable_docs, history
+    return last_value, displayable_docs, history, gr.Button("Ref 1", visible=yn_display[0]), gr.Button("Ref 2", visible=yn_display[1]), gr.Button("Ref 3", visible=yn_display[2]), gr.Button("Ref 4", visible=yn_display[3]), gr.Button("Ref 5", visible=yn_display[4])
 
 
 def auth_user(ui_session_id):
@@ -378,11 +380,11 @@ with gr.Blocks() as demo:
             query_input = gr.Textbox(placeholder="Type your question", label="Question ❔", scale=9, visible=False)
             btn_askGPT = gr.Button("▶", scale=1, visible=False)
         with gr.Row():
-            btn1 = gr.Button("Ref 1")
-            btn2 = gr.Button("Ref 2")
-            btn3 = gr.Button("Ref 3")
-            btn4 = gr.Button("Ref 4")
-            btn5 = gr.Button("Ref 5")
+            btn1 = gr.Button("Ref 1", visible=False)
+            btn2 = gr.Button("Ref 2", visible=False)
+            btn3 = gr.Button("Ref 3", visible=False)
+            btn4 = gr.Button("Ref 4", visible=False)
+            btn5 = gr.Button("Ref 5", visible=False)
         
         
         tb_sources = gr.Textbox(label='Sources', show_copy_button=True, visible=False)
@@ -397,8 +399,8 @@ with gr.Blocks() as demo:
 
     docs_input.upload(embed_files, inputs=[docs_input,tb_session_id], outputs=[db_output,tb_session_id, query_input])
     btn_reset_db.click(reset_database,inputs=[tb_session_id],outputs=[db_output])
-    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])
+    btn_askGPT.click(ask_gpt, inputs=[query_input, tb_session_id, tb_history], outputs=[answer_output, tb_sources, tb_history, btn1, btn2, btn3, btn4, btn5])
+    query_input.submit(ask_gpt, inputs=[query_input, tb_session_id, tb_history], outputs=[answer_output, tb_sources, tb_history, btn1, btn2, btn3, btn4, btn5])
 
     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])