Update secretlanguage.py

secretlanguage.py

@@ -1,6 +1,6 @@
 import param
 import panel as pn
-pn.extension(sizing_mode="stretch_width")
+pn.extension(sizing_mode="stretch_width", notifications=True)
 import pickle
 import pandas as pd
 
@@ -38,26 +38,52 @@ if (document.readyState === "complete") {
 import pandas as pd
 
 class ReactiveTables(param.Parameterized):
-    datas = pickle.load(open('all_secret_language.pkl', 'rb'))
-    name_ = param.String('deliberately')
-
     
+    name_ = param.String('Asian')
+    record_start_index = param.Integer(1)
+    record_stop_index = param.Integer(200)
+    old_data = None
+    datas = None
     
-    @param.depends('name_')
+    @param.depends('name_', 'record_start_index', 'record_stop_index')
     def data(self):
-        return pd.DataFrame(self.datas[self.name_])
+        print('new name', self.name_)
+        if len(self.name_) > 0:
+            if ord(self.name_[0]) in list(range(48, 57)):
+                file_name = 'num_dict.pkl'
+            elif ord(self.name_[0]) in list(range(97, 122)) + list(range(65, 90)):
+                file_name = f'{ord(self.name_[0])}_dict.pkl'
+            else:
+                file_name = 'other_dict.pkl'
+        print(f'all_secret_langauge_by_fist/{file_name}')
+        self.datas = pickle.load(open(f'all_secret_langauge_by_fist/{file_name}', 'rb'))
+        
+        if self.name_ in self.datas.keys():
+            self.old_data = pd.DataFrame(self.datas[self.name_])
+        return self.old_data
 
     @param.depends('name_')
     def summary(self):
         return self.data().describe()
     
-#     @param.depends('data', 'stop')
-#     def table(self):
-#         return self.data()[self.start:self.stop]
-    
-    @param.depends('name_')
+    @param.depends('name_', 'record_start_index', 'record_stop_index')
     def table_ours(self):
-        return pn.pane.HTML(self.data().to_html(classes=['example', 'panel-df']) + script, 
+        data = self.data()
+        return pn.pane.HTML(data[max(0, self.record_start_index):
+                                        min(self.record_stop_index + 1, len(data))].to_html(
+            classes=['example', 'panel-df']) + script, 
+                                   sizing_mode='stretch_width')
+    @param.depends('name_', 'record_start_index', 'record_stop_index')
+    def notification(self):
+        if self.datas:
+            if self.name_ in self.datas.keys():
+                return pn.pane.HTML(f'<h1>Found {self.name_}</h1>' ,
+                                   sizing_mode='stretch_width')
+            else:
+                return pn.pane.HTML(f'<h1>Error. {self.name_} not in the dict.</h1>' ,
+                                   sizing_mode='stretch_width')
+        else:
+            return pn.pane.HTML('<h1>initing</h1>' ,
                                    sizing_mode='stretch_width')
     
     
@@ -65,6 +91,7 @@ class ReactiveTables(param.Parameterized):
         return pn.Row(
             pn.Param(self, name="Settings", width=300, sizing_mode="fixed"),
             pn.Column(
+                self.notification,
                 "## Description", self.summary, 
                 "## Table", self.table_ours, 
                 width=2000, 
@@ -77,7 +104,7 @@ component = ReactiveTables().panel()
 
 pn.template.FastListTemplate(site="ACL 23 Submission", title="Finding Secret Language of Language Models", 
                              main=[
-                                 "This page presents some secret languages discovered by our proposed SecretFinding algorithm on three multi-sentence datasets. To ensure optimal performance of this webapp, we recommend presenting less than 2000 data at a time.",
+                                 "This page presents all secret languages discovered by our proposed SecretFinding algorithm on three multi-sentence datasets. To ensure optimal performance of this webapp, at a time at most 2000 entries can be showed. \n``Name'' is the word that you want to find secret languages for. record start index and record stop index mark the start and end indices of the entries.",
                                  component 
                              ]).servable()