Update app.py

app.py

@@ -30,6 +30,23 @@ def image_to_binary_labels(img: Image.Image, max_pixels: int = 256) -> list[int]
     binarized = (flat > 128).astype(int)  # Threshold at 128 to 0/1
     return binarized.tolist()
 
+def binary_labels_to_image(binary_labels: list[int], width: int = None, height: int = None) -> Image.Image:
+    """
+    Convert binary labels (0/1) back into an image.
+    If width and height are not provided, assumes a square image.
+    """
+    total_pixels = len(binary_labels)
+    if width is None or height is None:
+        side = int(np.ceil(np.sqrt(total_pixels)))
+        width = height = side
+    needed_pixels = width * height
+    if total_pixels < needed_pixels:
+        binary_labels += [0] * (needed_pixels - total_pixels)
+    array = np.array(binary_labels, dtype=np.uint8) * 255
+    image_array = array.reshape((height, width))
+    img = Image.fromarray(image_array, mode='L')
+    return img
+
 # Predefined headers for the 32 mutation sites
 mutation_site_headers = [
     3244, 3297, 3350, 3399, 3455, 3509, 3562, 3614,
@@ -48,21 +65,17 @@ with tab1:
     user_input = st.text_input("Text Input", value="DNA")
 
     if user_input:
-        # Compute ASCII codes
         ascii_codes = [ord(c) for c in user_input]
-        # Compute binary labels
         binary_labels = string_to_binary_labels(user_input)
 
         st.subheader("ASCII Codes")
         st.write(ascii_codes)
 
         st.subheader("Binary Labels per Character")
-        # Display bits grouped per character for readability
         grouped_chars = [binary_labels[i:i+8] for i in range(0, len(binary_labels), 8)]
         for idx, bits in enumerate(grouped_chars):
             st.write(f"'{user_input[idx]}' → {bits}")
 
-        # New section: 32-bit group display in table format
         st.subheader("Binary Labels (32-bit groups)")
         num_groups = (len(binary_labels) + 31) // 32
         table_data = []
@@ -121,4 +134,9 @@ with tab2:
             mime="text/csv"
         )
 
+        st.subheader("Reconstruct Image from Binary Labels")
+        if st.button("Reconstruct Image"):
+            reconstructed_img = binary_labels_to_image(binary_labels)
+            st.image(reconstructed_img, caption="Reconstructed Image", use_column_width=True)
+
 # Future: integrate DNA editor mapping for each mutation site here