Update app.py

app.py

@@ -9,6 +9,28 @@ from sklearn.ensemble import RandomForestRegressor  # Important pentru deseriali
 from pydantic import BaseModel, ValidationError, Field, field_validator, model_validator
 from typing import Any
 
+class RobustModelWrapper:
+    """Wrapper robust pentru model, compatibil cu FastAPI."""
+    def __init__(self, model, feature_names):
+        self.model = model
+        self.feature_names_in_ = np.array(feature_names)
+    
+    def predict(self, X):
+        """Realizează predicții asigurându-se că datele sunt în formatul corect."""
+        # Convertim la DataFrame dacă nu este deja
+        if not isinstance(X, pd.DataFrame):
+            X = pd.DataFrame(X, columns=self.feature_names_in_)
+        
+        # Asigură-te că DataFrame-ul are exact coloanele necesare în ordinea corectă
+        prediction_df = pd.DataFrame()
+        for feature in self.feature_names_in_:
+            if feature in X.columns:
+                prediction_df[feature] = X[feature]
+            else:
+                raise ValueError(f"Caracteristica '{feature}' lipsește din datele de intrare")
+        
+        # Acum realizăm predicția cu coloanele în ordinea corectă
+        return self.model.predict(prediction_df)
 
 app = FastAPI()