Update app.py

app.py

@@ -33,9 +33,12 @@ from annotation_questions import (
 ########################################################################################
 
 # Function to get user ID from URL
-def get_param_from_url(param):
-    user_id = st.query_params.get(param, "")
-    return user_id
+def get_param_from_url(param: str, default: str = ""):
+    return st.query_params.get(param, default)
+    
+#def get_param_from_url(param):
+#    user_id = st.query_params.get(param, "")
+#    return user_id
 
 # 'local' or 'hf'. hf is for Hugging Face file system but has limits on the number of access per hour
 filesystem = 'hf'
@@ -342,7 +345,7 @@ def show_field(f: Field, index: int, data_collected):
                 #    )
 
                 if f.name == 'dialogue_name' and value:
-                    render_markdown_simple(os.path.join(input_repo_path, 'dialogues', value), height_px=720)
+                    render_markdown_simple(os.path.join(input_repo_path, 'dialogues', st.session_state.batch, value), height_px=720)
                 
                 elif f.name == 'role_name' and value:
                     render_markdown_simple(os.path.join(input_repo_path, 'role_descriptions', value), height_px=520)
@@ -359,7 +362,7 @@ def show_field(f: Field, index: int, data_collected):
                 if f.other_params and f.other_params.get("use_dialogue_name"):
                     row = st.session_state.data.iloc[index]
                     dialogue_name = row["dialogue_name"]
-                    md_path = os.path.join(input_repo_path, "dialogues", dialogue_name)
+                    md_path = os.path.join(input_repo_path, "dialogues", st.session_state.batch, dialogue_name)
                     content = load_text(md_path)
                     st.markdown(content, unsafe_allow_html=True)
                 elif f.other_params and f.other_params.get("use_roledesc_name"):
@@ -558,6 +561,7 @@ def prep_and_save_data(index, skip_sample, completed: bool):
         **existing,
         **base,
         'user_id': st.session_state.user_id,
+        'batch': st.session_state.batch,
         'index': st.session_state.current_index,
         'skip': skip_sample,
         'completed': int(completed),
@@ -624,7 +628,12 @@ if 'data' not in st.session_state:
 user_id_from_url = get_param_from_url("user_id")
 if user_id_from_url and not st.session_state.get("user_id"):
     st.session_state.user_id = user_id_from_url
-    
+
+# batch
+batch_from_url = get_param_from_url("batch")
+if batch_from_url and not st.session_state.get("batch"):
+    st.session_state.batch = batch_from_url
+
 # current index
 if 'current_index' not in st.session_state:
     start_index = get_start_index()
@@ -679,7 +688,7 @@ if st.session_state.current_index == -2:
         add_checked_submit()
 
 elif st.session_state.current_index == -1:
-    if st.session_state.get('user_id'):
+    if st.session_state.get('user_id') and st.session_state.get("batch"):
         navigate(1)
     else:
         with st.form("user_id_form"):