annotator update

app.py

@@ -28,7 +28,7 @@ if "annotator" not in state:
     for idx, option in enumerate(OPTIONS):
         c[idx].button(f"{option}", on_click=annotate, args=(option,))
 
-if "annotations" not in state:
+if "annotations" not in state and "annotator" in state:
     state.annotations = {}
     state.annotations['data_id', 'index', 'output', 'model', 'is_faithfull'] = ''
     f_0 = open(source_path)
@@ -66,8 +66,9 @@ def submit(index_0, index_1, index_2):
         state.submit = 0
 
 
-if state.files:
+if state.files and "annotator" in state:
     st.header("Dataset annotation")
+    st.header(state.annotator)
     selected_file = state.current_file
     # source_file = selected_file[0] 
     # ori_file = selected_file[1] 
@@ -80,6 +81,7 @@ if state.files:
         state.indexes = [1, 2, 3]
         random.shuffle(state.indexes)
 
+    st.write('-'*50)
     st.write(f"Output: {selected_file[state.indexes[0]]}")
     # c = st.columns(len(OPTIONS))
     state.a = st.radio('a', OPTIONS, key=f'{0}')