Update app.py

app.py

@@ -257,142 +257,117 @@ def update_image(image_url):
     else:
         return None
 
-# Gradio layout setup
+# Función para procesar la entrada del chatbot
+def process_input(message, history):
+    # Implementa tu lógica de procesamiento aquí
+    response = f"Procesando: {message}"
+    history.append((message, response))
+    return history, history, None
+
+# Función para enviar una pregunta preestablecida
+def send_preset_question(question, history):
+    return process_input(question, history)
+
+# Lista ampliada de categorías y preguntas
+categories = [
+    ("Diagnósticos", [
+        "¿Cuál es el diagnóstico más común registrado en el dataset?",
+        "¿Cuáles son los diagnósticos más frecuentes que requieren hospitalización prolongada?",
+        "¿Qué tan frecuentes son los diagnósticos relacionados con nacimientos prematuros?",
+        "¿Cuál es la incidencia de diagnósticos graves como el 'Síndrome de Dificultad Respiratoria' (P220)?",
+        "¿Cuáles son los códigos CIE-10 relacionados con infecciones neonatales?",
+        "¿Qué diagnósticos están clasificados bajo el código P00-P96 en neonatos?",
+        "¿Cuáles son los diagnósticos menos comunes en el dataset?",
+        "¿Cómo se distribuyen los diagnósticos entre pacientes masculinos y femeninos?",
+        "¿Existen diferencias significativas en los diagnósticos entre diferentes departamentos?"
+    ]),
+    ("Medicamentos", [
+        "¿Qué medicamentos son más usados en neonatos?",
+        "¿Cuál es el medicamento más usado para infecciones neonatales?",
+        "¿Qué tipo de forma farmacéutica es la más utilizada en neonatos?",
+        "¿Cuáles son los medicamentos más costosos administrados a recién nacidos?",
+        "¿Qué proporción de medicamentos administrados son antibióticos?",
+        "¿Qué porcentaje del total de medicamentos administrados corresponde a la Tetraciclina Clorhidrato?",
+        "¿Qué medicamentos se utilizan con mayor frecuencia en los departamentos con menos recursos?",
+        "¿Cuáles son los medicamentos menos comunes administrados en el dataset?",
+        "¿Qué relación existe entre el tipo de medicamento y el diagnóstico principal?"
+    ]),
+    ("Estadísticas y Frecuencia", [
+        "¿Qué porcentaje del total de registros corresponde al diagnóstico 'Nacido Vivo Único' (Z370)?",
+        "¿Cuál es la distribución por edad gestacional de los neonatos en el dataset?",
+        "¿Cuál es la tasa de readmisión hospitalaria en neonatos?",
+        "¿Cómo varía la mortalidad neonatal entre diferentes departamentos?",
+        "¿Hay patrones estacionales en la administración de medicamentos?",
+        "¿Cuál es la duración promedio de la hospitalización para los recién nacidos?",
+        "¿Qué proporción de recién nacidos es hospitalizada por más de 7 días?",
+        "¿Cómo varía la duración de la hospitalización entre los diferentes departamentos?",
+        "¿Qué diagnósticos están asociados con hospitalizaciones de corta duración?"
+    ]),
+    ("Impacto y Calidad de Datos", [
+        "¿Cómo impacta el uso de antibióticos en la recuperación neonatal?",
+        "¿Qué mejoras se pueden implementar en la atención neonatal?",
+        "¿Cuál es la efectividad de las medidas preventivas en la reducción de infecciones neonatales?",
+        "¿Cuáles son las limitaciones de los datos sobre salud neonatal?",
+        "¿Qué tan completos y precisos son los registros de diagnóstico?",
+        "¿Cómo se manejaron los valores faltantes en la variable de peso (PESO)?",
+        "¿Qué método se utilizó para imputar valores faltantes en la variable de días de hospitalización?",
+        "¿Cómo se gestionaron los outliers en la variable de peso por género?",
+        "¿Qué recomendaciones se hicieron para mejorar la calidad de los datos en futuros estudios?"
+    ]),
+    ("Análisis Geográfico", [
+        "¿Cuál es el departamento con el mayor número de registros de administración de medicamentos?",
+        "¿Qué departamentos tienen los menores registros de administración de medicamentos?",
+        "¿Cuál es la diferencia en la administración de medicamentos entre Lima y La Libertad?",
+        "¿Cómo se distribuyen los registros de administración de medicamentos en áreas urbanas versus rurales?",
+        "¿Qué departamento muestra la mayor variedad de medicamentos administrados?",
+        "¿Existen diferencias en el costo de los medicamentos entre diferentes provincias dentro de un mismo departamento?",
+        "¿Qué provincia tiene el menor registro de administración de medicamentos?",
+        "¿Cómo varía la administración de medicamentos en los distritos más poblados versus los menos poblados?"
+    ]),
+    ("Análisis Temporal", [
+        "¿Cuál es el rango de fechas de administración de medicamentos registrado en el dataset?",
+        "¿En qué meses se registra la mayor cantidad de administración de medicamentos?",
+        "¿Cuál es la distribución de la administración de medicamentos a lo largo de los días de la semana?",
+        "¿Existen picos en la administración de medicamentos durante períodos específicos?",
+        "¿Cómo ha evolucionado la administración de medicamentos desde 2022 hasta 2023?",
+        "¿Qué impacto tienen las estaciones del año en la administración de medicamentos?",
+        "¿Cómo varía la administración de medicamentos durante días festivos o feriados?",
+        "¿Qué variación se observa en la administración de medicamentos al comparar el primer y el último trimestre de 2023?"
+    ])
+]
+
+# Configuración de la interfaz de Gradio
 with gr.Blocks() as demo:
     with gr.Row():
-        with gr.Column(scale=1):
+        with gr.Column(scale=2):
             chatbot_input = gr.Textbox(label="Tu mensaje")
             chatbot_output = gr.Chatbot(label="ChatBot")
             chatbot_history = gr.State(value=[])
             image_url = gr.State(value=None)
             submit_button = gr.Button("Enviar")
 
-            # Main accordion for categories
-            with gr.Accordion("Categorías de Preguntas", open=True):
-                
-                # 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")
 
-    def process_input(message, history):
-        full_response, image = chat_function(message, history)
-        history.append((message, full_response))
-        return history, history, image
-
-    # Function to update chatbot input with preset question
-    def send_preset_question(question, history):
-        return process_input(question, history)
-
-    # Assign the buttons to send the respective categorized questions
-    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)
-
-# Launch the interface
+    # Crear filas de categorías
+    for i in range(0, len(categories), 3):
+        with gr.Row():
+            for category, questions in categories[i:i+3]:
+                with gr.Column():
+                    with gr.Accordion(category):
+                        for question in questions:
+                            button = gr.Button(question)
+                            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]
+    )
+
+# Lanzar la interfaz
 demo.launch(debug=True)