Update app.py

app.py

@@ -481,8 +481,33 @@ table {
     font-size: x-small !important;
 }
 """
+
+choices_with_tools = [
+    ("[1] Guía BPL", retriever_1_tool),
+    ("[2] Guía BPM Medicamentos", retriever_2_tool),
+    ("[3] MANUAL DE NORMAS TÉCNICAS DE CALIDAD", retriever_3_tool),
+    ("[4] Resolución 1160 - GMP - 2016", retriever_4_tool),
+#    ("[5] Resolución 2266 - DECRETO FITOTERAPÉUTICOS - 2004", retriever_5_tool),
+    ("[6] Resolución 3619 - GLP - 2013", retriever_6_tool),
+    ("[7] Resolución 005107 - FITOTERAPEUTICOS - 2005", retriever_7_tool),    
+    ("[8] Resolución 3690 - 2016", retriever_8_tool),
+    ("[9] Resolución 3157 - 2018", retriever_9_tool),
+    ("[10] Preguntas y respuestas Res. 3157 - 2018", retriever_10_tool),
+    ("General", retriever_all),
+    
+]
+
+# Solo extraer los nombres para mostrarlos en la interfaz
+selected_choices = [label for label, _ in choices_with_tools]
+choice_labels = [label for label, _ in choices_with_tools]
+
 with gr.Blocks(theme='sudeepshouche/minimalist', css=css) as demo:
 
+    # Actualizar las choices seleccionadas
+    def update_selected_choices(choices):
+        global selected_choices
+        selected_choices = choices
+    
     def get_ref():
         return {mkdn: gr.Markdown(result_metadata), texts: gr.HTML(str(result_texts))}
 
@@ -559,6 +584,7 @@ with gr.Blocks(theme='sudeepshouche/minimalist', css=css) as demo:
         global result_texts
         global final_response
         global query
+        global selected_choices
 
         # Inicializar el historial de chat si está vacío con el mensaje del usuario actual
         if not chat_history:
@@ -570,6 +596,9 @@ with gr.Blocks(theme='sudeepshouche/minimalist', css=css) as demo:
         # Resumir los mensajes previos en chat_history_engine
         chat_history_engine = summarize_assistant_messages(chat_history_engine)
 
+        # Engine:
+        exec(os.environ.get('engine'))
+        
         # Generar la respuesta usando el motor de chat
         response = chat_engine.stream_chat(message, chat_history=chat_history_engine)
 
@@ -615,18 +644,9 @@ with gr.Blocks(theme='sudeepshouche/minimalist', css=css) as demo:
 
         with gr.Column():
             with gr.Accordion(elem_classes="accordion", label="Bases de datos del conocimiento", open=False):
-                gr.Markdown("""
-                ###### [1] Guía BPL
-                ###### [2] Guía BPM Medicamentos
-                ###### [3] MANUAL DE NORMAS TÉCNICAS DE CALIDAD
-                ###### [4] Resolución 1160 - GMP - 2016
-                ###### [5] Resolución 2266 - DECRETO FITOTERAPÉUTICOS - 2004
-                ###### [6] Resolución 3619 - GLP - 2013
-                ###### [7] Resolución 005107 - FITOTERAPEUTICOS - 2005
-                ###### [8] Resolución 3690 - 2016
-                ###### [9] Resolución 3157 - 2018
-                ###### [10] Preguntas y respuestas Res. 3157 - 2018
-                """)
+                # CheckboxGroup para seleccionar bases de datos
+                chx = gr.CheckboxGroup(choices=choice_labels, value=choice_labels, label="Bases de datos del conocimiento")
+                chx.select(fn=update_selected_choices, inputs=chx)
 
             with gr.Accordion(elem_classes="accordion", label="Referencias", open=True):
                 mkdn = gr.Markdown()