Update app.py

app.py

@@ -1,6 +1,6 @@
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel, validator, model_validator
-from pydantic.errors import PydanticValueError
+# from pydantic.errors import PydanticValueError
 import pandas as pd
 import joblib
 import numpy as np
@@ -18,18 +18,18 @@ except Exception as e:
     model = None  # Setăm modelul ca None în caz de eroare
     FEATURE_ORDER = []  # Inițializăm o listă goală pentru a evita erorile ulterioare
 
-# Definim clase personalizate pentru erori
-class CementPercentError(PydanticValueError):
-    msg_template = "Cement percentage must be between 0% and 15%"
+# # Definim clase personalizate pentru erori
+# class CementPercentError(PydanticValueError):
+#     msg_template = "Cement percentage must be between 0% and 15%"
 
-class CuringPeriodError(PydanticValueError):
-    msg_template = "Curing period must be between 1 and 90 days"
+# class CuringPeriodError(PydanticValueError):
+#     msg_template = "Curing period must be between 1 and 90 days"
 
-class CompactionRateError(PydanticValueError):
-    msg_template = "Compaction velocity must be between 0.5 and 1.5 mm/min"
+# class CompactionRateError(PydanticValueError):
+#     msg_template = "Compaction velocity must be between 0.5 and 1.5 mm/min"
     
 
-class SoilInput(BaseModel):
+class SoilInput__(BaseModel):
     cement_perecent: float
     curing_period: float
     compaction_rate: float
@@ -41,24 +41,58 @@ class SoilInput(BaseModel):
             self.curing_period = 0
         else:
             if not (1 <= self.curing_period <= 90):
-                raise CuringPeriodError()
-                # raise ValueError("Perioada de maturare trebuie să fie între 1 și 90 zile dacă cement_perecent > 0")
+                # raise CuringPeriodError()
+                raise ValueError("Curing period must be between 1 and 90 days")
         return self
     
     @validator('cement_perecent')
     def validate_cement(cls, v):
         if not 0 <= v <= 15:
-            raise CementPercentError()
-            # raise ValueError("Procentul de ciment trebuie să fie între 0% și 15%")
+            # raise CementPercentError()
+            raise ValueError("Cement percentage must be between 0% and 15%")
         return v
 
     @validator('compaction_rate')
     def validate_compaction(cls, v):
         if not 0.5 <= v <= 1.5:
-            raise CompactionRateError()
-            # raise ValueError("Viteza de compactare trebuie să fie între 0.5 și 1.5 mm/min")
+            # raise CompactionRateError()
+            raise ValueError("Compaction velocity must be between 0.5 and 1.5 mm/min")
+        return v
+
+class SoilInput(BaseModel):
+    cement_perecent: float = Field(
+        ...,  # This means the field is required
+        description="Cement percentage in the mixture"
+    )
+    curing_period: float = Field(
+        ...,
+        description="Number of days for curing"
+    )
+    compaction_rate: float = Field(
+        ...,
+        description="Rate of compaction in mm/min"
+    )
+
+    @field_validator('cement_perecent')
+    @classmethod
+    def validate_cement(cls, v: float) -> float:
+        if not 0 <= v <= 15:
+            raise ValueError("Cement percentage must be between 0% and 15%")
+        return v
+
+    @field_validator('curing_period')
+    @classmethod
+    def validate_curing(cls, v: float) -> float:
+        if not 1 <= v <= 90:
+            raise ValueError("Curing period must be between 1 and 90 days")
         return v
 
+    @field_validator('compaction_rate')
+    @classmethod
+    def validate_compaction(cls, v: float) -> float:
+        if not 0.5 <= v <= 1.5:
+            raise ValueError("Compaction rate must be between 0.5 and 1.5 mm/min")
+        return v
         
 @app.post("/predict")
 async def predict(soil_data: SoilInput):
@@ -69,8 +103,23 @@ async def predict(soil_data: SoilInput):
         raise HTTPException(status_code=500, detail="Modelul nu a fost încărcat corect")
 
     try:
-        # Construim DataFrame-ul pentru predicție
         input_data = soil_data.dict()
+        
+        # Aplicăm regula fizică: dacă nu avem ciment, perioada de maturare este 0
+        if input_data['cement_perecent'] == 0:
+            # Păstrăm datele originale pentru răspuns
+            original_data = input_data.copy()
+            # Modificăm perioada de maturare pentru predicție
+            input_data['curing_period'] = 0
+            
+            # Adăugăm o notă explicativă în răspuns
+            explanation = "Pentru amestecuri fără ciment, perioada de maturare nu influențează rezistența."
+        else:
+            original_data = input_data
+            explanation = None
+        
+        # Construim DataFrame-ul pentru predicție
+        # input_data = soil_data.dict()
         input_df = pd.DataFrame([input_data])
 
         # Ne asigurăm că ordinea caracteristicilor este corectă