Update app.py

app.py

@@ -855,14 +855,58 @@ def generate_video(user_input, resolution, caption_option):
 
 
 
+import gradio as gr
+import shutil
+import os
+import random
+
+# Human-friendly voice list
+VOICE_CHOICES = {
+    'Emma (Female)': 'af_heart',
+    'Bella (Female)': 'af_bella',
+    'Nicole (Female)': 'af_nicole',
+    'Aoede (Female)': 'af_aoede',
+    'Kore (Female)': 'af_kore',
+    'Sarah (Female)': 'af_sarah',
+    'Nova (Female)': 'af_nova',
+    'Sky (Female)': 'af_sky',
+    'Alloy (Female)': 'af_alloy',
+    'Jessica (Female)': 'af_jessica',
+    'River (Female)': 'af_river',
+    'Michael (Male)': 'am_michael',
+    'Fenrir (Male)': 'am_fenrir',
+    'Puck (Male)': 'am_puck',
+    'Echo (Male)': 'am_echo',
+    'Eric (Male)': 'am_eric',
+    'Liam (Male)': 'am_liam',
+    'Onyx (Male)': 'am_onyx',
+    'Santa (Male)': 'am_santa',
+    'Adam (Male)': 'am_adam',
+    'Emma 🇬🇧 (Female)': 'bf_emma',
+    'Isabella 🇬🇧 (Female)': 'bf_isabella',
+    'Alice 🇬🇧 (Female)': 'bf_alice',
+    'Lily 🇬🇧 (Female)': 'bf_lily',
+    'George 🇬🇧 (Male)': 'bm_george',
+    'Fable 🇬🇧 (Male)': 'bm_fable',
+    'Lewis 🇬🇧 (Male)': 'bm_lewis',
+    'Daniel 🇬🇧 (Male)': 'bm_daniel'
+}
+
 def handle_music_upload(music_file):
     if music_file is not None:
         target_path = "music.mp3"
         shutil.copy(music_file, target_path)
         print(f"Uploaded music saved as: {target_path}")
 
-def generate_video_with_music(user_input, resolution, caption_option, music_file):
+def generate_video_with_music(user_input, resolution, caption_option, music_file, voice, video_clip_probability):
     handle_music_upload(music_file)
+
+    # Set global or pass values to your other functions as needed:
+    selected_voice = VOICE_CHOICES[voice]
+    print(f"Selected Voice: {selected_voice}")
+    print(f"Video Clip Probability: {video_clip_probability}%")
+
+    # Your existing generate_video() call — you might want to pass the voice and clip probability
     return generate_video(user_input, resolution, caption_option)
 
 iface = gr.Interface(
@@ -871,12 +915,13 @@ iface = gr.Interface(
         gr.Textbox(label="Video Concept", placeholder="Enter your video concept here..."),
         gr.Radio(["Full", "Short"], label="Resolution", value="Full"),
         gr.Radio(["Yes", "No"], label="Captions", value="Yes"),
-        gr.File(label="Upload Background Music (MP3)", file_types=[".mp3"])
+        gr.File(label="Upload Background Music (MP3)", file_types=[".mp3"]),
+        gr.Dropdown(choices=list(VOICE_CHOICES.keys()), label="Choose Voice", value="Emma (Female)"),
+        gr.Slider(0, 100, value=25, step=1, label="Video Clip Usage Probability (%)")
     ],
     outputs=gr.Video(label="Generated Video"),
     title="AI Documentary Video Generator",
-    description="Upload your background music and create a funny documentary-style video."
+    description="Upload your background music, choose a voice, set video clip ratio, and create a funny AI documentary!"
 )
 
 iface.launch(share=True)
-