Update app.py

app.py

@@ -31,35 +31,23 @@ def obtener_extractos(pregunta):
     return [(doc.page_content, doc.metadata.get("url", "URL no disponible")) for doc in docs_relevantes]
 
 def extract_unique_citations_paragraph(response):
-    """
-    Extrae todas las citas URL únicas de un objeto Response
-    y devuelve un párrafo de texto con formato legible, separado por puntos.
-    """
-    seen_urls = set()
-    fragments = []
+    """Extrae todas las URL y títulos de un objeto Response."""
+    citations = []
 
     for item in getattr(response, "output", []):
+        # Solo buscamos en mensajes del asistente
         if getattr(item, "type", None) == "message":
             for block in getattr(item, "content", []) or []:
                 if getattr(block, "type", None) == "output_text":
                     for ann in getattr(block, "annotations", []) or []:
                         if getattr(ann, "type", None) == "url_citation":
-                            url = getattr(ann, "url", None)
-                            title = getattr(ann, "title", None)
-                            if url and url not in seen_urls:
-                                seen_urls.add(url)
-                                # Construir una frase tipo "Título (URL)"
-                                if title:
-                                    fragments.append(f"{title} ({url})")
-                                else:
-                                    fragments.append(f"{url}")
-
-    # Unir todas las frases en un párrafo separado por puntos
-    paragraph = ". ".join(fragments)
-    if paragraph and not paragraph.endswith("."):
-        paragraph += "."
-
-    return paragraph
+                            citations.append({
+                                "title": getattr(ann, "title", None),
+                                "url": getattr(ann, "url", None),
+                                "start": getattr(ann, "start_index", None),
+                                "end": getattr(ann, "end_index", None)
+                            })
+    return citations
 
 
 
@@ -162,8 +150,10 @@ def respond(message, history: list[tuple[str, str]], domain_table):
             elif event.type == "response.completed":
                 response_stream = stream.get_final_response()
                 citations = extract_unique_citations_paragraph(response_stream)
-                response += "\n\nFuentes:\n" + citations
-                yield response
+                response += "\n\n🔗 Fuentes citadas: \n"
+                for c in citations:
+                    response += f"• {c['title'] or 'Sin título'} → {c['url']}")
+                    yield response
                 break
 
 
@@ -184,10 +174,11 @@ def add_domain(domains, new_domain):
     return domains, ""  # limpia el textbox
 
 with gr.Blocks() as demo:
-    gr.Markdown("### 🧠 Buscador especializado en investigación médica")
+    gr.Markdown("### 🧠 Chat Médico AI")
+    gr.Markdown("Este es un chatbot especializdo en la búsqueda de información médica con fuentes verídicas. Puedes colocar que busque sólo en dominios específicos.")
 
     with gr.Row():
-        new_domain = gr.Textbox(label="Agregar dominio")
+        new_domain = gr.Textbox(label="Búscar en dominios específicos (opcional)")
         add_button = gr.Button("➕ Añadir")
     
     domain_table = gr.Dataframe(
@@ -203,8 +194,6 @@ with gr.Blocks() as demo:
         fn=respond,
         chatbot=custom_chatbot,   
         additional_inputs=[domain_table],
-        title="🧠 Buscador especializado en investigación médica",
-        description="Agrega tus dominios arriba para limitar la búsqueda"
     )
 
     add_button.click(add_domain, [domain_table, new_domain], [domain_table, new_domain])