Update install_model.py

install_model.py

@@ -1,34 +1,37 @@
 import os
 import zipfile
 import argostranslate.package
+import shutil
 
 model_dir = "/app/models"
 install_dir = "/app/.local/share/argos-translate/packages"
 
+# Cleanup: Remove invalid model folder if it exists
+bad_path = os.path.join(install_dir, "model")
+if os.path.isdir(bad_path):
+    shutil.rmtree(bad_path)
+    print(f"Removed invalid leftover package: {bad_path}")
+
 def is_valid_argosmodel(path):
     try:
-        with zipfile.ZipFile(path, 'r') as z:
-            return 'metadata.json' in z.namelist()
+        with zipfile.ZipFile(path, 'r') as archive:
+            return 'metadata.json' in archive.namelist()
     except zipfile.BadZipFile:
         return False
 
-print(f"Looking for models in: {model_dir}")
-
+# Install all valid .argosmodel files
 for filename in os.listdir(model_dir):
     if filename.endswith(".argosmodel"):
-        model_path = os.path.join(model_dir, filename)
+        full_path = os.path.join(model_dir, filename)
 
-        if not is_valid_argosmodel(model_path):
-            print(f"Skipping invalid model: {filename}")
+        if not is_valid_argosmodel(full_path):
+            print(f"Skipping invalid model file: {filename}")
             continue
 
         try:
             print(f"Installing model: {filename}")
-            argostranslate.package.install_from_path(model_path)
+            argostranslate.package.install_from_path(full_path)
         except Exception as e:
             print(f"Error installing {filename}: {e}")
 
-print("\nInstallation complete. Installed model directories:")
-for root, dirs, files in os.walk(install_dir):
-    for d in dirs:
-        print(f"Folder {os.path.join(root, d)}")
+print("All valid models installed.")