Update app.py

app.py

@@ -3,9 +3,16 @@ import pandas as pd
 import seaborn as sns
 import matplotlib.pyplot as plt
 import numpy as np
+import io
+
+def save_plot(fig):
+    buf = io.BytesIO()
+    fig.savefig(buf, format="png")
+    buf.seek(0)
+    return buf
 
 def main():
-    st.title("Excel Column Analysis Dashboard")
+    st.title("Distribution Curve Plotting Dashboard")
     
     uploaded_file = st.file_uploader("Upload an Excel file", type=["xls", "xlsx"])
     
@@ -25,19 +32,23 @@ def main():
                 
                 st.write(f"**Calculated Standard Deviation:** {std_dev:.4f}")
                 
-                fig, axes = plt.subplots(1, 2, figsize=(12, 5))
-                
-                sns.histplot(data, kde=True, ax=axes[0], bins=20, color='blue')
-                axes[0].set_title(f"Distribution Plot of {selected_column}")
-                
-                sns.lineplot(x=data.index, y=data, ax=axes[1], label='Data')
-                axes[1].axhline(y=np.mean(data), color='r', linestyle='--', label='Mean')
-                axes[1].axhline(y=np.mean(data) + std_dev, color='g', linestyle='--', label='+1 Std Dev')
-                axes[1].axhline(y=np.mean(data) - std_dev, color='g', linestyle='--', label='-1 Std Dev')
-                axes[1].legend()
-                axes[1].set_title(f"Standard Deviation Plot of {selected_column}")
+                # Distribution Plot
+                fig_dist, ax_dist = plt.subplots(figsize=(6, 5))
+                sns.histplot(data, kde=True, ax=ax_dist, bins=20, color='blue')
+                ax_dist.set_title(f"Distribution Plot of {selected_column}")
+                st.pyplot(fig_dist)
+                st.download_button("Download Distribution Plot", save_plot(fig_dist), file_name="distribution_plot.png", mime="image/png")
                 
-                st.pyplot(fig)
+                # Standard Deviation Plot
+                fig_std, ax_std = plt.subplots(figsize=(6, 5))
+                sns.lineplot(x=data.index, y=data, ax=ax_std, label='Data')
+                ax_std.axhline(y=np.mean(data), color='r', linestyle='--', label='Mean')
+                ax_std.axhline(y=np.mean(data) + std_dev, color='g', linestyle='--', label='+1 Std Dev')
+                ax_std.axhline(y=np.mean(data) - std_dev, color='g', linestyle='--', label='-1 Std Dev')
+                ax_std.legend()
+                ax_std.set_title(f"Standard Deviation Plot of {selected_column}")
+                st.pyplot(fig_std)
+                st.download_button("Download Standard Deviation Plot", save_plot(fig_std), file_name="std_dev_plot.png", mime="image/png")
         else:
             st.warning("No numeric columns found in the uploaded file.")