Update app.py

app.py

@@ -391,81 +391,77 @@ with tab4:
         st.subheader("Decoded String (continuous across rows)")
         st.write(decoded_string)
 
-# Tab 5: Binary → String
 # Tab 5: Binary → String
 with tab5:
     st.header("Decode Binary Labels to String")
 
+    # ========== 32-BIT DECODING ==========
     st.subheader("🔘 Option 1: 32-bit Binary per Row")
     st.write("Upload CSV with 32 columns (0 or 1), no headers, from EF Binary format.")
     binary32_file = st.file_uploader("Upload 32-bit Binary CSV", type=["csv"], key="binary_32")
 
     if binary32_file:
         df_32 = pd.read_csv(binary32_file, header=None)
-        
-        # Apply known ascending order headers manually:
-        df_32.columns = [str(h) for h in mutation_site_headers_actual_3614]
-        
-        # Reorder for 4402→3244 + 4882→4455 visualization:
-        df_32 = df_32[[str(h) for h in mutation_site_headers_3614]]
-        
-        if df_32.shape[1] != 32:
-            st.warning("⚠️ CSV must have exactly 32 columns.")
-        else:
-            # Reordered: 4402 → 3244, 4882 → 4455
-            df_32.columns = [str(h) for h in mutation_site_headers_3614]
-
-            # Show reordered binary
-            st.subheader("Binary Labels (Reordered 4402→3244, 4882→4455)")
-            reordered_df_32 = df_32[[str(h) for h in mutation_site_headers_3614 if str(h) in df_32.columns]]
-            st.dataframe(reordered_df_32.style.applymap(lambda v: "background-color: lightgreen" if v == 1 else "background-color: lightcoral"))
-            st.download_button("Download Reordered CSV", reordered_df_32.to_csv(index=False), "decoded_binary_32_reordered.csv", key="download_csv_tab5_32_reordered")
-
-            # Decode reordered binary to string
-            all_bits = reordered_df_32.values.flatten().astype(int).tolist()
-            decoded_reordered = binary_labels_to_string(all_bits)
-            st.subheader("Decoded String (Reordered 4402→3244, 4882→4455)")
-            st.write(decoded_reordered)
-
-            # Ascending: 3244 → 4882
-            df_ascending = df_32[[str(h) for h in mutation_site_headers_actual_3614 if str(h) in df_32.columns]]
-            st.subheader("Binary Labels (Ascending 3244→4882)")
-            st.dataframe(df_ascending.style.applymap(lambda v: "background-color: lightgreen" if v == 1 else "background-color: lightcoral"))
-            st.download_button("Download Ascending CSV", df_ascending.to_csv(index=False), "decoded_binary_32_ascending.csv", key="download_csv_tab5_32_ascend")
-
-            decoded_asc = binary_labels_to_string(df_ascending.values.flatten().astype(int).tolist())
-            st.subheader("Decoded String (Ascending 3244→4882)")
-            st.write(decoded_asc)
+        df_32.columns = [str(h) for h in mutation_site_headers_actual_3614]  # assume ascending
+
+        # Reordered: 4402 → 3244, 4882 → 4455
+        reordered_df_32 = df_32[[str(h) for h in mutation_site_headers_3614 if str(h) in df_32.columns]]
+        st.subheader("Binary Labels (Reordered 4402→3244, 4882→4455)")
+        st.dataframe(reordered_df_32.style.applymap(lambda v: "background-color: lightgreen" if v == 1 else "background-color: lightcoral"))
+        st.download_button("Download Reordered CSV", reordered_df_32.to_csv(index=False), "decoded_binary_32_reordered.csv", key="download_csv_tab5_32_reordered")
+
+        decoded_reordered = binary_labels_to_string(reordered_df_32.values.flatten().astype(int).tolist())
+        st.subheader("Decoded String (Reordered 4402→3244, 4882→4455)")
+        st.write(decoded_reordered)
+
+        # Ascending: 3244 → 4882
+        df_ascending = df_32[[str(h) for h in mutation_site_headers_actual_3614 if str(h) in df_32.columns]]
+        st.subheader("Binary Labels (Ascending 3244→4882)")
+        st.dataframe(df_ascending.style.applymap(lambda v: "background-color: lightgreen" if v == 1 else "background-color: lightcoral"))
+        st.download_button("Download Ascending CSV", df_ascending.to_csv(index=False), "decoded_binary_32_ascending.csv", key="download_csv_tab5_32_ascend")
+
+        decoded_asc = binary_labels_to_string(df_ascending.values.flatten().astype(int).tolist())
+        st.subheader("Decoded String (Ascending 3244→4882)")
+        st.write(decoded_asc)
 
     st.markdown("---")
 
