some update for annotators

app.py

@@ -14,22 +14,36 @@ CONTEXT_SENT_1_h_1_RES = f'DUC/output_dir/sent_window_1_h_1_clusters/{generated_
 source_path = 'DUC/sent_window_1_h_1_clusters/test.source'
 
 OPTIONS = ["faithfull", "Not faithfull"]
+Annotators = ['Ruben', 'Arie']
+annotators_dic = {'Ruben':10, 'Arie':60}
 
 if "number_button_fill" not in state:
     state.number_button_fill = 0
 
+def annotate(annotator):
+    state.annotator = annotator
+
+if "annotator" not in state:
+    c = st.columns(len(Annotators))
+    for idx, option in enumerate(OPTIONS):
+        c[idx].button(f"{option}", on_click=annotate, args=(option,))
+
 if "annotations" not in state:
     state.annotations = {}
     state.annotations['data_id', 'index', 'output', 'model', 'is_faithfull'] = ''
     f_0 = open(source_path)
-    source = f_0.read().split('\n')[:N]
+    source_ = f_0.read().split('\n')
+    source = source_[:N] + source_[annotators_dic[state.annotators]:annotators_dic[state.annotators]+50]
     f_1 = open(ORI_RES)
-    ori_res = f_1.read().split('\n')[:N]
+    ori_res_ = f_1.read().split('\n')
+    ori_res = ori_res_[:N] + ori_res_[annotators_dic[state.annotators]:annotators_dic[state.annotators]+50]
     # f_2 = open(CONTEXT_SENT_0_h_0_RES)
     f_3 = open(CONTEXT_SENT_0_h_1_RES)
-    sent_0_h_1 = f_3.read().split('\n')[:N]
+    sent_0_h_1_ = f_3.read().split('\n')
+    sent_0_h_1 = sent_0_h_1_[:N] + sent_0_h_1_[annotators_dic[state.annotators]:annotators_dic[state.annotators]+50]
     f_4 = open(CONTEXT_SENT_1_h_1_RES)
-    sent_1_h_1 = f_4.read().split('\n')[:N]
+    sent_1_h_1_ = f_4.read().split('\n')
+    sent_1_h_1 = sent_1_h_1_[:N] + sent_1_h_1_[annotators_dic[state.annotators]:annotators_dic[state.annotators]+50]
     state.files = list(zip(source, ori_res, sent_0_h_1, sent_1_h_1))
     state.current_file = state.files[0]
     state.counter = 0
@@ -51,9 +65,9 @@ def submit(index_0, index_1, index_2):
         state.counter += 1
         state.submit = 0
 
-st.header("Dataset annotation")
 
 if state.files:
+    st.header("Dataset annotation")
     selected_file = state.current_file
     # source_file = selected_file[0] 
     # ori_file = selected_file[1]