Update openvoice_app.py

openvoice_app.py

@@ -2,7 +2,10 @@ import os
 import torch
 import argparse
 import gradio as gr
-import requests
+import smtplib
+from email.mime.text import MIMEText
+from email.mime.multipart import MIMEMultipart
+from email.mime.application import MIMEApplication
 from openvoice import se_extractor
 from openvoice.api import BaseSpeakerTTS, ToneColorConverter
 from dotenv import load_dotenv
@@ -36,24 +39,40 @@ os.makedirs(output_dir, exist_ok=True)
 api_key = os.environ.get("ELEVENLABS_API_KEY")
 supported_languages = ['zh', 'en']
 
-# Function to get all voices
-def get_voices(api_key):
-    url = "https://api.elevenlabs.io/v1/voices"
-    headers = {"xi-api-key": api_key}
-    response = requests.request("GET", url, headers=headers)
-    return response.json()
+# Gmail configuration
+GMAIL_USER = os.environ.get('GMAIL_USER')
+GMAIL_PASSWORD = os.environ.get('GMAIL_PASSWORD')  # This should be an app password, not your regular password
 
-# Function to delete a voice by ID
-def delete_voice(api_key, voice_id):
-    url = f"https://api.elevenlabs.io/v1/voices/{voice_id}"
-    headers = {"xi-api-key": api_key}
-    response = requests.request("DELETE", url, headers=headers)
-    return response.status_code, response.text
+# Function to send email with downloadable file using Gmail SMTP
+def send_email_with_file(recipient_email, file_path, subject, body):
+    try:
+        msg = MIMEMultipart()
+        msg['From'] = GMAIL_USER
+        msg['To'] = recipient_email
+        msg['Subject'] = subject
+
+        msg.attach(MIMEText(body, 'plain'))
+
+        with open(file_path, "rb") as file:
+            part = MIMEApplication(file.read(), Name=os.path.basename(file_path))
+        part['Content-Disposition'] = f'attachment; filename="{os.path.basename(file_path)}"'
+        msg.attach(part)
+
+        server = smtplib.SMTP_SSL('smtp.gmail.com', 465)
+        server.ehlo()
+        server.login(GMAIL_USER, GMAIL_PASSWORD)
+        server.send_message(msg)
+        server.close()
+
+        return True
+    except Exception as e:
+        print(f"An error occurred while sending email: {e}")
+        return False
 
 # Predict function with rate limiting based on IP address
-#@limiter.limit("100 per minute")
-def predict(prompt, style, audio_file_pth, voice_name):
-    text_hint = ''
+@limiter.limit("100 per minute")
+def predict(prompt, style, audio_file_pth, voice_name, customer_email):
+    text_hint = 'Your file will only be saved for 24 hours.\n'
     if len(prompt) < 2:
         text_hint += "[ERROR] Please provide a longer prompt text.\n"
         return text_hint, None, None
@@ -69,20 +88,16 @@ def predict(prompt, style, audio_file_pth, voice_name):
     )
     # Generate audio from text
     audio = client.generate(text=prompt, voice=voice)
-    save(audio, f'{output_dir}/output.wav')
-    
     save_path = f'{output_dir}/output.wav'
-    data = get_voices(api_key)
-    # Find all voice IDs with the name provided by the user
-    trial_voice_ids = [voice.get("voice_id") for voice in data['voices'] if voice.get("name") == voice_name]
-
-    # # Delete each voice with the name provided by the user
-    # for voice_id in trial_voice_ids:
-    #     status_code, response_text = delete_voice(api_key, voice_id)
-    #     print(f"Deleted voice ID {voice_id}: Status Code {status_code}, Response {response_text}")
-
-    # if not trial_voice_ids:
-    #     print("No voices with the name provided by the user found.")
+    save(audio, save_path)
+    
+    # Send email with downloadable file
+    subject = "Your Voice Clone File"
+    body = "Thank you for using our Voice Clone service. Your file is attached."
+    if send_email_with_file(customer_email, save_path, subject, body):
+        text_hint += "Email sent successfully with the voice file.\n"
+    else:
+        text_hint += "Failed to send email with the voice file. Please try again later.\n"
 
     return text_hint, save_path, audio_file_pth
 
@@ -111,6 +126,10 @@ with gr.Blocks(gr.themes.Glass()) as demo:
                 label="Your name and Product you bought",
                 value="Sam"
             )
+            customer_email_gr = gr.Textbox(
+                label="Your Email",
+                info="We'll send you a downloadable file to this email address."
+            )
             tts_button = gr.Button("Start", elem_id="send-btn", visible=True)
 
         with gr.Column():
@@ -118,7 +137,7 @@ with gr.Blocks(gr.themes.Glass()) as demo:
             audio_gr = gr.Audio(label="Replicated Sound", autoplay=True)
             ref_audio_gr = gr.Audio(label="Original Audio Used ")
 
-            tts_button.click(predict, [input_text_gr, style_gr, ref_gr, voice_name_gr], outputs=[out_text_gr, audio_gr, ref_audio_gr])
+            tts_button.click(predict, [input_text_gr, style_gr, ref_gr, voice_name_gr, customer_email_gr], outputs=[out_text_gr, audio_gr, ref_audio_gr])
 
     demo.queue()
     demo.launch(debug=True, show_api=False, share=args.share)