diff --git a/libs/ktem/ktem/assets/css/main.css b/libs/ktem/ktem/assets/css/main.css index 4fe2019d..c722a6df 100644 --- a/libs/ktem/ktem/assets/css/main.css +++ b/libs/ktem/ktem/assets/css/main.css @@ -105,6 +105,10 @@ mark { background-color: #1496bb; } +mark.warning { + background-color: rgb(216 120 104); +} + /* clpse */ .clpse { background-color: var(--background-fill-secondary); diff --git a/libs/ktem/ktem/pages/chat/verification.py b/libs/ktem/ktem/pages/chat/verification.py index aea3c470..74d9620b 100644 --- a/libs/ktem/ktem/pages/chat/verification.py +++ b/libs/ktem/ktem/pages/chat/verification.py @@ -43,7 +43,10 @@ class VerificationPage(BasePage): highlighted_text = text[: spans[0]["start"]] for idx, span in enumerate(spans): to_highlight = text[span["start"] : span["end"]] - highlighted_text += Render.highlight(to_highlight) + highlighted_text += Render.highlight( + to_highlight, + elem_classes="warning", + ) if idx < len(spans) - 1: highlighted_text += text[span["end"] : spans[idx + 1]["start"]] highlighted_text += text[spans[-1]["end"] :] @@ -86,9 +89,16 @@ class VerificationPage(BasePage): for claim in result["ungroundedDetails"]: rationale += Render.collapsible_with_header( - Document(text=claim["reason"], metadata={"file_name": claim["text"]}) + Document( + text="{}".format(claim["reason"]), + metadata={ + "file_name": "{}".format( + claim["text"] + ), + }, + ) ) - verification_output += f"
{rationale}
" + verification_output += f"
{rationale}
" return gr.update(visible=True), verification_output diff --git a/libs/ktem/ktem/utils/render.py b/libs/ktem/ktem/utils/render.py index eeb4fc27..0d4f1d2e 100644 --- a/libs/ktem/ktem/utils/render.py +++ b/libs/ktem/ktem/utils/render.py @@ -101,9 +101,10 @@ class Render: """ # noqa @staticmethod - def highlight(text: str) -> str: + def highlight(text: str, elem_classes: str | None = None) -> str: """Highlight text""" - return f"{text}" + class_str = "" if not elem_classes else f' class="{elem_classes}"' + return f"{text}" @staticmethod def image(url: str, text: str = "") -> str: