Update app.py

app.py

@@ -29,6 +29,9 @@ kmeans_artifact = joblib.load("kmeans_model.joblib")
 kmeans_model = kmeans_artifact["model"]
 cluster_scaler = kmeans_artifact["scaler"]
 cluster_features_names = kmeans_artifact["feature_names"]
+if "cluster" in cluster_features_names:
+    raise ValueError("cluster_features_names must NOT include 'cluster'")
+
 
 
 ################### FEATURE GATHERING
@@ -129,11 +132,14 @@ def predict_volatility(date):
 
     # Feature engineering
     df = engineer_features(df)
-    df = df.dropna()
 
+    # Clustering
     # Clustering
     df = assign_cluster(df)
 
+    assert "cluster" in df.columns, "Cluster feature missing after assignment"
+
+
     # ---------------- NOWCASTING ----------------
     X_xgb = df.iloc[-1:][xgb_features]
     X_ngb = df.iloc[-1:][ngb_features]