Updated app.py

Tab5 now contains the full EpMotion script, with TS_10/TS_50, Source D wells, Name and barcode id columns

app.py

@@ -388,7 +388,6 @@ def get_well_position(sample_index):
     col_number = ((sample_index - 1) % 12) + 1
     return f"{row_letter}{col_number}"
 
-# Tab 5: Binary → String
 # Tab 5: Binary → String
 with tab5: 
     st.header("Decode Binary Labels to String")
@@ -415,7 +414,6 @@ with tab5:
                 new_src = f"{new_row_letter}{col_number}"
                 entry['Source'] = new_src
 
-                # Reset volume tracking for new source
                 if new_src not in source_volumes:
                     source_volumes[new_src] = 0
                 source_volumes[new_src] += vol
@@ -425,12 +423,52 @@ with tab5:
 
         return adjusted_sources, source_volumes
 
+    # Utility: Track D source volume separately for fixed-volume dispensing
+    def generate_fixed_d_source_instructions(df_robot, fixed_volume=16, volume_limit=170):
+        d_source_volumes = {}
+        d_source_script = []
+        current_d_index = 1
+
+        for row_idx, sample in df_robot.iterrows():
+            for col in df_robot.columns[1:]:
+                if int(sample[col]) == 1:
+                    # Check current D well volume
+                    current_d_well = f"D{current_d_index}"
+                    if current_d_well not in d_source_volumes:
+                        d_source_volumes[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
+
+                    dest = get_well_position(int(sample['Sample']))
+                    tool = 'TS_10' if fixed_volume < 10 else 'TS_50'
+
+                    d_source_script.append({
+                        'Source': current_d_well,
+                        'Destination': dest,
+                        'Volume': fixed_volume,
+                        'Tool': tool
+                    })
+
+        return d_source_script, d_source_volumes
+
     # ========== 32-BIT DECODING ==========
     st.subheader("32-bit Binary per Row")
     st.write("Upload CSV with 32 columns (0 or 1), no headers, from EF Binary format or enter manually below.")
 
     binary32_file = st.file_uploader("Upload 32-bit Binary CSV", type=["csv"], key="binary_32")
 
+    # === Optional fields ===
+    st.subheader("Optional Metadata (Optional)")
+    barcode_id_input = st.text_input("Barcode ID (applied to all rows, optional)", value="")
+    labware_source_input = st.text_input("Labware for Source (optional, default = 1)", value="1")
+    labware_dest_input = st.text_input("Labware for Destination (optional, default = 1)", value="1")
+    name_input = st.text_input("Name field (optional, default = blank)", value="")
+
     if binary32_file:
         df_32 = pd.read_csv(binary32_file, header=None)
         df_32.columns = [str(h) for h in mutation_site_headers_actual_3614]
@@ -484,18 +522,34 @@ with tab5:
                     source = source_wells_32[i]
                     dest = get_well_position(int(sample['Sample']))
                     vol = round(sample['volume donors (µl)'], 2)
-                    robot_script_32.append({'Source': source, 'Destination': dest, 'Volume': vol})
+                    tool = 'TS_10' if vol < 10 else 'TS_50'
+                    robot_script_32.append({
+                        'Source': source,
+                        'Destination': dest,
+                        'Volume': vol,
+                        'Tool': tool
+                    })
 
-        # Adjust for source well volume limit
         robot_script_32, source_volumes_32 = track_and_replace_source(source_wells_32, robot_script_32)
 
-        robot_script_32_df = pd.DataFrame(robot_script_32)
+        # Add fixed 16ul D sources
+        d_script, d_volumes = generate_fixed_d_source_instructions(df_32_robot)
+        full_robot_script = robot_script_32 + d_script
+
+        robot_script_32_df = pd.DataFrame(full_robot_script)
+        robot_script_32_df.insert(0, 'Barcode ID', barcode_id_input)
+        robot_script_32_df.insert(1, 'Labware_1', labware_source_input)
+        robot_script_32_df.insert(3, 'Labware_2', labware_dest_input)
+        robot_script_32_df['Name'] = name_input
+        robot_script_32_df = robot_script_32_df[['Barcode ID', 'Labware_1', 'Source', 'Labware_2', 'Destination', 'Volume', 'Tool', 'Name']]
+        robot_script_32_df.columns = ['Barcode ID', 'Labware', 'Source', 'Labware', 'Destination', 'Volume', 'Tool', 'Name']
+
         st.dataframe(robot_script_32_df)
         st.download_button("Download Robot Script (32-bit)", robot_script_32_df.to_csv(index=False), "robot_script_32bit.csv", key="download_robot_32")
 
-        # Show total volume per source well
         st.subheader("Total Volume Used Per Source")
-        source_volume_df = pd.DataFrame(list(source_volumes_32.items()), columns=['Source', 'Total Volume (µl)'])
+        combined_volumes = {**source_volumes_32, **d_volumes}
+        source_volume_df = pd.DataFrame(list(combined_volumes.items()), columns=['Source', 'Total Volume (µl)'])
         st.dataframe(source_volume_df)
         st.download_button("Download Source Volumes", source_volume_df.to_csv(index=False), "source_total_volumes.csv", key="download_volume_32")