Daniel00611 commited on
Commit
a294ac4
verified
1 Parent(s): c9bd98a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -2
app.py CHANGED
@@ -36,9 +36,20 @@ vectorstore = Chroma(
36
  retriever = vectorstore.as_retriever()
37
 
38
  def obtener_extractos(pregunta: str):
39
- """Obtiene documentos relevantes desde ChromaDB"""
40
  docs_relevantes = retriever.invoke(pregunta)
41
- return [(doc.page_content, doc.metadata.get("url", "URL no disponible")) for doc in docs_relevantes]
 
 
 
 
 
 
 
 
 
 
 
42
 
43
  def enviar_contacto(access_token: str, phone_number_id: str, recipient_number: str, formatted_name: str, first_name: str):
44
  url = f"https://graph.facebook.com/v19.0/{phone_number_id}/messages"
 
36
  retriever = vectorstore.as_retriever()
37
 
38
  def obtener_extractos(pregunta: str):
39
+ """Obtiene documentos relevantes desde ChromaDB y los formatea como texto plano"""
40
  docs_relevantes = retriever.invoke(pregunta)
41
+
42
+ if not docs_relevantes:
43
+ return "No se encontr贸 informaci贸n relevante en la base de datos."
44
+
45
+ extractos = []
46
+ for i, doc in enumerate(docs_relevantes, start=1):
47
+ contenido = doc.page_content.strip()
48
+ url = doc.metadata.get("url", "URL no disponible")
49
+ extractos.append(f"馃敼 Extracto {i}:\n{contenido}\n馃敆 Fuente: {url}")
50
+
51
+ return "\n\n".join(extractos)
52
+
53
 
54
  def enviar_contacto(access_token: str, phone_number_id: str, recipient_number: str, formatted_name: str, first_name: str):
55
  url = f"https://graph.facebook.com/v19.0/{phone_number_id}/messages"