Send cryolo model when flushing spa

Author: stephen-riggs

src/murfey/workflows/spa/flush_spa_preprocess.py

@@ -8,6 +8,7 @@
 
 from murfey.server import _transport_object
 from murfey.server.api.auth import MurfeySessionID
+from murfey.server.api.spa import _cryolo_model_path
 from murfey.server.feedback import _murfey_id
 from murfey.util import sanitise, secure_path
 from murfey.util.config import get_machine_config, get_microscope
@@ -314,15 +315,12 @@ def flush_spa_preprocess(message: dict, murfey_db: Session, demo: bool = False)
     ).all()
     if not stashed_files:
         return True
-    instrument_name = (
-        murfey_db.exec(
-            select(MurfeySession).where(MurfeySession.id == message["session_id"])
-        )
-        .one()
-        .instrument_name
-    )
-    machine_config = get_machine_config(instrument_name=instrument_name)[
-        instrument_name
+
+    murfey_session = murfey_db.exec(
+        select(MurfeySession).where(MurfeySession.id == message["session_id"])
+    ).one()
+    machine_config = get_machine_config(instrument_name=murfey_session.instrument_name)[
+        murfey_session.instrument_name
     ]
     recipe_name = machine_config.recipes.get("em-spa-preprocess", "em-spa-preprocess")
     collected_ids = murfey_db.exec(
@@ -424,6 +422,11 @@ def flush_spa_preprocess(message: dict, murfey_db: Session, demo: bool = False)
                     else f.eer_fractionation_file
                 ),
                 "do_icebreaker_jobs": default_spa_parameters.do_icebreaker_jobs,
+                "cryolo_model_weights": str(
+                    _cryolo_model_path(
+                        murfey_session.visit, murfey_session.instrument_name
+                    )
+                ),
                 "foil_hole_id": foil_hole_id,
             },
         }