Update app.py

app.py

@@ -1,5 +1,5 @@
 from fastapi import FastAPI, HTTPException
-from pydantic import BaseModel, validator
+from pydantic import BaseModel, validator, root_validator
 import pandas as pd
 import joblib
 import numpy as np
@@ -19,30 +19,60 @@ except Exception as e:
 
 
 # Definirea clasei pentru inputuri
+# class SoilInput(BaseModel):
+#     cement_perecent: float
+#     curing_period: float
+#     compaction_rate: float
+
+#     @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 SoilInput(BaseModel):
     cement_perecent: float
     curing_period: float
     compaction_rate: float
 
+    @root_validator
+    def check_cement_and_curing(cls, values):
+        cement = values.get('cement_perecent')
+        curing = values.get('curing_period')
+        # Dacă cement_perecent este 0, forțăm curing_period să fie 0,
+        # indiferent de valoarea transmisă.
+        if cement == 0:
+            values['curing_period'] = 0
+        else:
+            # În cazul în care cimentul nu e 0, ne asigurăm că curing_period este între 1 și 90.
+            if not (1 <= curing <= 90):
+                raise ValueError("Perioada de maturare trebuie să fie între 1 și 90 zile dacă cement_perecent > 0")
+        return values
+
     @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
-
-
+        
 @app.post("/predict")
 async def predict(soil_data: SoilInput):
     """