Update app.py

app.py

@@ -128,7 +128,11 @@ with tab1:
 
         # Step 3: Generate the robot script
         robot_script = []
-        source_wells = robot_template['Source'].unique().tolist()[:33]
+        source_wells = robot_template['Source'].unique().tolist()
+        if len(source_wells) < 32:
+            source_wells += [f"Fake{i}" for i in range(32 - len(source_wells))]
+        source_wells = source_wells[:32]
+
 
         st.write(f"Number of source wells: {len(source_wells)}")
         st.write(f"Number of binary columns: {len(df_sorted.columns[1:33])}")