Update backend/app/services/cefr_predictor.py

backend/app/services/cefr_predictor.py

@@ -1,18 +1,27 @@
+import os
 from pathlib import Path
 import joblib
 
-# رجّع root ديال المشروع (learnlanguage)
-ROOT = Path(__file__).resolve().parents[3]  # .../learnlanguage
-MODEL_PATH = ROOT / "ml" / "models" / "cefr_model.pkl"
+MODEL_PATH = Path(os.getenv("CEFR_MODEL_PATH", "/app/ml/models/cefr_model.pkl"))
+
+HF_REPO_ID = os.getenv("CEFR_MODEL_REPO", "Zinebhm/cefr-model-learnlanguage")
+HF_FILENAME = os.getenv("CEFR_MODEL_FILENAME", "cefr_model.pkl")
 
 class CEFRPredictor:
     def __init__(self):
         if not MODEL_PATH.exists():
-            raise FileNotFoundError(f"Model not found: {MODEL_PATH}")
+            MODEL_PATH.parent.mkdir(parents=True, exist_ok=True)
+            # download from Hugging Face
+            from huggingface_hub import hf_hub_download
+            downloaded = hf_hub_download(
+                repo_id=HF_REPO_ID,
+                filename=HF_FILENAME,
+                repo_type="model",
+            )
+            # copy to expected path
+            os.replace(downloaded, MODEL_PATH)
+
         self.model = joblib.load(MODEL_PATH)
 
-    def predict(self, text: str) -> str:
-        text = (text or "").strip()
-        if not text:
-            return "A2"
+    def predict(self, text: str):
         return self.model.predict([text])[0]