Update ui.py

ui.py

@@ -4,26 +4,26 @@ import gradio as gr
 from config import SIMILARITY_THRESHOLD_DEFAULT, SYSTEM_PROMPT, MAX_LENGTH_DEFAULT
 import os
 
-# Definir el tema (puedes mantener tu tema personalizado)
+# Define the theme (you can keep your custom theme)
 def get_theme():
     theme = gr.themes.Default(
         primary_hue="indigo",
-        # ... (resto de la configuración de tu tema)
+        # ... (rest of your theme configuration)
     )
     return theme
 
-# Cargar imágenes y descripciones
+# Load images and descriptions
 def load_images():
     image_carousel_data = {
         "Análisis Geográfico": [
-            {"image": "images/rId101.png", "description": "Descripción de la imagen 1"},
-            {"image": "images/rId105.png", "description": "Descripción de la imagen 2"},
+            {"image": "images/rId101.png", "description": "📊 Lima (150101) has the highest number of records..."},
+            {"image": "images/rId105.png", "description": "🏥 Level 3 establishments are the most represented..."},
         ],
-        # Añade más categorías y sus imágenes según necesites
+        # Add more categories and images as needed
     }
     return image_carousel_data
 
-# Construir la interfaz
+# Build the interface
 def build_interface(process_input, send_preset_question, update_image):
     theme = get_theme()
     image_carousel_data = load_images()
@@ -31,20 +31,25 @@ def build_interface(process_input, send_preset_question, update_image):
     with gr.Blocks(theme=theme) as demo:
         with gr.Row():
             with gr.Column(scale=1):
-                # Agregar el video
+                # Add the video
                 video = gr.Video(value="video.mp4", label="Video de Introducción")
 
-                # Carruseles de imágenes
+                # Image Galleries
                 gr.Markdown("### Carruseles de Imágenes")
                 for category, images_list in image_carousel_data.items():
                     gr.Markdown(f"#### {category}")
-                    with gr.Carousel():
-                        for item in images_list:
-                            with gr.Card():
-                                gr.Image(value=item["image"])
-                                gr.Markdown(item["description"])
+                    # Use gr.Gallery instead of gr.Carousel
+                    gallery_items = []
+                    for item in images_list:
+                        gallery_items.append((item["image"], item["description"]))
+                    gr.Gallery(
+                        value=gallery_items,
+                        label=category,
+                        show_label=False,
+                        elem_id=None
+                    )
 
-                # Botón de descarga
+                # Download button
                 download_button = gr.File(label="Descargar Informe", value="Reporte.pdf")
 
                 # Chatbot
@@ -55,7 +60,7 @@ def build_interface(process_input, send_preset_question, update_image):
                         submit_button = gr.Button("Enviar")
                         chatbot_history = gr.State(value=[])
 
-                        # Añadir opciones de selección
+                        # Add selection options
                         selection = gr.Radio(
                             ["Solo Búsqueda Vectorial", "Solo Yi-Coder", "Ambos (basado en umbral de similitud)"],
                             label="Seleccione el modo de búsqueda",
@@ -77,22 +82,19 @@ def build_interface(process_input, send_preset_question, update_image):
                         image_url = gr.State(value=None)
                         image_output = gr.Image(label="Imagen asociada")
 
-                    # Aquí puedes incluir tus botones y categorías
-                    # ...
+                        # Define processing functions
+                        def on_submit(message, history, selected_option, similarity_threshold, system_prompt, max_length):
+                            history, new_history, image = process_input(
+                                message, history, selected_option, similarity_threshold, system_prompt, max_length
+                            )
+                            return history, new_history, image
 
-                    # Definir las funciones de procesamiento
-                    def on_submit(message, history, selected_option, similarity_threshold, system_prompt, max_length):
-                        history, new_history, image = process_input(
-                            message, history, selected_option, similarity_threshold, system_prompt, max_length
+                        # Configure click events for the chatbot
+                        submit_button.click(
+                            on_submit,
+                            inputs=[chatbot_input, chatbot_history, selection, similarity_threshold_slider, system_prompt_input, max_length_slider],
+                            outputs=[chatbot_output, chatbot_history, image_url]
                         )
-                        return history, new_history, image
-
-                    # Configurar eventos de clic para el chatbot
-                    submit_button.click(
-                        on_submit,
-                        inputs=[chatbot_input, chatbot_history, selection, similarity_threshold_slider, system_prompt_input, max_length_slider],
-                        outputs=[chatbot_output, chatbot_history, image_url]
-                    )
-                    image_url.change(fn=update_image, inputs=image_url, outputs=image_output)
+                        image_url.change(fn=update_image, inputs=image_url, outputs=image_output)
 
     return demo