Update app.py

app.py

@@ -39,7 +39,8 @@ ds = load_data(educational_topic)
 
 with col_2:
     category_types = ['afaik', 'defined_by_llm', 'defined_by_llm_under_afaik']
-    selected_category_type = st.selectbox("Select Category Type", category_types)
+    default_index = 0 if educational_topic == "Yes" else 1
+    selected_category_type = st.selectbox("Select Category Type", category_types, index=default_index)
 with col_3:
     categories = get_categories_by_type(ds, selected_category_type)
     selected_category = st.selectbox("Select Category", categories)
@@ -47,8 +48,19 @@ with col_3:
 selected_cluster = ds.filter(lambda x: x['category'] == selected_category)
 
 # Select sample index
-n_samples = len(selected_cluster["examples"])
-index_example = st.number_input(f"Index of a sample: 0 - {n_samples}",  min_value=0, max_value=n_samples-1, value=0, step=1)
+print(f"selected_cluster is {selected_cluster}")
+n_samples = len(selected_cluster)
+if n_samples > 0:
+    col_1, col_2 = st.columns(2)
+    with col_1:
+        index_cluster = st.number_input(f"Found {len(selected_cluster)} clusters, choose one",  min_value=0, max_value=len(selected_cluster)-1, value=0, step=1)
 
-sample = selected_cluster["examples"][index_example]
-st.markdown(sample)
+    files = selected_cluster[index_cluster]["examples"]
+
+    with col_2:
+        index_example = st.number_input(f"Found {len(files)} files in the cluster, choose one",  min_value=0, max_value=len(files)-1, value=0, step=1)
+
+    sample = files[index_example]
+    st.markdown(sample)
+else:
+    st