refactor: replace load_model with get_model for improved model loading

app.py

@@ -1,9 +1,9 @@
 from interface import create_interface
-from model_utils import load_model
+from model_utils import get_model
 
 def main():
     print("Loading model...")
-    model = load_model()
+    model = get_model()
     
     if model is None:
         print("Warning: Model could not be loaded. Please ensure 'best_model.pkl' exists.")

model_utils.py

@@ -5,17 +5,6 @@ from config import MODEL_PATH
 
 warnings.filterwarnings('ignore')
 
-model = None
-
-def load_model():
-    global model
-    try:
-        model = joblib.load(MODEL_PATH)
-        return model
-    except Exception as e:
-        print(f"Error loading model: {e}")
-        return None
-
 def calculate_derived_features(age, weight, height, neutrophil, lymphocyte, platelet):
     height_m = height / 100
     bmi = weight / (height_m ** 2)
@@ -26,12 +15,7 @@ def calculate_derived_features(age, weight, height, neutrophil, lymphocyte, plat
 def predict_outcome(age, weight, height, gravidity, parity, h_abortion, 
                    living_child, gestational_age, hemoglobin, hematocrit, 
                    platelet, mpv, pdw, neutrophil, lymphocyte):
-    global model
-    
-    if model is None:
-        model = load_model()
-        if model is None:
-            return "خطا: مدل بارگذاری نشد", ""
+    model = get_model()
     
     try:
         bmi, nlr, plr = calculate_derived_features(age, weight, height, neutrophil, lymphocyte, platelet)
@@ -72,8 +56,16 @@ def predict_outcome(age, weight, height, gravidity, parity, h_abortion,
     except Exception as e:
         return f"خطا در پردازش: {str(e)}", ""
 
+
+model = None
+
 def get_model():
     global model
     if model is None:
-        model = load_model()
+        try:
+            model = joblib.load(MODEL_PATH)
+            return model
+        except Exception as e:
+            print(f"Error loading model: {e}")
+            return None
     return model