Update install_model.py

install_model.py

@@ -1,17 +1,34 @@
-import argostranslate.package
 import os
+import zipfile
+import argostranslate.package
 
 model_dir = "/app/models"
+install_dir = "/app/.local/share/argos-translate/packages"
+
+def is_valid_argosmodel(path):
+    try:
+        with zipfile.ZipFile(path, 'r') as z:
+            return 'metadata.json' in z.namelist()
+    except zipfile.BadZipFile:
+        return False
 
-if not os.path.isdir(model_dir):
-    print(f"Model directory {model_dir} not found.")
-    exit(1)
+print(f"Looking for models in: {model_dir}")
 
 for filename in os.listdir(model_dir):
     if filename.endswith(".argosmodel"):
         model_path = os.path.join(model_dir, filename)
+
+        if not is_valid_argosmodel(model_path):
+            print(f"Skipping invalid model: {filename}")
+            continue
+
         try:
-            print(f"Installing model: {model_path}")
+            print(f"Installing model: {filename}")
             argostranslate.package.install_from_path(model_path)
         except Exception as e:
-            print(f"Failed to install {model_path}: {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)}")