Update app.py

app.py

@@ -25,7 +25,7 @@ def main_func(age, sex, cp, trtbps, chol, fbs, restecg, thalachh,exng,oldpeak,sl
     shap_values = explainer(new_row)
     # plot = shap.force_plot(shap_values[0], matplotlib=True, figsize=(30,30), show=False)
     # plot = shap.plots.waterfall(shap_values[0], max_display=6, show=False)
-    plot = shap.plots.bar(shap_values[0], max_display=6, order=shap.Explanation.abs, show_data='auto', show=False)
+    plot = shap.plots.bar(shap_values[0], max_display=8, order=shap.Explanation.abs, show_data='auto', show=False)
 
     plt.tight_layout()
     local_plot = plt.gcf()
@@ -35,49 +35,96 @@ def main_func(age, sex, cp, trtbps, chol, fbs, restecg, thalachh,exng,oldpeak,sl
 
 # Create the UI
 title = "**Heart Attack Predictor & Interpreter** 🪐"
-description1 = """This app takes info from subjects and predicts their heart attack likelihood. Do not use for medical diagnosis."""
+description1 = """This app takes info from subjects and predicts their heart attack likelihood. Do not use these results for an actual medical diagnosis."""
 
 description2 = """
-To use the app, click on one of the examples, or adjust the values of the factors, and click on Analyze. 🤞
+To use the app, simply adjust the inputs and click the "Analyze" button. You can also click one of the examples below to see how it's done!
 """ 
 
 with gr.Blocks(title=title) as demo:
-    gr.Markdown(f"## {title}")
-    gr.Markdown(description1)
-    gr.Markdown("""---""")
-    gr.Markdown(description2)
-    gr.Markdown("""---""")
 
-    age = gr.Number(label="age Score", value=80)
-    sex = gr.Radio(["Male", "Female"], label="Gender", type = "index", info = "What is your gender?")
-    cp = gr.Radio([0,1,2,3], label="Chest Pain", info="Rate the severity of your chest pain [0: none, 5: ER RIGHT NOW]:")
-    trtbps = gr.Number(label="Resting Blood Pressure", info = "What is your resting blood pressure?", minimum=1, maximum=200, value=4) 
-    chol = gr.Slider(label="Cholesterol", info="What is your Cholesterol level?", minimum=1, maximum=570, value=4, step=1)
-    fbs = gr.Radio([1, 2, 3, 4, 5], label="Blood Sugar", info="Do you have high blood sugar? [1: Low, 5: Very High]")
-
-    restecg = gr.Slider(label="EKG Score", info="Rate your EKG score, 1 being low, 5 being high", minimum=1, maximum=5, value=4, step=1)
-    thalachh = gr.Number(label="Maximum Heart rate", value=4)
-   
-    exng = gr.Radio(["No", "Yes"], label="Exercise Risk", info="Do you have heart risk during exercise?", type="index")
-    oldpeak = gr.Slider(label="Rate your ST depression during exercise:", minimum=1, maximum=10, value=5, step=0.1)
-    slp = gr.Radio(["unslopping", "flat", "downsloping"], label="Slope" ,info="What is the slope of your peak ST segment", type="index")
-    caa = gr.Slider(label="Major Blood Vessels", info="How many major blood vessels do you have?", minimum=0, maximum=4, value=2, step=1)
-    thall = gr.Slider(label="thall Score", info="What is your thall score?", minimum=0, maximum=3, value=2, step=1)
+    with gr.Row():
+        with gr.Column():
+            gr.Markdown(f"# {title}")
+            gr.Markdown(f"## How does it work?")
+            gr.Markdown(description1)
+            gr.Markdown("""---""")
+            gr.Markdown(description2)
+        with gr.Column():
+            gr.Image("dr.jpg")        
     
-    submit_btn = gr.Button("Analyze")
+    gr.Markdown("""---""")
+    
+    with gr.Row():
+        with gr.Column():
+            gr.Markdown(f"## Edit the Inputs Below:")
+            gr.Markdown("""---""")
+
+            with gr.Row():
+                age = gr.Number(label="Age", info="How old are you?", value=40)
+                # sex = gr.Radio(["Male", "Female"], label = "What Gender are you?", type = "index")
+                sex = gr.Radio(["Male", "Female"], label="Sex", info="What gender are you?", type="index")
+                # sex = gr.Radio(choices=["Male", "Female"])
+            
+            cp = gr.Radio(["Typical Angina", "Atypical Angina", "Non-anginal Pain", "Asymptomatic"], label="Chest Pain", info="What kind of chest pain do you have?", type="index")
+            # cp = gr.Slider(label="Chest Pain Type", minimum=1, maximum=5, value=4, step=1)
+             # trtbps = gr.Slider(label="Resting blood pressure (in mm Hg)", minimum=1, maximum=200, value=4, step=1) 
+            trtbps = gr.Number(label="trtbps", value=100) 
+            chol = gr.Number(label="chol", value=70) 
+            fbs = gr.Radio(["False", "True"], label="fbs",  info="Is your fasting blood sugar > 120 mg/dl?" , type="index")
+
+            # restecg = gr.Slider(label="Resting ECG Score", minimum=1, maximum=5, value=4, step=1)
+            restecg = gr.Dropdown(["Normal", "Having ST-T wave abnormality", "Showing probable or definite left ventricular hypertrophy by Estes' criteria"], label="rest_ecg", type="index")
+            thalachh = gr.Slider(label="thalach Score", minimum=1, maximum=205, value=4, step=1)
+            exng = gr.Radio(["No", "Yes"], label="Exercise Induced Angina", type="index")
+            oldpeak = gr.Slider(label="Oldpeak Score", minimum=1, maximum=10, value=4, step=1)
+            slp = gr.Slider(label="Slp Score", minimum=1, maximum=5, value=4, step=1)
+            caa = gr.Slider(label="Number of Major Vessels", minimum=1, maximum=3, value=3, step=1)
+            thall = gr.Slider(label="Thall Score", minimum=1, maximum=5, value=4, step=1)
+
+            
+            
+            
+
+        with gr.Column():
+            gr.Markdown(f"## Output:")
+            gr.Markdown("""---""")
+            with gr.Column(visible=True) as output_col:
+                label = gr.Label(label = "Predicted Label")
+                local_plot = gr.Plot(label = 'Shap:')
+
+            gr.Markdown(f"## Examples:")
+            gr.Markdown("""---""")
+            gr.Markdown("### Click on any of the examples below to see how it works:")
+            gr.Examples([[24,"Male","Typical Angina",4,5,"True","Normal",4,"No",5,1,2,3], [24,"Female","Asymptomatic",4,5,"False","Normal",2,"Yes",1,1,2,3]], [age, sex, cp, trtbps, chol, fbs, restecg, thalachh,exng,oldpeak,slp,caa,thall], [label,local_plot], main_func, cache_examples=True)
 
-    with gr.Column(visible=True) as output_col:
-        label = gr.Label(label = "Predicted Label")
-        local_plot = gr.Plot(label = 'Shap:')
+        
+    submit_btn = gr.Button("Analyze", variant="primary")
+
+
+    gr.Markdown("""---""")
+    gr.Markdown(f"## Data Dictionary:")
+    gr.Markdown("""
+    
+Age : Age of the patient
+Sex : Sex of the patient
+trtbps : resting blood pressure (in mm Hg)
+chol : cholestoral in mg/dl fetched via BMI sensor
+fbs : (fasting blood sugar > 120 mg/dl) (1 = true; 0 = false)
+rest_ecg : resting electrocardiographic results
+    Value 0: normal
+    Value 1: having ST-T wave abnormality (T wave inversions and/or ST elevation or depression of > 0.05 mV)
+    Value 2: showing probable or definite left ventricular hypertrophy by Estes' criteria
+    
+thalach : maximum heart rate achieved
+target : 0 = less chance of heart attack 1= more chance of heart attack""")
+    
 
     submit_btn.click(
         main_func,
         [age, sex, cp, trtbps, chol, fbs, restecg, thalachh,exng,oldpeak,slp,caa,thall],
         [label,local_plot], api_name="Heart_Predictor"
     )
-    
-    gr.Markdown("### Click on any of the examples below to see how it works:")
-    gr.Examples([[77,"Male",3,200,564, 1,2,202,1,6.2,2,4,1], [24,"Male",4,4,5,3,3,2,1,1,1,2,3]], [age, sex, cp, trtbps, chol, fbs, restecg, thalachh,exng,oldpeak,slp,caa,thall], [label,local_plot], main_func, cache_examples=True)
-
 
+    
 demo.launch()