Add document information

app.py

@@ -185,7 +185,7 @@ class ComparisonMenu:
 #
 #
 class ParameterLayout:
-    def __init__(self, name, ci=0.95):
+    def __init__(self, name, ci=0.95, **kwargs):
         self.name = name
         self.ci = ci
         self.summarize = DataSummarizer(self.name, self.ci)
@@ -215,8 +215,8 @@ class ParameterLayout:
         raise NotImplementedError()
 
 class PersonLayout(ParameterLayout):
-    def __init__(self, ci=0.95):
-        super().__init__('ability', ci)
+    def __init__(self, ci=0.95, **kwargs):
+        super().__init__('ability', ci, **kwargs)
 
     def extra(self, df, *args):
         with gr.Row():
@@ -245,18 +245,25 @@ class PersonLayout(ParameterLayout):
                 button.click(menu, inputs=inputs, outputs=[display])
 
 class ItemLayout(ParameterLayout):
+    _prefix = Path('jerome-white', 'leaderboard-documents')
+
+    def __init__(self, name, framework, ci=0.95, **kwargs):
+        super().__init__(name, ci, **kwargs)
+        ds = load_dataset(f'{self._prefix}-{framework}', split='train')
+        self.df = (ds
+                   .to_pandas()
+                   .set_index('doc'))
+
     def extra(self, df, *args):
         (frame, ) = args
         frame.select(self.df_select_callback,
                      inputs=[frame],
                      outputs=[gr.JSON()])
 
-    @staticmethod
-    def df_select_callback(df: pd.DataFrame,  evt: gr.SelectData):
+    def df_select_callback(self, df: pd.DataFrame,  evt: gr.SelectData):
         index = df.columns.get_loc('Source')
-        return {
-            'source': evt.row_value[index],
-        }
+        doc = evt.row_value[index]
+        return self.df.loc[(doc, 'info')]
 
 #
 #
@@ -265,20 +272,18 @@ with gr.Blocks() as demo:
     path = str(Path('jerome-white', 'leaderboard-item-response'))
     splits = get_dataset_split_names(path)
     pmap = {
-        'alpha': ItemLayout('discrimination'),
-        'beta': ItemLayout('difficulty'),
-        'theta': PersonLayout(),
+        'alpha': ft.partial(ItemLayout, name='discrimination'),
+        'beta': ft.partial(ItemLayout, name='difficulty'),
+        'theta': PersonLayout,
     }
 
     for s in splits:
-        data = load_dataset(path, split=s)
-        df = data.to_pandas()
-
+        df = load_dataset(path, split=s).to_pandas()
         with gr.Tab(s):
             for (i, g) in df.groupby('parameter'):
-                layout = pmap[i]
+                layout = pmap.get(i)(framework=s)
                 tab = str(layout)
                 with gr.Tab(tab.title()):
                     layout(g)
 
-demo.launch(server_name='0.0.0.0', debug=True)
+demo.launch()