Update openvoice_app.py

openvoice_app.py

@@ -84,8 +84,18 @@ 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)
+
 # Predict function
-def predict(prompt, style, audio_file_pth, voice_name, customer_email):
+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
+    
     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"
@@ -116,74 +126,73 @@ def predict(prompt, style, audio_file_pth, voice_name, customer_email):
 
     return text_hint, save_path, audio_file_pth
 
-# Consent agreement text
-CONSENT_TEXT = """
-By using this app, you agree to the following:
-
-1. This app is intended for use only with CuddleBuddy products.
-2. CuddleBuddy and any of its partners are not responsible for any misuse of this app or its output.
-3. You will not use this app for any illegal or unethical purposes.
-4. You understand that the voice cloning technology should be used responsibly and with respect for others' privacy and rights.
-
-Do you agree to these terms?
-"""
-
-# Function to handle consent
-def handle_consent(agree):
-    if agree:
-        return gr.update(visible=True), gr.update(visible=False)
-    else:
-        return gr.update(visible=False), gr.update(visible=True)
-
 # Gradio interface setup
 with gr.Blocks(gr.themes.Glass()) as demo:
-    consent_view = gr.Column(visible=True)
-    main_view = gr.Column(visible=False)
-    
-    with consent_view:
-        gr.Markdown("# Consent Agreement")
-        gr.Markdown(CONSENT_TEXT)
-        agree_button = gr.Button("I Agree")
-        disagree_button = gr.Button("I Disagree")
+    # 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")
     
-    with main_view:
-        with gr.Row():
-            with gr.Column():
-                input_text_gr = gr.Textbox(
-                    label="Create This",
-                    info="One or two sentences at a time is better. Up to 200 text characters.",
-                    value="He hoped there would be stew for dinner, turnips and carrots and bruised potatoes and fat mutton pieces to be ladled out in thick, peppered, flour-fattened sauce.",
-                )
-                style_gr = gr.Dropdown(
-                    label="Style",
-                    choices=['default', 'whispering', 'cheerful', 'terrified', 'angry', 'sad', 'friendly'],
-                    info="Please upload a reference audio file that is at least 1 minute long. For best results, ensure the audio is clear. You can use Adobe Podcast Enhance(https://podcast.adobe.com/enhance) to improve the audio quality before uploading.",
-                    max_choices=1,
-                    value="default",
-                )
-                ref_gr = gr.Audio(
-                    label="Original Audio",
-                    type="filepath",
-                    sources=["upload"],
-                )
-                voice_name_gr = gr.Textbox(
-                    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():
-                out_text_gr = gr.Text(label="Info")
-                audio_gr = gr.Audio(label="Replicated Sound", autoplay=True)
-                ref_audio_gr = gr.Audio(label="Original Audio Used ")
-
-    agree_button.click(handle_consent, inputs=[gr.Boolean(True)], outputs=[main_view, consent_view])
-    disagree_button.click(handle_consent, inputs=[gr.Boolean(False)], outputs=[main_view, consent_view])
-    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])
+    # Main application view
+    with gr.Row(visible=False) as main_app:
+        with gr.Column():
+            input_text_gr = gr.Textbox(
+                label="Create This",
+                info="One or two sentences at a time is better. Up to 200 text characters.",
+                value="He hoped there would be stew for dinner, turnips and carrots and bruised potatoes and fat mutton pieces to be ladled out in thick, peppered, flour-fattened sauce.",
+            )
+            style_gr = gr.Dropdown(
+                label="Style",
+                choices=['default', 'whispering', 'cheerful', 'terrified', 'angry', 'sad', 'friendly'],
+                info="Please upload a reference audio file that is at least 1 minute long. For best results, ensure the audio is clear. You can use Adobe Podcast Enhance(https://podcast.adobe.com/enhance) to improve the audio quality before uploading.",
+                max_choices=1,
+                value="default",
+            )
+            ref_gr = gr.Audio(
+                label="Original Audio",
+                type="filepath",
+                sources=["upload"],
+            )
+            voice_name_gr = gr.Textbox(
+                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():
+            out_text_gr = gr.Text(label="Info")
+            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]
+    )
 
     demo.queue()
     demo.launch(debug=True, show_api=False, share=args.share)