Update website.py

website.py

@@ -19,47 +19,60 @@ def webai(user_input):
     user_input_clear = user_input
     input_data = [user_input_clear]
     input_data_scaled = (input_data - X_min) / (X_max - X_min)
-    print(input_data_scaled)
+    
     # Получаем предсказание от модели
     predicted_result_scaled = model.predict(input_data_scaled)
-    print(predicted_result_scaled[0][0] * 100)
-    # 35 0 190 75 120 80 1 1 0 0 1
-    # 35 0 170 90 130 90 1 1 0 0 0
-    # 39 0 156 45 110 80 2 1 0 0 0
-    # 47 1 168 87 120 80 2 1 1 1 1
-    # 37 0 185 75 120 80 2 1 1 1 0
-    return f"{round(predicted_result_scaled[0][0] * 100, 2)}%"
+    otv = round(predicted_result_scaled[0][0] * 100, 2)
+    if otv < 0:
+        otv = 0
+    elif otv > 100:
+        otv = 100
+    chans = ''
+    if otv >=0 and otv < 30:
+        chans = "Низкий"
+    elif otv >=30 and otv <50:
+        chans = "Средний"
+    elif otv >=50 and otv <70:
+        chans = "Высокий, обратитесь к кардиологу"
+    else:
+        chans = "Крайне высокий, обязательно обратитесь к кардиологу"
+    
+    return f'Вероятность заболевания: {otv}. Шанс вашего заболевания: ' + chans
 
 
 def pomoch(age, gender, height, weight, ap_hi, ap_lo, cholesterol, gluc, smoke, alco, active):
-    X = [int(age), gender, int(height), int(weight), int(ap_hi), int(ap_lo), float(cholesterol), float(gluc), smoke,
-         alco, active]
-    X[0] = today().year - X[0]
-    if X[1] == "Мужской":
-        X[1] = 0
-    else:
-        X[1] = 1
-    for i in range(8, 11):
-        if X[i] == "Да":
-            X[i] = 1
+    try:
+        X = [int(age), gender, int(height), int(weight), int(ap_hi), int(ap_lo), float(cholesterol), float(gluc), smoke,
+             alco, active]
+        X[0] = today().year - X[0]
+        if X[1] == "Мужской":
+            X[1] = 0
         else:
-            X[i] = 0
-    if X[6] <= 5:
-        X[6] = 1
-    else:
-        if X[6] >= 7.8:
-            X[6] = 3
+            X[1] = 1
+        for i in range(8, 11):
+            if X[i] == "Да":
+                X[i] = 1
+            else:
+                X[i] = 0
+        if X[6] <= 5:
+            X[6] = 1
         else:
-            X[6] = 2
-
-    if X[7] <= 5.5 and X[7] >= 3.3:
-        X[7] = 1
-    else:
-        if (X[7] > 5.5 and X[7] < 11) or X[7] < 3.3:
-            X[7] = 2
+            if X[6] >= 7.8:
+                X[6] = 3
+            else:
+                X[6] = 2
+    
+        if X[7] <= 5.5 and X[7] >= 3.3:
+            X[7] = 1
         else:
-            X[7] = 3
-    return webai(X)
+            if (X[7] > 5.5 and X[7] < 11) or X[7] < 3.3:
+                X[7] = 2
+            else:
+                X[7] = 3
+        return webai(X)
+    except ValueError:
+        return "Пожалуйста, убедитесь, что все значения числовые."
+    
 
 
 demo = gr.Interface(