Update app.py

app.py

@@ -17,24 +17,27 @@ def feature_select(dataset, feature, hue = None):
     df = pd.read_csv(dataset.name)
     non_numeric_cols = df.select_dtypes('object').columns.tolist()
     
-    plot1 = plt.figure()
-    if hue:
-      sns.histplot(data = df, x = df[feature], kde = True, hue = hue)
-    else:
-      sns.histplot(data = df, x = df[feature], kde = True)
-    
     if feature in non_numeric_cols:
+      kde = False
       plot2 = plt.figure()
       if hue:
-        sns.countplot(x = df[feature], data = df, palette='rainbow', hue = hue)
+        sns.countplot(x = feature, data = df, palette='rainbow', hue = hue)
       else:
-        sns.countplot(x = df[feature], data = df, palette='rainbow')
+        sns.countplot(x = feature, data = df, palette='rainbow')
     else:
+      kde = True
       plot2 = plt.figure()
       if hue:
-        sns.boxplot(x = df[feature], hue = hue)
+        sns.boxplot(x = feature, data = df, hue = hue)
       else:
-        sns.boxplot(x = df[feature])
+        sns.boxplot(x = feature, data = df )
+
+    plot1 = plt.figure()
+    if hue:
+      sns.histplot(data = df, x = feature, kde = kde, hue = hue, multiple="stack")
+    else:
+      sns.histplot(data = df, x = feature, kde = kde)
+   
 
     return plot1, plot2