Update app.py

app.py

@@ -739,7 +739,7 @@ voyager_table = {
 }
 reverse_voyager_table = {v: k for k, v in voyager_table.items()}
 
-# === Binary ↔ String conversion ===
+# === Binary → String conversion ===
 def binary_labels_to_string(bits: list[int]) -> str:
     chars = []
     for i in range(0, len(bits), 6):
@@ -752,10 +752,16 @@ def binary_labels_to_string(bits: list[int]) -> str:
 
 # === Well mapping ===
 def get_well_position(sample_index):
-    row_letter = chr(65 + (sample_index - 1) // 12)
-    col_number = ((sample_index - 1) % 12) + 1
+    """Convert sample index (1-based) into A1–H12 pattern within its plate"""
+    row_letter = chr(65 + ((sample_index - 1) % 96) // 12)  # 8 rows (A–H)
+    col_number = ((sample_index - 1) % 12) + 1              # 12 columns
     return f"{row_letter}{col_number}"
 
+def get_plate_id(sample_index):
+    """Return Plate number based on 96 samples per plate"""
+    plate_number = math.ceil(sample_index / 96)
+    return f"Plate {plate_number}"
+
 # === Track and replace source if volume exceeded ===
 def track_and_replace_source(source_list, robot_script, volume_limit=150):
     source_volumes = {}
@@ -782,28 +788,24 @@ def generate_fixed_d_source_instructions_to_all_samples(n_samples, fixed_volume=
     current_d_index = 1
     for i in range(n_samples):
         dest = get_well_position(i + 1)
+        plate = get_plate_id(i + 1)
         current_d_well = f"D{current_d_index}"
         d_source_volumes.setdefault(current_d_well, 0)
-
         if d_source_volumes[current_d_well] + fixed_volume > volume_limit:
             current_d_index += 1
             current_d_well = f"D{current_d_index}"
             d_source_volumes[current_d_well] = 0
-
         d_source_volumes[current_d_well] += fixed_volume
-
-        # ✅ Updated: use TS_50 if volume >10 µL, else TS_10
         tool = 'TS_50' if fixed_volume > 10 else 'TS_10'
         d_source_script.append({
+            'Plate': plate,
             'Source': current_d_well,
             'Destination': dest,
             'Volume': fixed_volume,
             'Tool': tool
         })
-
     return d_source_script, d_source_volumes
 
-# === Source well generation ===
 def generate_source_wells(n):
     wells, rows = [], 'ABCDEFGHIJKLMNOPQRSTUVWXYZ'
     for i in range(n):
@@ -813,7 +815,6 @@ def generate_source_wells(n):
 
 # === Main UI ===
 st.header("Upload Binary Data (0/1)")
-
 binary_file = st.file_uploader("Upload Binary CSV", type=["csv"])
 st.divider()
 
@@ -831,7 +832,7 @@ if binary_file:
 else:
     st.info("No file uploaded — manually enter binary data below.")
     df_binary = st.data_editor(
-        pd.DataFrame([str(i) for i in range(1, 20)]),
+        pd.DataFrame(columns=[str(i) for i in range(1, 33)]),
         num_rows="dynamic", key="manual_input"
     )
 
@@ -861,19 +862,24 @@ if not df_binary.empty:
     for i, col in enumerate(df_robot.columns[1:]):
         for _, sample in df_robot.iterrows():
             if int(sample[col]) == 1:
+                sample_id = int(sample['Sample'])
+                plate = get_plate_id(sample_id)
                 source = source_wells[i]
-                dest = get_well_position(int(sample['Sample']))
+                dest = get_well_position(sample_id)
                 vol = round(sample['volume donors (µL)'], 2)
-
-                # ✅ Updated: use TS_50 for volumes >10 µL, TS_10 otherwise
                 tool = 'TS_50' if vol > 10 else 'TS_10'
-                robot_script.append({'Source': source, 'Destination': dest, 'Volume': vol, 'Tool': tool})
+                robot_script.append({
+                    'Plate': plate,  # ✅ New Column
+                    'Source': source,
+                    'Destination': dest,
+                    'Volume': vol,
+                    'Tool': tool
+                })
 
     robot_script, source_volumes = track_and_replace_source(source_wells, robot_script, volume_limit=volume_limit_input)
     d_script, d_volumes = generate_fixed_d_source_instructions_to_all_samples(
         len(df_robot), fixed_volume=16, volume_limit=volume_limit_input
     )
-
     full_script = robot_script + d_script
 
     robot_script_df = pd.DataFrame(full_script)
@@ -881,8 +887,9 @@ if not df_binary.empty:
     robot_script_df.insert(1, 'Labware_Source', labware_source_input)
     robot_script_df.insert(3, 'Labware_Destination', labware_dest_input)
     robot_script_df['Name'] = name_input
-    robot_script_df = robot_script_df[['Barcode ID', 'Labware_Source', 'Source',
-                                       'Labware_Destination', 'Destination', 'Volume', 'Tool', 'Name']]
+    robot_script_df = robot_script_df[['Barcode ID', 'Labware_Source', 'Plate',
+                                       'Source', 'Labware_Destination', 'Destination',
+                                       'Volume', 'Tool', 'Name']]
 
     st.dataframe(robot_script_df)
     st.download_button("⬇️ Download Robot Script", robot_script_df.to_csv(index=False), "robot_script.csv")