Update app.py

app.py

@@ -13,13 +13,6 @@ import gradio as gr
 
 model_name = "BAAI/bge-m3"
 
-# APIs personales
-#PINECONE_ENVIRONMENT = us-east-1
-#PINECONE_API_KEY = 3a3e9022-381d-436e-84cb-ba93464d283e
-
-#os.environ["PINECONE_ENVIRONMENT"] = "us-east-1"
-#os.environ["PINECONE_API_KEY"] = "3a3e9022-381d-436e-84cb-ba93464d283e"
-
 # Retrieve the Pinecone API key from the user
 PINECONE_API_KEY = "3a3e9022-381d-436e-84cb-ba93464d283e"  # Use the key you set in the secrets
 PINECONE_ENVIRONMENT = "us-east-1"  # Use the environment you set in the secrets
@@ -274,54 +267,80 @@ with gr.Blocks() as demo:
             image_url = gr.State(value=None)
             submit_button = gr.Button("Enviar")
 
-            # Categorized neonatal-related questions
+            # Main accordion for categories
             with gr.Accordion("Categorías de Preguntas", open=True):
-                with gr.Row():
-                    with gr.Column():
-                        gr.Markdown("**Diagnóstico Principal**")
-                        button_1 = gr.Button("¿Cuáles son los diagnósticos más comunes en neonatos?")
-                        button_2 = gr.Button("¿Cuál es la prevalencia de la sepsis neonatal?")
-
-                    with gr.Column():
-                        gr.Markdown("**Tipo de Medicamento**")
-                        button_3 = gr.Button("¿Qué medicamentos son más usados en neonatos?")
-                        button_4 = gr.Button("¿Cuál es el medicamento más usado para infecciones neonatales?")
-
-                    with gr.Column():
-                        gr.Markdown("**Duración de Hospitalización**")
-                        button_5 = gr.Button("¿Cuál es la duración promedio de hospitalización de neonatos?")
-                        button_6 = gr.Button("¿Qué diagnósticos requieren hospitalizaciones más largas?")
-
-                with gr.Row():
-                    with gr.Column():
-                        gr.Markdown("**Costo de Medicamentos**")
-                        button_7 = gr.Button("¿Cuáles son los medicamentos más costosos en el tratamiento neonatal?")
-
-                    with gr.Column():
-                        gr.Markdown("**Diferencias Geográficas**")
-                        button_8 = gr.Button("¿Cómo varía el acceso a medicamentos entre regiones?")
-
-                    with gr.Column():
-                        gr.Markdown("**Análisis Temporal**")
-                        button_9 = gr.Button("¿Hay patrones estacionales en la administración de medicamentos?")
-
-                with gr.Row():
-                    with gr.Column():
-                        gr.Markdown("**Relación entre Diagnósticos y Medicamentos**")
-                        button_10 = gr.Button("¿Qué medicamentos se usan para tratar sepsis neonatal?")
-
-                    with gr.Column():
-                        gr.Markdown("**Impacto en la Salud Neonatal**")
-                        button_11 = gr.Button("¿Cómo impacta el uso de antibióticos en la recuperación neonatal?")
-
-                    with gr.Column():
-                        gr.Markdown("**Limitaciones en la Calidad de Datos**")
-                        button_12 = gr.Button("¿Cuáles son las limitaciones de los datos sobre salud neonatal?")
-
-                with gr.Row():
-                    with gr.Column():
-                        gr.Markdown("**Recomendaciones y Mejoras**")
-                        button_13 = gr.Button("¿Qué mejoras se pueden implementar en la atención neonatal?")
+                
+                # Diagnósticos Accordion
+                with gr.Accordion("Diagnósticos", open=False):
+                    
+                    # Subaccordion for Diagnósticos Comunes
+                    with gr.Accordion("Diagnósticos Comunes", open=False):
+                        button_1 = gr.Button("¿Cuál es el diagnóstico más común registrado en el dataset?")
+                        button_2 = gr.Button("¿Cuáles son los diagnósticos más frecuentes que requieren hospitalización prolongada?")
+                        button_3 = gr.Button("¿Qué tan frecuentes son los diagnósticos relacionados con nacimientos prematuros?")
+                        button_4 = gr.Button("¿Cuáles son los diagnósticos menos comunes en el dataset?")
+                    
+                    # Subaccordion for Diagnósticos Críticos
+                    with gr.Accordion("Diagnósticos Críticos", open=False):
+                        button_5 = gr.Button("¿Cuál es la incidencia de diagnósticos graves como el 'Síndrome de Dificultad Respiratoria' (P220)?")
+                        button_6 = gr.Button("¿Cuál es la tasa de mortalidad asociada a estos diagnósticos graves?")
+                        button_7 = gr.Button("¿Qué porcentaje de neonatos con sepsis requiere cuidados intensivos?")
+                    
+                    # Subaccordion for Codificación y Clasificación
+                    with gr.Accordion("Codificación y Clasificación", open=False):
+                        button_8 = gr.Button("¿Cuáles son los códigos CIE-10 relacionados con infecciones neonatales?")
+                        button_9 = gr.Button("¿Qué diagnósticos están clasificados bajo el código P00-P96 en neonatos?")
+
+                # Tratamientos Accordion
+                with gr.Accordion("Tratamientos", open=False):
+
+                    # Subaccordion for Medicamentos Comunes
+                    with gr.Accordion("Medicamentos Comunes", open=False):
+                        button_10 = gr.Button("¿Qué medicamentos son más usados en neonatos?")
+                        button_11 = gr.Button("¿Cuál es el medicamento más usado para infecciones neonatales?")
+                        button_12 = gr.Button("¿Cuáles son los efectos secundarios más comunes de estos medicamentos?")
+
+                    # Subaccordion for Intervenciones Médicas
+                    with gr.Accordion("Intervenciones Médicas", open=False):
+                        button_13 = gr.Button("¿Cuál es la proporción de diagnósticos que requieren intervención quirúrgica?")
+                        button_14 = gr.Button("¿Cuáles son las complicaciones más frecuentes durante las intervenciones quirúrgicas en neonatos?")
+                        button_15 = gr.Button("¿Cuál es la tasa de éxito de las intervenciones quirúrgicas en neonatos?")
+
+                # Estadísticas y Frecuencia Accordion
+                with gr.Accordion("Estadísticas y Frecuencia", open=False):
+
+                    # Subaccordion for Estadísticas Generales
+                    with gr.Accordion("Estadísticas Generales", open=False):
+                        button_16 = gr.Button("¿Qué porcentaje del total de registros corresponde al diagnóstico 'Nacido Vivo Único' (Z370)?")
+                        button_17 = gr.Button("¿Cuál es la distribución por edad gestacional de los neonatos en el dataset?")
+                        button_18 = gr.Button("¿Cuál es la tasa de readmisión hospitalaria en neonatos?")
+
+                    # Subaccordion for Diferencias Geográficas
+                    with gr.Accordion("Diferencias Geográficas", open=False):
+                        button_19 = gr.Button("¿Cómo varía el acceso a medicamentos entre regiones?")
+                        button_20 = gr.Button("¿Existen diferencias significativas en las tasas de diagnóstico entre diferentes regiones?")
+                        button_21 = gr.Button("¿Cómo varía la mortalidad neonatal entre diferentes departamentos?")
+
+                    # Subaccordion for Análisis Temporal
+                    with gr.Accordion("Análisis Temporal", open=False):
+                        button_22 = gr.Button("¿Hay patrones estacionales en la administración de medicamentos?")
+                        button_23 = gr.Button("¿Cómo han cambiado las tasas de mortalidad neonatal a lo largo de los años?")
+                        button_24 = gr.Button("¿Existen variaciones en la incidencia de diagnósticos críticos a lo largo del año?")
+
+                # Impacto y Calidad de Datos Accordion
+                with gr.Accordion("Impacto y Calidad de Datos", open=False):
+
+                    # Subaccordion for Impacto en la Salud Neonatal
+                    with gr.Accordion("Impacto en la Salud Neonatal", open=False):
+                        button_25 = gr.Button("¿Cómo impacta el uso de antibióticos en la recuperación neonatal?")
+                        button_26 = gr.Button("¿Qué mejoras se pueden implementar en la atención neonatal?")
+                        button_27 = gr.Button("¿Cuál es la efectividad de las medidas preventivas en la reducción de infecciones neonatales?")
+
+                    # Subaccordion for Calidad de Datos
+                    with gr.Accordion("Calidad de Datos", open=False):
+                        button_28 = gr.Button("¿Cuáles son las limitaciones de los datos sobre salud neonatal?")
+                        button_29 = gr.Button("¿Qué tan completos y precisos son los registros de diagnóstico?")
+                        button_30 = gr.Button("¿Cómo afecta la calidad de los datos a la investigación y atención neonatal?")
 
         with gr.Column(scale=1):
             image_output = gr.Image(label="Imagen asociada")
