Update risk_model.py

risk_model.py

@@ -29,4 +29,30 @@ def load_model():
 
 model = load_model()
 
-def predict_risk_
+def predict_risk(temp, duration):
+    global model
+    pred = model.predict([[temp, duration]])[0]
+    timestamp = get_ist_time()
+    HISTORY.append({
+        "Temperature": temp,
+        "Duration": duration,
+        "Risk": pred,
+        "Timestamp": timestamp
+    })
+    return pred, timestamp
+
+def retrain_model():
+    try:
+        data = pd.read_csv(DATA_PATH)
+        X = data[["temperature", "duration"]]
+        y = data["risk_level"]
+        clf = RandomForestClassifier().fit(X, y)
+        dump(clf, MODEL_PATH)
+        global model
+        model = clf
+        return "✅ Model retrained successfully"
+    except Exception as e:
+        return f"❌ Error: {str(e)}"
+
+def get_history_df():
+    return pd.DataFrame(HISTORY)