Update app.py

app.py

@@ -13,6 +13,8 @@ import plotly.express as px
 import json
 from difflib import get_close_matches
 from xlsxwriter import Workbook
+import io
+
 
 
 
@@ -318,16 +320,18 @@ def fuel_norm_distribution_dashboard(root_dir, state_mapping_file, start_year=20
     
             # Download button
             st.markdown("### 📥 Download Filtered Data")
-            download_format = st.radio("Select format", ["CSV", "Excel"], horizontal=True)
-            if download_format == "CSV":
-                csv_data = filtered_df.to_csv(index=False).encode("utf-8")
-                st.download_button("Download CSV", csv_data, file_name="fuel_distribution_filtered.csv", mime="text/csv")
-            else:
-                output = BytesIO()
-                with pd.ExcelWriter(output, engine='xlsxwriter') as writer:
-                    filtered_df.to_excel(writer, index=False, sheet_name="Filtered Data")
-                st.download_button("Download Excel", output.getvalue(), file_name="fuel_distribution_filtered.xlsx", mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet")
-    
+            
+            output = BytesIO()
+            with pd.ExcelWriter(output, engine='xlsxwriter') as writer:
+                filtered_df.to_excel(writer, index=False, sheet_name="Filtered Data")
+            
+            st.download_button(
+                label="Download Excel",
+                data=output.getvalue(),
+                file_name="fuel_distribution_filtered.xlsx",
+                mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
+            )formats-officedocument.spreadsheetml.sheet")
+                
 
     with tab2:
         st.subheader("Emission Norm Distribution by State")
@@ -349,17 +353,18 @@ def fuel_norm_distribution_dashboard(root_dir, state_mapping_file, start_year=20
             st.plotly_chart(fig2, use_container_width=True)
     
             # Download button
-            st.markdown("### 📥 Download Filtered Norms Data")
-            download_format_norms = st.radio("Select format", ["CSV", "Excel"], horizontal=True, key="norm_download_format")
-            if download_format_norms == "CSV":
-                csv_norm_data = filtered_norms.to_csv(index=False).encode("utf-8")
-                st.download_button("Download CSV", csv_norm_data, file_name="norm_distribution_filtered.csv", mime="text/csv")
-            else:
-                norm_output = BytesIO()
-                with pd.ExcelWriter(norm_output, engine='xlsxwriter') as writer:
-                    filtered_norms.to_excel(writer, index=False, sheet_name="Filtered Norms")
-                st.download_button("Download Excel", norm_output.getvalue(), file_name="norm_distribution_filtered.xlsx", mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet")
-
+           st.markdown("### 📥 Download Filtered Norms Data")
+            
+            norm_output = BytesIO()
+            with pd.ExcelWriter(norm_output, engine='xlsxwriter') as writer:
+                filtered_norms.to_excel(writer, index=False, sheet_name="Filtered Norms")
+            
+            st.download_button(
+                label="Download Excel",
+                data=norm_output.getvalue(),
+                file_name="norm_distribution_filtered.xlsx",
+                mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
+            )
     with tab3:
         st.subheader("Norm Emissions by Fuel Type and Year")