Update app.py

app.py

@@ -391,6 +391,7 @@ 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")
@@ -406,20 +407,28 @@ with tab5:
         else:
             # Reordered: 4402 → 3244, 4882 → 4455
             df_32.columns = [str(h) for h in mutation_site_headers_3614]
-            all_bits = df_32.values.flatten().astype(int).tolist()
-            decoded_reordered = binary_labels_to_string(all_bits)
 
+            # 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)
-            st.download_button("Download Reordered CSV", df_32.to_csv(index=False), "decoded_binary_32_reordered.csv", key="download_csv_tab5_32_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]]
-            decoded_asc = binary_labels_to_string(df_ascending.values.flatten().astype(int).tolist())
+            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.download_button("Download Ascending CSV", df_ascending.to_csv(index=False), "decoded_binary_32_ascending.csv", key="download_csv_tab5_32_ascend")
 
     st.markdown("---")
 
@@ -430,18 +439,22 @@ with tab5:
     if binary31_file:
         df_31 = pd.read_csv(binary31_file, header=None)
 
-        def trim_row(row):
-            bits = row.dropna().astype(int).tolist()[:31]
-            return bits
+        # Reordered: 4402 → 3244, 4882 → 4455
+        df_31.columns = [str(h) for h in mutation_site_headers]
+
+        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 = trim_row(row)
+            bits = row.dropna().astype(int).tolist()[:31]
             decoded_rows.append(binary_labels_to_string(bits))
 
-        st.subheader("Decoded String from 31-bit Chunks")
         full_decoded = "".join(decoded_rows)
+        st.subheader("Decoded String from 31-bit Groups (Reordered 4402→3244, 4882→4455)")
         st.write(full_decoded)
 
         st.download_button("Download Concatenated Output", full_decoded, "decoded_31bit_string.txt", key="download_csv_tab5_31")
 
+