From 25bc544854ea1d84014a38fd84020883fbec85f1 Mon Sep 17 00:00:00 2001
From: Alexander Druz <alexandersdruz@gmail.com>
Date: Mon, 20 Nov 2023 15:05:57 +0100
Subject: [PATCH] Add ROUGE score lens to python API

---
 renumics/spotlight/layout/lenses.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/renumics/spotlight/layout/lenses.py b/renumics/spotlight/layout/lenses.py
index e31c604b..17c9ee55 100644
--- a/renumics/spotlight/layout/lenses.py
+++ b/renumics/spotlight/layout/lenses.py
@@ -232,3 +232,12 @@ def spectrogram(
         name=name,
         settings={"freqScale": frequency_scale, "ampScale": amplitude_scale},
     )
+
+
+def rouge_score(column: str, reference_column: str, name: Optional[str] = None) -> Lens:
+    """
+    Add ROUGE score viewer to Spotlight inspector widget.
+
+    Supports a pair of string columns.
+    """
+    return Lens(type="RougeScoreView", columns=[column, reference_column], name=name)