Update app.py

app.py

@@ -1,5 +1,6 @@
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel, validator, model_validator
+from pydantic.errors import PydanticValueError
 import pandas as pd
 import joblib
 import numpy as np
@@ -17,75 +18,45 @@ 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%"
 
-# Definirea clasei pentru inputuri
-# class SoilInput(BaseModel):
-#     cement_perecent: float
-#     curing_period: float
-#     compaction_rate: float
+class CuringPeriodError(PydanticValueError):
+    msg_template = "Curing period must be between 1 and 90 days"
 
-#     @validator('cement_perecent')
-#     def validate_cement(cls, v):
-#         if not 0 <= v <= 15:
-#             raise ValueError("Procentul de ciment trebuie să fie între 0% și 15%")
-#         return v
-
-#     @validator('curing_period')
-#     def validate_curing(cls, v):
-#         if not 1 <= v <= 90:
-#             raise ValueError("Perioada de maturare trebuie să fie între 1 și 90 zile")
-#         return v
-
-#     @validator('compaction_rate')
-#     def validate_compaction(cls, v):
-#         if not 0.5 <= v <= 1.5:
-#             raise ValueError("Viteza de compactare trebuie să fie între 0.5 și 1.5 mm/min")
-#         return v
+class CompactionRateError(PydanticValueError):
+    msg_template = "Compaction velocity must be between 0.5 and 1.5 mm/min"
+    
 
 class SoilInput(BaseModel):
     cement_perecent: float
     curing_period: float
     compaction_rate: float
 
-    # @model_validator(mode='after')
-    # def check_cement_and_curing(self):
-    #     cement = values.get('cement_perecent')
-    #     curing = values.get('curing_period')
-    #     if cement == 0:
-    #         values['curing_period'] = 0
-    #     else:
-    #         if not (1 <= curing <= 90):
-    #             raise ValueError("Perioada de maturare trebuie să fie între 1 și 90 zile dacă cement_perecent > 0")
-    #     return values
-
+    
     @model_validator(mode="after")
     def check_cement_and_curing(self):
         if self.cement_perecent == 0:
             self.curing_period = 0
         else:
             if not (1 <= self.curing_period <= 90):
-                raise ValueError("Perioada de maturare trebuie să fie între 1 și 90 zile dacă cement_perecent > 0")
+                raise CuringPeriodError()
+                # raise ValueError("Perioada de maturare trebuie să fie între 1 și 90 zile dacă cement_perecent > 0")
         return self
-        
-# @model_validator(mode="after")
-# def check_cement_and_curing(self):
-#     if self.cement_percent == 0:
-#         self.curing_period = 0
-#     else:
-#         if not (1 <= self.curing_period <= 90):
-#             raise ValueError("Invalid curing period")
-#     return self
     
     @validator('cement_perecent')
     def validate_cement(cls, v):
         if not 0 <= v <= 15:
-            raise ValueError("Procentul de ciment trebuie să fie între 0% și 15%")
+            raise CementPercentError()
+            # raise ValueError("Procentul de ciment trebuie să fie între 0% și 15%")
         return v
 
     @validator('compaction_rate')
     def validate_compaction(cls, v):
         if not 0.5 <= v <= 1.5:
-            raise ValueError("Viteza de compactare trebuie să fie între 0.5 și 1.5 mm/min")
+            raise CompactionRateError()
+            # raise ValueError("Viteza de compactare trebuie să fie între 0.5 și 1.5 mm/min")
         return v