Update app.py (#2)

- Update app.py (b066f66edf36de43aaec7ed09190bdf6889e4cea)


Co-authored-by: JESUS SANCHEZ <jsr90@users.noreply.huggingface.co>

app.py

@@ -2,24 +2,22 @@ import gradio as gr
 import pandas as pd
 from joblib import load
 
-
-
-def cardio(age,is_male,ap_hi,ap_lo,cholesterol,gluc,smoke,alco,active,height,weight,BMI):
+def cardio(age,gender,ap_hi,ap_lo,cholesterol,gluc,smoke,alco,active,height,weight):
     model = load('cardiosight.joblib')
     df = pd.DataFrame.from_dict(
         {
             "age": [age*365],
-            "gender":[0 if is_male else 1],
+            "gender":[0 if gender=='Male' else 1],
             "ap_hi": [ap_hi],
             "ap_lo": [ap_lo],
             "cholesterol": [cholesterol + 1],
             "gluc": [gluc + 1],
-            "smoke":[1 if smoke else 0],
-            "alco": [1 if alco else 0],
-            "active": [1 if active else 0],
+            "smoke":[1 if smoke=='Yes' else 0],
+            "alco": [1 if alco=='Yes' else 0],
+            "active": [1 if active=='Yes' else 0],
             "newvalues_height": [height],
             "newvalues_weight": [weight],
-            "New_values_BMI": [BMI],
+            "New_values_BMI": weight/((height/100)**2),
             
         }
     )
@@ -29,37 +27,35 @@ def cardio(age,is_male,ap_hi,ap_lo,cholesterol,gluc,smoke,alco,active,height,wei
       predicted="Tiene un riesgo alto de sufrir problemas cardiovasculares"
     else:
       predicted="Su riesgo de sufrir problemas cardiovasculares es muy bajo. Siga así."
-    return predicted
+    return "Su IMC es de "+str(round(df['New_values_BMI'][0], 2))+'. '+predicted
     
 iface = gr.Interface(
     cardio,
     [
-        gr.inputs.Slider(1,99,label="Age"),
-        "checkbox",
-        gr.inputs.Slider(10,250,label="Diastolic Preassure"),
-        gr.inputs.Slider(10,250,label="Sistolic Preassure"),
-        gr.inputs.Radio(["Normal","High","Very High"],type="index",label="Cholesterol"),
-        gr.inputs.Radio(["Normal","High","Very High"],type="index",label="Glucosa Level"),
-        "checkbox",
-        "checkbox",
-        "checkbox",
-        gr.inputs.Slider(30,220,label="Height in cm"),
-        gr.inputs.Slider(10,300,label="Weight in Kg"),
-        gr.inputs.Slider(1,50,label="BMI"),
+        gr.Slider(1,99,label="Age"),
+        gr.Dropdown(choices=['Male', 'Female'], label='Gender', value='Female'),
+        gr.Slider(10,250,label="Diastolic Preassure"),
+        gr.Slider(10,250,label="Sistolic Preassure"),
+        gr.Radio(["Normal","High","Very High"],type="index",label="Cholesterol"),
+        gr.Radio(["Normal","High","Very High"],type="index",label="Glucosa Level"),
+        gr.Dropdown(choices=['Yes', 'No'], label='Smoke', value='No'),
+        gr.Dropdown(choices=['Yes', 'No'], label='Alcohol', value='No'),
+        gr.Dropdown(choices=['Yes', 'No'], label='Active', value='Yes'),
+        gr.Slider(30,220,label="Height in cm"),
+        gr.Slider(10,300,label="Weight in Kg"),
     ],
 
     "text",
     examples=[
-        [40,True,120,80,"High","Normal",0,0,1,168,62,21],
-        [35,False,150,60,"Very High","Normal",0,0,1,143,52,31],
-        [60,True,160,70,"High","High",1,1,0,185,90,23],
+        [20,'Male',110,60,"Normal","Normal",'No','No','Yes',168,60],
+        [30,'Female',120,70,"High","High",'No','Yes','Yes',143,70],
+        [40,'Male',130,80,"Very High","Very High",'Yes','Yes','No',185,80],
+        [50,'Female',140,90,"Normal","High",'Yes','No','No',165,90],
+        [60,'Male',150,100,"High","Very High",'No','No','Yes',175,100],
+        [70,'Female',160,90,"Very High","Normal",'Yes','Yes','No',185,110],
     ],
-    interpretation="default",
     title = 'Calculadora de Riesgo Cardiovascular mediante Inteligencia Artificial',
-    description = 'El proyecto de CARDIOSIGHT nace debido a la presente necesidad en nuestro país de crear métodos y herramientas de identificación temprana para los individuos con alto riesgo de sufrir enfermedades cardiovasculares. Con el fin de prevenir eventos cardíacos primarios y ayudar a disminuir la incidencia de nuevos casos, por medio de hábitos de prevención. Mas información: https://saturdays.ai/2022/03/16/cardiosight-machine-learning-para-calcular-riesgo-cardiovascular/',
-    theme = 'grass'
+    description = 'Duplicación del proyecto de CARDIOSIGHT. He cambiado los botones tipo check por dropdown y calculado el IMC a partir de la altura y el peso. Más información: https://saturdays.ai/2022/03/16/cardiosight-machine-learning-para-calcular-riesgo-cardiovascular/'
 )
 
-
-   
 iface.launch()