Update ui.py

ui.py

@@ -16,7 +16,6 @@ def get_theme():
 def load_images():
     image_carousel_data = {
         "NASA": [
-            {"image": "images/rld0.png", "description": "NASA"},
             {"image": "images/rld1.png", "description": "NASA"},
             {"image": "images/rld2.png", "description": "NASA"},
         ],
@@ -29,7 +28,7 @@ def build_interface(process_input, send_preset_question, update_image):
     theme = get_theme()
     image_carousel_data = load_images()
 
-    with gr.Blocks(theme=theme) as demo:
+    with gr.Blocks(theme='upsatwal/mlsc_tiet') as demo:
         with gr.Row():
             with gr.Column(scale=0.8):
                 # Add the video
@@ -80,7 +79,8 @@ def build_interface(process_input, send_preset_question, update_image):
                         )
 
                     with gr.Column(scale=1):
-                        image_output = gr.Image(label="Imagen asociada", interactive=False)
+                        image_url = gr.State(value=None)
+                        image_output = gr.Image(label="Imagen asociada")
 
                         # Define processing functions
                         def on_submit(message, history, selected_option, similarity_threshold, system_prompt, max_length):
@@ -91,17 +91,10 @@ def build_interface(process_input, send_preset_question, update_image):
 
                         # Configure click events for the chatbot
                         submit_button.click(
-                            fn=on_submit,
+                            on_submit,
                             inputs=[chatbot_input, chatbot_history, selection, similarity_threshold_slider, system_prompt_input, max_length_slider],
-                            outputs=[chatbot_output, chatbot_history, image_output]
+                            outputs=[chatbot_output, chatbot_history, image_url]
                         )
+                        image_url.change(fn=update_image, inputs=image_url, outputs=image_output)
 
-                        # Opcional: Manejar preguntas predefinidas
-                        preset_question = gr.Button("Pregunta Predefinida")
-                        preset_question.click(
-                            fn=send_preset_question,
-                            inputs=[preset_question, chatbot_history, selection, similarity_threshold_slider, system_prompt_input, max_length_slider],
-                            outputs=[chatbot_output, chatbot_history, image_output]
-                        )
-
-        return demo
+    return demo