+    # ========== 31-BIT DECODING ==========
     st.subheader("🔘 Option 2: 31-bit Binary Grouped per Row")
-    st.write("Upload CSV with 31 columns per row (each row = one 6-bit ASCII chunk group).")
-    binary31_file = st.file_uploader("Upload 31-bit Group CSV", type=["csv"], key="binary_31")
+    st.write("Upload CSV with 31 columns (no headers), each row = one 6-bit ASCII character group.")
 
+    binary31_file = st.file_uploader("Upload 31-bit Group CSV", type=["csv"], key="binary_31")
     if binary31_file:
         df_31 = pd.read_csv(binary31_file, header=None)
-        # Assume known ascending order and assign:
-        df_31.columns = [str(h) for h in mutation_site_headers_actual]
-        
-        # Reorder:
-        df_31 = df_31[[str(h) for h in mutation_site_headers]]
-        # Reordered: 4402 → 3244, 4882 → 4455
-        df_31.columns = [str(h) for h in mutation_site_headers]
+        df_31.columns = [str(h) for h in mutation_site_headers_actual]  # assume ascending
 
+        # Reordered: 4402 → 3244, 4882 → 4455
+        reordered_df_31 = df_31[[str(h) for h in mutation_site_headers if str(h) in df_31.columns]]
         st.subheader("Binary Labels (Reordered 4402→3244, 4882→4455)")
-        st.dataframe(df_31.style.applymap(lambda v: "background-color: lightgreen" if v == 1 else "background-color: lightcoral"))
-        st.download_button("Download Reordered CSV", df_31.to_csv(index=False), "decoded_binary_31_reordered.csv", key="download_csv_tab5_31_reordered")
-
-        decoded_rows = []
-        for _, row in df_31.iterrows():
-            bits = row.dropna().astype(int).tolist()[:31]
-            decoded_rows.append(binary_labels_to_string(bits))
+        st.dataframe(reordered_df_31.style.applymap(lambda v: "background-color: lightgreen" if v == 1 else "background-color: lightcoral"))
+        st.download_button("Download Reordered CSV", reordered_df_31.to_csv(index=False), "decoded_binary_31_reordered.csv", key="download_csv_tab5_31_reordered")
 
-        full_decoded = "".join(decoded_rows)
+        decoded_rows_reordered = [
+            binary_labels_to_string(row.dropna().astype(int).tolist()[:31])
+            for _, row in reordered_df_31.iterrows()
+        ]
+        full_decoded_reordered = "".join(decoded_rows_reordered)
         st.subheader("Decoded String from 31-bit Groups (Reordered 4402→3244, 4882→4455)")
-        st.write(full_decoded)
+        st.write(full_decoded_reordered)
+        st.download_button("Download Concatenated Output", full_decoded_reordered, "decoded_31bit_string_reordered.txt", key="download_csv_tab5_31")
 
-        st.download_button("Download Concatenated Output", full_decoded, "decoded_31bit_string.txt", key="download_csv_tab5_31")
+        # Ascending: 3244 → 4882
+        df_31_asc = df_31[[str(h) for h in mutation_site_headers_actual if str(h) in df_31.columns]]
+        st.subheader("Binary Labels (Ascending 3244→4882)")
+        st.dataframe(df_31_asc.style.applymap(lambda v: "background-color: lightgreen" if v == 1 else "background-color: lightcoral"))
+        st.download_button("Download Ascending CSV", df_31_asc.to_csv(index=False), "decoded_binary_31_ascending.csv", key="download_csv_tab5_31_ascend")
 
+        decoded_rows_asc = [
+            binary_labels_to_string(row.dropna().astype(int).tolist()[:31])
+            for _, row in df_31_asc.iterrows()
+        ]
+        full_decoded_asc = "".join(decoded_rows_asc)
+        st.subheader("Decoded String from 31-bit Groups (Ascending 3244→4882)")
+        st.write(full_decoded_asc)
+        st.download_button("Download Concatenated Output", full_decoded_asc, "decoded_31bit_string_ascending.txt", key="download_csv_tab5_31_asc")