Update app.py

app.py

@@ -8,6 +8,13 @@ from config import SIMILARITY_THRESHOLD_DEFAULT, SYSTEM_PROMPT, MAX_LENGTH_DEFAU
 from decorators import gpu_decorator
 import torch
 
+########################
+
+from utils import process_tags_chat
+
+########################
+
+
 # Cargar modelos
 embedding_model = load_embedding_model()
 tokenizer, yi_coder_model, yi_coder_device = load_yi_coder_model()
@@ -90,11 +97,15 @@ def process_input(message, history, selected_option, similarity_threshold, syste
     history.append((message, response))
     return history, history, image
 
-def update_image(image_url):
-    if image_url:
-        return image_url
-    else:
-        return None
+def update_image(message, history):
+    # Realizar búsqueda vectorial
+    search_results = vector_search(message, embedding_model, index)
+
+    # Llamar a process_tags_chat para procesar los resultados y obtener la imagen
+    full_response, image_url = process_tags_chat(search_results)
+
+    return image_url if image_url else None
+
 
 def send_preset_question(question, history, selected_option, similarity_threshold, system_prompt, max_length):
     return process_input(question, history, selected_option, similarity_threshold, system_prompt, max_length)