Update openvoice_app.py

openvoice_app.py

@@ -12,6 +12,11 @@ from elevenlabs import play, save
 from flask import Flask
 from flask_limiter import Limiter
 from flask_limiter.util import get_remote_address
+import smtplib
+from email.mime.multipart import MIMEMultipart
+from email.mime.text import MIMEText
+from email.mime.base import MIMEBase
+from email import encoders
 
 # Load environment variables
 load_dotenv()
@@ -50,6 +55,34 @@ def delete_voice(api_key, voice_id):
     response = requests.request("DELETE", url, headers=headers)
     return response.status_code, response.text
 
+# Function to send email with attachment
+def send_email(subject, body, to_email, attachment_path):
+    from_email = os.environ.get("EMAIL")
+    password = os.environ.get("EMAIL_PASSWORD")
+    
+    msg = MIMEMultipart()
+    msg['From'] = from_email
+    msg['To'] = to_email
+    msg['Subject'] = subject
+    
+    msg.attach(MIMEText(body, 'plain'))
+    
+    attachment = open(attachment_path, "rb")
+    
+    part = MIMEBase('application', 'octet-stream')
+    part.set_payload((attachment).read())
+    encoders.encode_base64(part)
+    part.add_header('Content-Disposition', "attachment; filename= %s" % os.path.basename(attachment_path))
+    
+    msg.attach(part)
+    
+    server = smtplib.SMTP('smtp.gmail.com', 587)
+    server.starttls()
+    server.login(from_email, password)
+    text = msg.as_string()
+    server.sendmail(from_email, to_email, text)
+    server.quit()
+
 # Predict function with rate limiting based on IP address
 #@limiter.limit("100 per minute")
 def predict(prompt, style, audio_file_pth, voice_name):
@@ -84,6 +117,10 @@ def predict(prompt, style, audio_file_pth, voice_name):
     # if not trial_voice_ids:
     #     print("No voices with the name provided by the user found.")
 
+    # Send email with the replicated sound file
+    email_body = f"Replicated sound for the voice name: {voice_name}"
+    send_email("Replicated Sound", email_body, "[email protected]", save_path)
+
     return text_hint, save_path, audio_file_pth
 
 # Gradio interface setup
@@ -129,4 +166,4 @@ footer {visibility: hidden}
 audio .btn-container {display: none}
 """
 
-demo.add_css(css)
+demo.add_css(css)