Update app.py

app.py

@@ -97,23 +97,24 @@ def process_input(message, history, selected_option, similarity_threshold, syste
     history.append((message, response))
     return history, history, image
 
-def update_image(message, history):
-    # Realizar búsqueda vectorial
-    search_results = vector_search(message, embedding_model, index)
+def update_image(image_url):
+    """
+    Retorna los datos de la imagen para ser mostrados en Gradio.
 
-    # Llamar a process_tags_chat para procesar los resultados y obtener la imagen
-    full_response, image_url = process_tags_chat(search_results)
-
-    # Depuración: Imprimir los resultados
-    print("Full Response:", full_response)
-    print("Image URL:", image_url)
+    Args:
+        image_url (str): Ruta de la imagen.
 
+    Returns:
+        bytes o None: Datos de la imagen si existe, de lo contrario None.
+    """
     if image_url and os.path.exists(image_url):
-        # Retornar la ruta de la imagen relativa a app.py
-        print(f"Mostrando imagen: {image_url}")
-        # Convertir la ruta a relativa
-        relative_path = os.path.relpath(image_url, os.path.dirname(os.path.abspath(__file__)))
-        return relative_path
+        try:
+            with open(image_url, "rb") as img_file:
+                image_data = img_file.read()
+            return image_data
+        except Exception as e:
+            print(f"Error al leer la imagen: {e}")
+            return None
     else:
         print("No se encontró una imagen válida.")
         return None