Update risk_model.py

risk_model.py

@@ -14,25 +14,20 @@ def predict_risk(temp, duration):
     """
     
     # Define risk level, score, and alert based on temperature and duration conditions
-    if temp <= 160 and duration <= 45:
-        # Low Risk: temperature <= 160°C and duration <= 45 minutes
+    if temp <= 100 and duration <= 45:
+        # Low Risk: temperature between 0°C and 100°C, and duration <= 45 minutes
         risk_level = "Low"
-        risk_score = round(np.random.uniform(0, 40), 2)  # Low risk score (0 to 40%)
+        risk_score = np.random.uniform(0, 33)  # Low risk score (0 to 33%)
         alert = "Safe"
-    elif 161 <= temp <= 190 and 46 <= duration <= 90:
-        # Moderate Risk: temperature between 161°C and 190°C and duration between 46 and 90 minutes
+    elif 101 <= temp <= 150 and 46 <= duration <= 70:
+        # Moderate Risk: temperature between 101°C and 150°C, and duration between 46 and 70 minutes
         risk_level = "Moderate"
-        risk_score = round(np.random.uniform(40, 70), 2)  # Moderate risk score (40 to 70%)
+        risk_score = np.random.uniform(34, 66)  # Moderate risk score (34 to 66%)
         alert = "Risk"
-    elif temp > 190 and duration > 90:
-        # High Risk: temperature > 190°C and duration > 90 minutes
+    else:
+        # High Risk: temperature between 151°C and 200°C, and duration > 70 minutes
         risk_level = "High"
-        risk_score = round(np.random.uniform(70, 100), 2)  # High risk score (70 to 100%)
+        risk_score = np.random.uniform(67, 100)  # High risk score (67 to 100%)
         alert = "High Risk"
-    else:
-        # Conditions between Moderate and High Risk
-        risk_level = "Moderate"
-        risk_score = round(np.random.uniform(40, 70), 2)  # Moderate risk score (40 to 70%)
-        alert = "Risk"
 
-    return risk_level, risk_score, alert
+    return risk_level, round(risk_score, 2), alert