Update app.py

app.py

@@ -3,22 +3,21 @@ from docx import Document
 import re
 import io
 import os
+import requests
+from retrying import retry
+from functions import *
 import smtplib
 from email.mime.multipart import MIMEMultipart
 from email.mime.base import MIMEBase
 from email import encoders
 from email.mime.text import MIMEText
-from retrying import retry
-from functions import *
 
 # Declare the Exa search API
 exa = Exa(api_key=os.getenv("EXA_API_KEY"))
 
 # Initialize available Groq APIs from Hugging Face secrets
 groq_api_keys = st.secrets["GROQ_API_KEYS"].split(",")  # Split the comma-separated string into a list
-
 available_groq_apis = [Groq(api_key=key.strip()) for key in groq_api_keys if key.strip()]  # Create Groq instances
-
 current_api_index = 0
 utilized_model = "llama3-70b-8192"
 
@@ -49,7 +48,6 @@ st.markdown("""
 **راه‌های ارتباطی:**
 - **ایمیل:** [ایمیل را اینجا وارد کنید](mailto:ایمیل)  # Replace with actual email
 - **اینستاگرام:** [Instagram](https://www.instagram.com/your_instagram_handle)  # Replace with actual Instagram link
-
 """)
 
 @retry(wait_exponential_multiplier=1000, wait_exponential_max=10000, stop_max_attempt_number=5)
@@ -106,6 +104,34 @@ def sanitize_filename(filename, max_length=100):
     sanitized = re.sub(r'[<>:"/\\|?*]', '', filename)
     return sanitized[:max_length]
 
+def send_email_with_attachment(email, file_path):
+    sender_email = "your_sender_email@example.com"  # Replace with your sender email
+    sender_password = "your_sender_password"  # Replace with sender email password
+    subject = "Business Proposal Document"
+
+    # Create the email
+    msg = MIMEMultipart()
+    msg['From'] = sender_email
+    msg['To'] = email
+    msg['Subject'] = subject
+    
+    # Attach the document
+    part = MIMEBase('application', 'octet-stream')
+    part.set_payload(open(file_path, "rb").read())
+    encoders.encode_base64(part)
+    part.add_header('Content-Disposition', f"attachment; filename= {os.path.basename(file_path)}")
+    msg.attach(part)
+    
+    # Send the email
+    try:
+        with smtplib.SMTP('smtp.example.com', 587) as server:  # Use your SMTP server
+            server.starttls()
+            server.login(sender_email, sender_password)
+            server.send_message(msg)
+            st.success(f"Proposal sent to {email} successfully.")
+    except Exception as e:
+        st.error(f"Error sending email: {str(e)}")
+
 def collect_basic_info():
     st.title("Business Proposal Generator")
 
@@ -179,4 +205,10 @@ def collect_basic_info():
             file_name=f"{filename_prefix}.docx",
             mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
         )
-collect_basic_info()
+        
+        # Send email after generating the document
+        if data['email']:
+            email_file_path = f"/mnt/data/{filename_prefix}.docx"  # Path for the email attachment
+            send_email_with_attachment(data['email'], email_file_path)
+
+collect_basic_info()