Update openvoice_app.py

openvoice_app.py

@@ -7,6 +7,8 @@ from dotenv import load_dotenv
 from elevenlabs.client import ElevenLabs
 from elevenlabs import play, save
 import base64
+import psycopg2
+from urllib.parse import urlparse, parse_qs
 
 # Load environment variables
 load_dotenv()
@@ -84,18 +86,45 @@ def send_email_with_file(recipient_email, file_path, subject, body):
         print(f"An error occurred while sending email: {e}")
         return False
 
-# New function to handle consent
-def check_consent(agreement):
-    if agreement:
-        return gr.update(visible=True), gr.update(visible=False)
-    else:
-        return gr.update(visible=False), gr.update(visible=True)
+# Database connection details
+connection_string = os.environ.get("DATABASE_URL")
+result = urlparse(connection_string)
+user = result.username
+password = result.password
+host = result.hostname
+port = result.port
+database = result.path[1:]
+sslmode = parse_qs(result.query)['sslmode'][0]
+
+# Function to add user information to the database
+def add_user_info_to_db(email, order, sample_audio, name):
+    connection = psycopg2.connect(
+        dbname=database,
+        user=user,
+        password=password,
+        host=host,
+        port=port,
+        sslmode=sslmode
+    )
+    cursor = connection.cursor()
+    
+    insert_query = """
+    INSERT INTO example_table (email, order, sample_audio, name)
+    VALUES (%s, %s, %s, %s);
+    """
+    try:
+        cursor.execute(insert_query, (email, order, sample_audio, name))
+        connection.commit()
+        print("User information added to the database successfully")
+    except Exception as error:
+        print(f"Error occurred during data insertion: {error}")
+        connection.rollback()
+    finally:
+        cursor.close()
+        connection.close()
 
 # Predict function
-def predict(agreement, prompt, style, audio_file_pth, voice_name, customer_email):
-    if not agreement:
-        return "You must agree to the terms before using this application.", None, None
-    
+def predict(prompt, style, audio_file_pth, voice_name, customer_email, order):
     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"
@@ -124,29 +153,16 @@ def predict(agreement, prompt, style, audio_file_pth, voice_name, customer_email
     else:
         text_hint += "Failed to send email with the voice file. Please try again later.\n"
 
+    # Add user information to the database
+    with open(audio_file_pth, "rb") as f:
+        sample_audio = f.read()
+    add_user_info_to_db(customer_email, order, sample_audio, voice_name)
+
     return text_hint, save_path, audio_file_pth
 
 # Gradio interface setup
 with gr.Blocks(gr.themes.Glass()) as demo:
-    # Consent view
-    with gr.Row(visible=True) as consent_row:
-        consent_text = gr.Markdown(
-        """
-        # Consent Agreement
-        
-        By using this application, you agree to the following terms:
-        
-        1. This app will only be used for CuddleBuddy products.
-        2. CuddleBuddy and any of its partners are not responsible for any misuse of this application.
-        3. You understand that the voice cloning technology should be used responsibly and ethically.
-        
-        Do you agree to these terms?
-        """
-        )
-        consent_checkbox = gr.Checkbox(label="I agree to the terms")
-    
-    # Main application view
-    with gr.Row(visible=False) as main_app:
+    with gr.Row():
         with gr.Column():
             input_text_gr = gr.Textbox(
                 label="Create This",
@@ -166,9 +182,13 @@ with gr.Blocks(gr.themes.Glass()) as demo:
                 sources=["upload"],
             )
             voice_name_gr = gr.Textbox(
-                label="Your name and Product you bought",
+                label="Your name",
                 value="Sam"
             )
+            order_gr = gr.Textbox(
+                label="Your order",
+                value="Sample Order",
+            )
             customer_email_gr = gr.Textbox(
                 label="Your Email",
                 info="We'll send you a downloadable file to this email address."
@@ -180,19 +200,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 ")
 
-    # Connect the consent checkbox to show/hide views
-    consent_checkbox.change(
-        check_consent,
-        inputs=[consent_checkbox],
-        outputs=[main_app, consent_row]
-    )
-
-    # Modify the click event to include the consent check
-    tts_button.click(
-        predict,
-        inputs=[consent_checkbox, input_text_gr, style_gr, ref_gr, voice_name_gr, customer_email_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, order_gr], outputs=[out_text_gr, audio_gr, ref_audio_gr])
 
     demo.queue()
     demo.launch(debug=True, show_api=False, share=args.share)
@@ -202,4 +210,4 @@ footer {visibility: hidden}
 audio .btn-container {display: none}
 """
 
-demo.add_css(css)
+demo.add_css(css)