Update app.py

app.py

@@ -164,24 +164,31 @@ if st.session_state.authenticated:
         df = pd.DataFrame(emails)
 
         st.write("### Download Options:")
-        col1, col2, col3, col4 = st.columns(4)
 
-        with col1:
+        # Create a dropdown for file format selection
+        file_format = st.selectbox(
+            "Select Download Format",
+            ["CSV", "Excel", "PDF", "SQL"],
+            index=0  # CSV is default
+        )
+
+        # Download button logic based on the selected format
+        if file_format == "CSV":
             csv = df.to_csv(index=False).encode('utf-8')
-            st.download_button("Download as CSV", csv, f"{label}_emails.csv", "text/csv")
+            st.download_button("Download", csv, f"{label}_emails.csv", "text/csv")
 
-        with col2:
+        elif file_format == "Excel":
             excel_buffer = io.BytesIO()
             with pd.ExcelWriter(excel_buffer, engine='openpyxl') as writer:
                 df.to_excel(writer, index=False)
             excel_buffer.seek(0)
-            st.download_button("Download as Excel", excel_buffer, f"{label}_emails.xlsx", "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet")
+            st.download_button("Download", excel_buffer, f"{label}_emails.xlsx", "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet")
 
-        with col3:
+        elif file_format == "PDF":
             pdf_buffer = export_to_pdf(df)
-            st.download_button("Download as PDF", pdf_buffer, f"{label}_emails.pdf", "application/pdf")
+            st.download_button("Download", pdf_buffer, f"{label}_emails.pdf", "application/pdf")
 
-        with col4:
+        elif file_format == "SQL":
             conn = sqlite3.connect(':memory:')
             df.to_sql('emails', conn, index=False, if_exists='replace')
             sql_buffer = io.BytesIO()
@@ -189,6 +196,6 @@ if st.session_state.authenticated:
                 for line in conn.iterdump():
                     f.write(f'{line}\n'.encode('utf-8'))
             sql_buffer.seek(0)
-            st.download_button("Download as SQL", sql_buffer, "emails.db", "application/octet-stream")
+            st.download_button("Download", sql_buffer, "emails.db", "application/octet-stream")
 else:
     st.warning("You are not authenticated yet.")