Update app.py

app.py

@@ -58,33 +58,57 @@ UI_TEXT = {
 }
 
 # --- 2. Model Loading (EXPANDED TO 7 MODELS) ---
-MODEL_NAMES = {
-    "de": "de_core_news_md",
-    "en": "en_core_web_md",
-    "es": "es_core_news_md",
-    "la": "la_core_web_sm",  # Medium model not available for core Latin
-    "grc": "grc_core_news_md", # Ancient Greek
-    "he": "he_core_news_md", # Hebrew
-    "ar": "ar_core_news_md"  # Arabic
+
+# We now store a tuple: (model_name_to_load, install_package_name_or_url)
+# For 'la', we use the direct wheel URL as the package.
+# This URL is for spaCy v3.8.x, which matches the HF Space log
+LATIN_URL = "https://github.com/explosion/spacy-models/releases/download/la_core_web_sm-3.8.0/la_core_web_sm-3.8.0-py3-none-any.whl"
+
+MODEL_INFO = {
+    # lang_code: (model_name, install_package)
+    "de": ("de_core_news_md", "de_core_news_md"),
+    "en": ("en_core_web_md", "en_core_web_md"),
+    "es": ("es_core_news_md", "es_core_news_md"),
+    "la": ("la_core_web_sm", f"pip install {LATIN_URL}"), # <-- The Fix from previous step
+    "grc": ("grc_core_news_md", "grc_core_news_md"),
+    "he": ("he_core_news_md", "he_core_news_md"),
+    "ar": ("ar_core_news_md", "ar_core_news_md")
 }
 
-def load_model(model_name):
-    """Checks if model is installed and downloads it if not."""
+
+def load_model(model_info):
+    """
+    Checks if model is installed. If not, attempts to download it
+    using the specific install command provided in MODEL_INFO.
+    """
+    model_name, install_command = model_info
     try:
         nlp = spacy.load(model_name)
-        print(f"{model_name} loaded successfully.")
+        print(f"Model {model_name} loaded successfully.")
     except OSError:
-        print(f"{model_name} not found. Downloading...")
-        # This command runs in the Hugging Face environment
-        os.system(f"python -m spacy download {model_name}")
+        print(f"Model {model_name} not found. Attempting download...")
+        
+        # Use pip install for the Latin URL, spacy download for others
+        if not install_command.startswith("pip"):
+            install_command = f"python -m spacy download {install_command}"
+            
+        print(f"Running: {install_command}")
+        exit_code = os.system(install_command) # Run the correct command
+        
+        if exit_code != 0:
+            print(f"--- ERROR: Failed to download {model_name}. Exit code: {exit_code} ---")
+            # This will raise a final error, as the load will fail again
+        
+        # Try loading again after install attempt
         nlp = spacy.load(model_name)
-        print(f"{model_name} downloaded and loaded.")
+        print(f"Model {model_name} downloaded and loaded.")
     return nlp
 
+
 # Load all models at startup and store them in a dictionary
 print("Loading all models...")
 MODELS = {
-    lang_code: load_model(model_name) for lang_code, model_name in MODEL_NAMES.items()
+    lang_code: load_model(model_info) for lang_code, model_info in MODEL_INFO.items()
 }
 print("All models loaded successfully.")