From 79c145073020c8ddddf1b37fff4fac95baabfea0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Nielson=20Jann=C3=A9?= Date: Sun, 13 Apr 2025 15:27:21 +0200 Subject: [PATCH] Shows PDF to the side with a nice link --- generic_rag/app.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/generic_rag/app.py b/generic_rag/app.py index f2e54ae..50ee910 100644 --- a/generic_rag/app.py +++ b/generic_rag/app.py @@ -127,12 +127,9 @@ async def add_sources(chainlit_response: cl.Message, pdf_sources: dict, web_sour if len(pdf_sources) > 0: await chainlit_response.stream_token("\n\nThe following PDF source were consulted:\n") for source, page_numbers in pdf_sources.items(): - page_numbers = list(page_numbers) - page_numbers.sort() - # display="side" seems to be not supported by chainlit for PDF's, so we use "inline" instead. - chainlit_response.elements.append(cl.Pdf(name="pdf", display="inline", path=source, page=page_numbers[0])) - await chainlit_response.update() - await chainlit_response.stream_token(f"- '{source}' on page(s): {page_numbers}\n") + filename = Path(source).name + await chainlit_response.stream_token(f"- {filename} on page(s): {sorted(page_numbers)}\n") + chainlit_response.elements.append(cl.Pdf(name=filename, display="side", path=source, page=sorted(page_numbers)[0])) if len(web_sources) > 0: await chainlit_response.stream_token("\n\nThe following web sources were consulted:\n")