Update app.py

app.py

@@ -475,5 +475,33 @@ with tab5:
         decoded_flat_asc = binary_labels_to_string(df_31_asc.values.flatten().astype(int).tolist())
         st.subheader("Decoded String (Flattened 31-bit Ascending)")
         st.write(decoded_flat_asc)
-        st.download_button("Download Concatenated Output", decoded_flat_asc, "decoded_31bit_string_ascending.txt", key="download_csv_tab5_31_asc")
+        st.download_button("Download Concatenated Output", decoded_flat_asc, "decoded_31bit_string_ascending.txt", key="download_csv_tab5_31_asc
+        
+        # === Robot Preparation Script from 31-bit Binary ===
+        st.subheader("Robot Preparation Script from 31-bit Binary")
+        robot_template_31 = pd.read_csv("/home/user/app/Robot.csv", skiprows=3)
+        robot_template_31.columns = ['Labware', 'Source', 'Labware_2', 'Destination', 'Volume', 'Tool', 'Name']
+
+        df_31_robot = df_31.copy()
+        df_31_robot.insert(0, 'Sample', range(1, len(df_31_robot)+1))
+        df_31_robot['# donors'] = df_31_robot.iloc[:, 1:].sum(axis=1)
+        df_31_robot['volume donors (µl)'] = 32 / df_31_robot['# donors']
+
+        robot_script_31 = []
+        source_wells_31 = robot_template_31['Source'].unique().tolist()
+        if len(source_wells_31) < df_31.shape[1]:
+            source_wells_31 += [f"Fake{i}" for i in range(df_31.shape[1] - len(source_wells_31))]
+        source_wells_31 = source_wells_31[:df_31.shape[1]]
+
+        for i, col in enumerate(df_31.columns):
+            for row_idx, sample in df_31_robot.iterrows():
+                if sample[col] == 1:
+                    source = source_wells_31[i]
+                    dest = f"A{sample['Sample']}"
+                    vol = round(sample['volume donors (µl)'], 2)
+                    robot_script_31.append({'Source': source, 'Destination': dest, 'Volume': vol})
+
+        robot_script_31_df = pd.DataFrame(robot_script_31)
+        st.dataframe(robot_script_31_df)
+        st.download_button("Download Robot Script (31-bit)", robot_script_31_df.to_csv(index=False), "robot_script_31bit.csv", key="download_robot_31")