build: unzip knapsack_cache.zip if it exists

app.py

@@ -10,6 +10,7 @@ import logging
 from pathlib import Path
 from omegaconf import OmegaConf, DictConfig
 from gradio_log import Log
+import zipfile
 
 # --- InstaNovo Imports ---
 try:
@@ -127,6 +128,18 @@ def load_models_and_knapsack():
     # --- Knapsack Handling ---
     # Only attempt knapsack loading/generation if the Transformer model is loaded
     if INSTANOVO is not None and RESIDUE_SET is not None and KNAPSACK is None:
+        # Check if knapsack_cache.zip exists and unzip its contents
+        knapsack_zip_path = Path("knapsack_cache.zip")
+        if knapsack_zip_path.exists():
+            logger.info(f"Found {knapsack_zip_path}. Extracting contents to {KNAPSACK_DIR}...")
+            try:
+                with zipfile.ZipFile(knapsack_zip_path, 'r') as zip_ref:
+                    zip_ref.extractall(KNAPSACK_DIR)
+                logger.info("Knapsack cache extracted successfully.")
+            except Exception as e:
+                logger.error(f"Failed to extract {knapsack_zip_path}: {e}")
+                raise gr.Error(f"Failed to extract knapsack cache: {e}")
+
         knapsack_exists = (
             (KNAPSACK_DIR / "parameters.pkl").exists()
             and (KNAPSACK_DIR / "masses.npy").exists()