Update app.py

app.py

@@ -51,17 +51,17 @@ def plot_graphs(data):
     st.write("Explore key insights through visualizations.")
 
     # Histogram for depression
-    st.markdown("#### Histogram of Depression Levels")
-    fig, ax = plt.subplots(figsize=(6, 4))
-    ax.hist(data['depression'], bins=20, color='skyblue', edgecolor='black')
-    ax.set_title("Histogram of Depression Levels")
-    ax.set_xlabel("Depression Level")
+    st.markdown("#### Histogram of Stress Levels")
+    fig, ax = plt.subplots(figsize=(4, 2))
+    ax.hist(data['stress_level'], bins=20, color='skyblue', edgecolor='black')
+    ax.set_title("Histogram of Stress Levels")
+    ax.set_xlabel("Stress Level")
     ax.set_ylabel("Frequency")
     st.pyplot(fig)
 
     # Scatter plot for anxiety vs. depression
     st.markdown("#### Scatter Plot: Anxiety vs Depression")
-    fig, ax = plt.subplots(figsize=(6, 4))
+    fig, ax = plt.subplots(figsize=(4, 2))
     sns.scatterplot(x=data['anxiety_level'], y=data['depression'], ax=ax, color='blue')
     ax.set_title("Anxiety Level vs Depression")
     ax.set_xlabel("Anxiety Level")
@@ -70,7 +70,7 @@ def plot_graphs(data):
 
     # Correlation heatmap
     st.markdown("#### Correlation Heatmap")
-    fig, ax = plt.subplots(figsize=(10, 8))
+    fig, ax = plt.subplots(figsize=(8, 6))
     sns.heatmap(data.corr(), annot=True, cmap="coolwarm", ax=ax)
     ax.set_title("Correlation Heatmap")
     st.pyplot(fig)