update

app.py

@@ -1,51 +1,54 @@
 import os
 import joblib
+import shutil
 from huggingface_hub import hf_hub_download
-
 import streamlit as st
 import pandas as pd
-import os
-import joblib
-from huggingface_hub import hf_hub_download
-
-# Hugging Face Model Repo
-MODEL_REPO = "chagu13/is_click_predictor"
-MODEL_DIR = "models"
-os.makedirs(MODEL_DIR, exist_ok=True)  # Ensure directory exists
 
 # Hugging Face Model Repo
 MODEL_REPO = "chagu13/is_click_predictor"
 MODEL_DIR = "models"
 os.makedirs(MODEL_DIR, exist_ok=True)  # Ensure directory exists
 
-# Model Filenames
+# Model Filenames (on Hugging Face)
 CATBOOST_MODEL_FILENAME = "models/catboost_model.pkl"
 XGB_MODEL_FILENAME = "models/xgb_model.pkl"
 RF_MODEL_FILENAME = "models/rf_model.pkl"
 
-# Full Local Paths
+# Expected Local Paths
 CATBOOST_MODEL_PATH = os.path.join(MODEL_DIR, "catboost_model.pkl")
 XGB_MODEL_PATH = os.path.join(MODEL_DIR, "xgb_model.pkl")
 RF_MODEL_PATH = os.path.join(MODEL_DIR, "rf_model.pkl")
 
 
+def download_model(filename, local_path):
+    """Download model from Hugging Face and move it to the correct location."""
+    temp_path = hf_hub_download(repo_id=MODEL_REPO, filename=filename, local_dir=MODEL_DIR)
+
+    # Ensure correct file placement
+    if temp_path != local_path:
+        shutil.move(temp_path, local_path)
+
+    return local_path
+
+
 def load_models():
-    """Download models from Hugging Face and load them."""
+    """Download and load models from Hugging Face."""
     try:
+        print("🔄 Checking and downloading models...")
+
+        # Ensure models are downloaded and placed correctly
         if not os.path.exists(CATBOOST_MODEL_PATH):
-            print("🔄 Downloading CatBoost model...")
-            hf_hub_download(repo_id=MODEL_REPO, filename=CATBOOST_MODEL_FILENAME, local_dir=MODEL_DIR)
+            print("🚀 Downloading CatBoost model...")
+            download_model(CATBOOST_MODEL_FILENAME, CATBOOST_MODEL_PATH)
+
         if not os.path.exists(XGB_MODEL_PATH):
-            print("🔄 Downloading XGBoost model...")
-            hf_hub_download(repo_id=MODEL_REPO, filename=XGB_MODEL_FILENAME, local_dir=MODEL_DIR)
-        if not os.path.exists(RF_MODEL_PATH):
-            print("🔄 Downloading RandomForest model...")
-            hf_hub_download(repo_id=MODEL_REPO, filename=RF_MODEL_FILENAME, local_dir=MODEL_DIR)
+            print("🚀 Downloading XGBoost model...")
+            download_model(XGB_MODEL_FILENAME, XGB_MODEL_PATH)
 
-        # ✅ Debug: Check if models exist
-        assert os.path.exists(CATBOOST_MODEL_PATH), f"❌ Missing file: {CATBOOST_MODEL_PATH}"
-        assert os.path.exists(XGB_MODEL_PATH), f"❌ Missing file: {XGB_MODEL_PATH}"
-        assert os.path.exists(RF_MODEL_PATH), f"❌ Missing file: {RF_MODEL_PATH}"
+        if not os.path.exists(RF_MODEL_PATH):
+            print("🚀 Downloading RandomForest model...")
+            download_model(RF_MODEL_FILENAME, RF_MODEL_PATH)
 
         # ✅ Load models
         print("📦 Loading models...")