app

app.py

@@ -40,13 +40,19 @@ def outbreak(plot_type):
         return fig
     if plot_type == "Business Travel vs Attrition":
         fig = plt.figure()
-        ax = sns.countplot(x="Business Travel vs Attrition [0: No travel, 1: Travel Frequently, 2: Travel Rarely]", hue="Attrition", data=data_encoded)
+        ax = sns.countplot(x="BusinessTravel", hue="Attrition", data=data_encoded)
+        for p in ax.patches:
+            ax.annotate('{}'.format(p.get_height()), (p.get_x(), p.get_height()+1))
+        return fig
+    if plot_type == "Employee Experience vs Attrition":
+        fig = plt.figure()
+        ax = sns.countplot(x="EmployeeExperience", hue="Attrition", data=data_encoded)
         for p in ax.patches:
             ax.annotate('{}'.format(p.get_height()), (p.get_x(), p.get_height()+1))
         return fig
 
 inputs = [
-        gr.Dropdown(["Find Data Correlation", "Filter Correlation Data", "Business Travel vs Attrition", "Age vs Attrition",], label="Data Correlation and Visualization")
+        gr.Dropdown(["Find Data Correlation", "Filter Correlation Data", "Business Travel vs Attrition", "Employee Experience vs Attrition", "Age vs Attrition",], label="Data Correlation and Visualization")
     ]
 
 outputs = gr.Plot()