Update app.py

app.py

@@ -273,6 +273,56 @@ with gr.Blocks() as demo:
             chatbot_history = gr.State(value=[])
             image_url = gr.State(value=None)
             submit_button = gr.Button("Enviar")
+
+            # Categorized neonatal-related questions
+            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?")
+
         with gr.Column(scale=1):
             image_output = gr.Image(label="Imagen asociada")
 
@@ -281,6 +331,25 @@ with gr.Blocks() as demo:
         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_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])
+
     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)