Update app.py

app.py

@@ -81,14 +81,13 @@ def draw_graph():
     return Image.open(buf)
 
 
-def extraer_informacion_metadata(respuesta, max_results=10):
-    # Obtener source_nodes de la respuesta
-    source_nodes = respuesta.source_nodes
 
+def extraer_informacion_metadata(source_nodes, max_results=10):
     # Obtener page_labels, file_names y scores de source_nodes
     page_file_info = [
-        f"Página {node.node.metadata.get('page_label', '')} del archivo {node.node.metadata.get('file_name', '')} (Relevance: {node.score:.6f} - Id: {node.node.id_})"
+        f"Página {node.node.metadata.get('page_label', '')} del archivo {node.node.metadata['file_name']} (Relevance: {node.score:.2f} - Id: {node.node.id_})"
         for node in source_nodes
+        if 'file_name' in node.node.metadata and node.node.metadata['file_name']
     ]
 
     # Limitar la cantidad de resultados
@@ -96,6 +95,8 @@ def extraer_informacion_metadata(respuesta, max_results=10):
 
     return page_file_info
 
+    
+
 from typing import List
 from llama_index.core import Prompt
 from llama_index.core.llms import ChatMessage, MessageRole
@@ -149,14 +150,19 @@ with gr.Blocks(theme='sudeepshouche/minimalist') as demo:
     def respond(message, chat_history):      
         global chat_history_engine
         global result_metadata
- 
+    
+        # Verificar si el mensaje está vacío o no contiene nada
+        if not message.strip():
+            gr.Info("Escribe un mensaje en el chat")
+            return
+    
         # Si chat_history está vacío, inicialízalo con el mensaje del usuario actual
         if not chat_history:  
             chat_history = [[message, ""]]  
         else:
             # Si chat_history no está vacío, agrega el mensaje actual al final de la lista
             chat_history.append([message, ""])
-
+    
         chat_history_engine = summarize_assistant_messages(chat_history_engine)
         #chat_history_engine.append(ChatMessage(role=MessageRole.USER, content=message))   
         response = chat_engine.stream_chat(message, chat_history=chat_history_engine)
@@ -171,11 +177,9 @@ with gr.Blocks(theme='sudeepshouche/minimalist') as demo:
         for text in response.response_gen:
             chat_history[-1][1] += text  # Agrega el texto de respuesta al último mensaje en chat_history
             yield "", chat_history
-
+    
         print("----------")
         print(memory.get_all())
-        #chat_history_engine.append(ChatMessage(role=MessageRole.ASSISTANT, content=chat_history[-1][1]))
-        #return "", chat_history
         
             
     gr.Markdown("""