Add Population tab and convert .csv to .xml (May need more adjust)

app.py

@@ -27,6 +27,8 @@ from state import (
     updateMultiSelect, store_speaker_clips, register_file, analyze,
     load_demo_single, load_demo_single_sample, load_demo_multi, run_analysis_loop,
     build_all_csv_zip,
+    build_xml_download,
+    build_all_xml_zip,
 )
 
 # ---------------------------------------------------------------------------
@@ -259,8 +261,8 @@ try:
 
     st.header(f"Analysis of file {currFile}")
     TAB_NAMES = ["Speakers & Roles", "Pie Chart", "Sunburst",
-                 "Treemap", "Time Spoken", "Timeline", "Download"]
-    renameTab, pie2, sunburst1, treemap1, bar1, timeline, dataTab = st.tabs(TAB_NAMES)
+                 "Treemap", "Time Spoken", "Timeline", "Population", "Download"]
+    renameTab, pie2, sunburst1, treemap1, bar1, timeline, populationTab, dataTab = st.tabs(TAB_NAMES)
 
     currAnnotation, currTotalTime = st.session_state.results[currFile]
     speakerNames       = currAnnotation.labels()
@@ -340,6 +342,55 @@ try:
     # Tab: Data
     # -----------------------------------------------------------------------
 
+    # -----------------------------------------------------------------------
+    # Tab: Population
+    # -----------------------------------------------------------------------
+
+    with populationTab:
+        st.markdown("Enter the number of students present in each recording.")
+        pop_header_cols = st.columns([4, 2, 1])
+        pop_header_cols[0].markdown("**File Name**")
+        pop_header_cols[1].markdown("**Number of Students**")
+        st.markdown("<hr style='margin-top:2px; margin-bottom:4px;'>",
+                    unsafe_allow_html=True)
+        for fn in st.session_state.file_names:
+            if not (fn in st.session_state.results
+                    and len(st.session_state.results[fn]) == 2):
+                continue
+            plain    = fn.rsplit(".", 1)[0]
+            edit_key = f"pop_editing_{fn}"
+            input_key = f"pop_input_{fn}"
+            current_pop = st.session_state.studentPopulations.get(fn)
+
+            row_cols = st.columns([4, 2, 1])
+            row_cols[0].write(plain)
+
+            if st.session_state.get(edit_key, False):
+                with st.form(key=f"pop_form_{fn}", border=False):
+                    new_val = st.number_input(
+                        "Students", min_value=0, step=1,
+                        value=int(current_pop) if current_pop is not None else 0,
+                        key=input_key, label_visibility="collapsed",
+                    )
+                    c1, c2 = st.columns(2)
+                    if c1.form_submit_button("✓"):
+                        st.session_state.studentPopulations[fn] = int(new_val)
+                        st.session_state[edit_key] = False
+                        st.rerun()
+                    if c2.form_submit_button("✕"):
+                        st.session_state[edit_key] = False
+                        st.rerun()
+            else:
+                row_cols[1].write(str(current_pop) if current_pop is not None else "—")
+                if row_cols[2].button("✎", key=f"pop_edit_btn_{fn}",
+                                      help="Edit student count"):
+                    st.session_state[edit_key] = True
+                    st.rerun()
+
+    # -----------------------------------------------------------------------
+    # Tab: Download
+    # -----------------------------------------------------------------------
+
     with dataTab:
         # Build raw-speaker -> role lookup against the RAW currDF (before renames
         # are applied) so the map keys match the original SPEAKER_## labels.
@@ -356,19 +407,20 @@ try:
         displayDF = displayDF.rename(columns={"Resource": "Speaker"})
         if "Start" in displayDF.columns:
             displayDF = displayDF.sort_values("Start").reset_index(drop=True)
-        csv = convert_df(displayDF)
+
+        xml_bytes = build_xml_download(currFile)
         st.download_button(
-            f"Download {currPlainName}.csv", csv,
-            f"sonogram-analysis-{currPlainName}.csv", "text/csv",
-            key="download-csv", on_click="ignore",
+            f"Download {currPlainName}.xml", xml_bytes,
+            f"sonogram-analysis-{currPlainName}.xml", "application/xml",
+            key="download-xml", on_click="ignore",
         )
 
         analyzed_count = sum(
             1 for r in st.session_state.results.values() if len(r) == 2
         )
-        zip_bytes = build_all_csv_zip() if analyzed_count > 1 else b""
+        zip_bytes = build_all_xml_zip() if analyzed_count > 1 else b""
         st.download_button(
-            "Download all analyzed datas.csv in .zip archive", zip_bytes,
+            "Download all analyzed datas.xml in .zip archive", zip_bytes,
             "sonogram-analysis-all.zip", "application/zip",
             key="download-all-zip", on_click="ignore",
             disabled=(analyzed_count <= 1),