Update pages/EDA.py

pages/EDA.py

@@ -86,25 +86,28 @@ st.markdown("""
 # Streamlit App
 st.title("Crop Recommendation System - Feature Visualization")
 
-# Sidebar for visualization selection
-viz_option = st.sidebar.selectbox(
-    "Choose a visualization",
-    [
-        "Nitrogen Requirement per Crop", 
-        "Phosphorus Requirement per Crop", 
-        "Potassium Requirement per Crop",
-        "Temperature Distribution per Crop",
-        "Humidity Distribution per Crop",
-        "pH Value Distribution per Crop",
-        "Rainfall Distribution per Crop"
-    ]
+# List of available visualizations
+visualizations = [
+    "Nitrogen Requirement per Crop", 
+    "Phosphorus Requirement per Crop", 
+    "Potassium Requirement per Crop",
+    "Temperature Distribution per Crop",
+    "Humidity Distribution per Crop",
+    "pH Value Distribution per Crop",
+    "Rainfall Distribution per Crop"
+]
+
+# Slider for visualization selection
+viz_option = st.select_slider(
+    "Select a visualization:", 
+    options=visualizations
 )
 
 # Function to plot feature distributions per crop
 def plot_feature(feature, title):
     avg_feature_per_crop = df.groupby("Crop")[feature].mean().reset_index()
 
-    fig, axes = plt.subplots(1, 2, figsize=(14, 6))
+    fig, axes = plt.subplots(1, 2, figsize=(10, 4))  # Decreased plot size
 
     # Bar plot
     sns.barplot(data=avg_feature_per_crop, x=feature, y="Crop", palette="rainbow", ax=axes[0])
@@ -121,18 +124,15 @@ def plot_feature(feature, title):
     st.pyplot(fig)
 
 # Display selected plot
-if viz_option == "Nitrogen Requirement per Crop":
-    plot_feature("Nitrogen", "Nitrogen")
-elif viz_option == "Phosphorus Requirement per Crop":
-    plot_feature("Phosphorus", "Phosphorus")
-elif viz_option == "Potassium Requirement per Crop":
-    plot_feature("Potassium", "Potassium")
-elif viz_option == "Temperature Distribution per Crop":
-    plot_feature("Temperature", "Temperature")
-elif viz_option == "Humidity Distribution per Crop":
-    plot_feature("Humidity", "Humidity")
-elif viz_option == "pH Value Distribution per Crop":
-    plot_feature("pH_Value", "pH Value")
-elif viz_option == "Rainfall Distribution per Crop":
-    plot_feature("Rainfall", "Rainfall")
+feature_mapping = {
+    "Nitrogen Requirement per Crop": "Nitrogen",
+    "Phosphorus Requirement per Crop": "Phosphorus",
+    "Potassium Requirement per Crop": "Potassium",
+    "Temperature Distribution per Crop": "Temperature",
+    "Humidity Distribution per Crop": "Humidity",
+    "pH Value Distribution per Crop": "pH_Value",
+    "Rainfall Distribution per Crop": "Rainfall",
+}
+
+plot_feature(feature_mapping[viz_option], viz_option)