@@ -336,19 +355,41 @@ with gr.Blocks() as demo:
         return process_input(question, history)
 
     # Assign the buttons to send the respective categorized questions
-    button_1.click(send_preset_question, inputs=[gr.State(value="¿Cuáles son los diagnósticos más comunes en neonatos?"), chatbot_history], outputs=[chatbot_output, chatbot_history, image_url])
-    button_2.click(send_preset_question, inputs=[gr.State(value="¿Cuál es la prevalencia de la sepsis neonatal?"), chatbot_history], outputs=[chatbot_output, chatbot_history, image_url])
-    button_3.click(send_preset_question, inputs=[gr.State(value="¿Qué medicamentos son más usados en neonatos?"), chatbot_history], outputs=[chatbot_output, chatbot_history, image_url])
-    button_4.click(send_preset_question, inputs=[gr.State(value="¿Cuál es el medicamento más usado para infecciones neonatales?"), chatbot_history], outputs=[chatbot_output, chatbot_history, image_url])
-    button_5.click(send_preset_question, inputs=[gr.State(value="¿Cuál es la duración promedio de hospitalización de neonatos?"), chatbot_history], outputs=[chatbot_output, chatbot_history, image_url])
-    button_6.click(send_preset_question, inputs=[gr.State(value="¿Qué diagnósticos requieren hospitalizaciones más largas?"), chatbot_history], outputs=[chatbot_output, chatbot_history, image_url])
-    button_7.click(send_preset_question, inputs=[gr.State(value="¿Cuáles son los medicamentos más costosos en el tratamiento neonatal?"), chatbot_history], outputs=[chatbot_output, chatbot_history, image_url])
-    button_8.click(send_preset_question, inputs=[gr.State(value="¿Cómo varía el acceso a medicamentos entre regiones?"), chatbot_history], outputs=[chatbot_output, chatbot_history, image_url])
-    button_9.click(send_preset_question, inputs=[gr.State(value="¿Hay patrones estacionales en la administración de medicamentos?"), chatbot_history], outputs=[chatbot_output, chatbot_history, image_url])
-    button_10.click(send_preset_question, inputs=[gr.State(value="¿Qué medicamentos se usan para tratar sepsis neonatal?"), chatbot_history], outputs=[chatbot_output, chatbot_history, image_url])
-    button_11.click(send_preset_question, inputs=[gr.State(value="¿Cómo impacta el uso de antibióticos en la recuperación neonatal?"), chatbot_history], outputs=[chatbot_output, chatbot_history, image_url])
-    button_12.click(send_preset_question, inputs=[gr.State(value="¿Cuáles son las limitaciones de los datos sobre salud neonatal?"), chatbot_history], outputs=[chatbot_output, chatbot_history, image_url])
-    button_13.click(send_preset_question, inputs=[gr.State(value="¿Qué mejoras se pueden implementar en la atención neonatal?"), chatbot_history], outputs=[chatbot_output, chatbot_history, image_url])
+    button_assignments = [
+        (button_1, "¿Cuál es el diagnóstico más común registrado en el dataset?"),
+        (button_2, "¿Cuáles son los diagnósticos más frecuentes que requieren hospitalización prolongada?"),
+        (button_3, "¿Qué tan frecuentes son los diagnósticos relacionados con nacimientos prematuros?"),
+        (button_4, "¿Cuáles son los diagnósticos menos comunes en el dataset?"),
+        (button_5, "¿Cuál es la incidencia de diagnósticos graves como el 'Síndrome de Dificultad Respiratoria' (P220)?"),
+        (button_6, "¿Cuál es la tasa de mortalidad asociada a estos diagnósticos graves?"),
+        (button_7, "¿Qué porcentaje de neonatos con sepsis requiere cuidados intensivos?"),
+        (button_8, "¿Cuáles son los códigos CIE-10 relacionados con infecciones neonatales?"),
+        (button_9, "¿Qué diagnósticos están clasificados bajo el código P00-P96 en neonatos?"),
+        (button_10, "¿Qué medicamentos son más usados en neonatos?"),
+        (button_11, "¿Cuál es el medicamento más usado para infecciones neonatales?"),
+        (button_12, "¿Cuáles son los efectos secundarios más comunes de estos medicamentos?"),
+        (button_13, "¿Cuál es la proporción de diagnósticos que requieren intervención quirúrgica?"),
+        (button_14, "¿Cuáles son las complicaciones más frecuentes durante las intervenciones quirúrgicas en neonatos?"),
+        (button_15, "¿Cuál es la tasa de éxito de las intervenciones quirúrgicas en neonatos?"),
+        (button_16, "¿Qué porcentaje del total de registros corresponde al diagnóstico 'Nacido Vivo Único' (Z370)?"),
+        (button_17, "¿Cuál es la distribución por edad gestacional de los neonatos en el dataset?"),
+        (button_18, "¿Cuál es la tasa de readmisión hospitalaria en neonatos?"),
+        (button_19, "¿Cómo varía el acceso a medicamentos entre regiones?"),
+        (button_20, "¿Existen diferencias significativas en las tasas de diagnóstico entre diferentes regiones?"),
+        (button_21, "¿Cómo varía la mortalidad neonatal entre diferentes departamentos?"),
+        (button_22, "¿Hay patrones estacionales en la administración de medicamentos?"),
+        (button_23, "¿Cómo han cambiado las tasas de mortalidad neonatal a lo largo de los años?"),
+        (button_24, "¿Existen variaciones en la incidencia de diagnósticos críticos a lo largo del año?"),
+        (button_25, "¿Cómo impacta el uso de antibióticos en la recuperación neonatal?"),
+        (button_26, "¿Qué mejoras se pueden implementar en la atención neonatal?"),
+        (button_27, "¿Cuál es la efectividad de las medidas preventivas en la reducción de infecciones neonatales?"),
+        (button_28, "¿Cuáles son las limitaciones de los datos sobre salud neonatal?"),
+        (button_29, "¿Qué tan completos y precisos son los registros de diagnóstico?"),
+        (button_30, "¿Cómo afecta la calidad de los datos a la investigación y atención neonatal?"),
+    ]
+
+    for button, question in button_assignments:
+        button.click(send_preset_question, inputs=[gr.State(value=question), chatbot_history], outputs=[chatbot_output, chatbot_history, image_url])
 
     submit_button.click(process_input, inputs=[chatbot_input, chatbot_history], outputs=[chatbot_output, chatbot_history, image_url])
     image_url.change(fn=update_image, inputs=image_url, outputs=image_output)