Update app.py

app.py

@@ -3,6 +3,7 @@ import streamlit as st
 import os
 import google.generativeai as genai
 import random
+import datetime
 from streamlit import session_state as state
 from formulas import email_formulas  # Updated import statement
 from angles import angles
@@ -407,6 +408,19 @@ if submit:
                         <p>{generated_emails}</p>
                     </div>
                 """, unsafe_allow_html=True)
+                
+                # Add download button for the generated emails
+                if generated_emails:
+                    # Get current timestamp for the filename
+                    timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
+                    
+                    # Download button
+                    st.download_button(
+                        label="DESCARGAR EMAILS",
+                        data=generated_emails,
+                        file_name=f"emails_persuasivos_{timestamp}.txt",
+                        mime="text/plain"
+                    )
         except ValueError as e:
             col2.error(f"Error: {str(e)}")
     else: