add context/code titles

app.py

@@ -36,14 +36,11 @@ class App:
     def render_header(self, *args, **kwargs):
         pass
 
-    def render_body(self, *args, **kwargs):
-        data = self.data.train
-        st.write(f"> Train data `{data.shape[0]}` rows")
-        filter_ = st.text_input("search phrases", "")
-        if filter_:
-            data = search_df(self.data.train, filter_)
-        st.write(data)
+    @staticmethod
+    def render_body(*args, **kwargs):
+        Helper().display_train_data()
 
+        st.header("Visualize Phrases as a Network Graph")
         Helper().visualize()
 
     def render_footer(self, *args, **kwargs):
@@ -51,6 +48,18 @@ class App:
 
 
 class Helper(App):
+
+    def display_train_data(self):
+        data = self.data.train
+
+        data = self._add_code_title(data)
+
+        st.write(f"> Train data `{data.shape[0]}` rows")
+        filter_ = st.text_input("search phrases", "")
+        if filter_:
+            data = search_df(self.data.train, filter_)
+        st.write(data)
+
     def visualize(self, *args, **kwargs):
 
         data = self.data.train_kg
@@ -102,6 +111,14 @@ class Helper(App):
             g.add_edge(n1, n2, title=label, show_edge_weights=True)  # weight 42
         return g
 
+    @st.cache
+    def _add_code_title(self, data):
+        titles = self.data.titles
+        data = data.merge(titles, left_on="context", right_on="code", how="left")
+        cols = data.columns.tolist()[:7]
+        cols.remove("code")
+        return data[cols]
+
 
 if __name__ == "__main__":
     app = App()