Update app.py

app.py

@@ -38,6 +38,7 @@ except Exception as e:
     print(f"Eroare la încărcarea modelului: {str(e)}")
     raise
 
+FEATURE_ORDER = ['cement_perecent', 'curing_period', 'compaction_rate']
 @app.post("/predict")
 async def predict(soil_data: SoilInput):
     """
@@ -49,10 +50,11 @@ async def predict(soil_data: SoilInput):
         input_df = pd.DataFrame([input_data])
         
         # Ne asigurăm că ordinea coloanelor este corectă
-        feature_order = ['cement_perecent', 'curing_period', 'compaction_rate']
+        feature_order = FEATURE_ORDER
         input_df = input_df[feature_order]
+        # input_df = pd.DataFrame([soil_data.dict()])[FEATURE_ORDER]
 
-        expected_features = model.feature_names_in_  # Extrage ordinea corectă din model
+        # expected_features = model.feature_names_in_  # Extrage ordinea corectă din model
        # input_df = input_df[expected_features]  # Reordonează caracteristicile
         
         # Facem predicția