Update spaces.py

spaces.py

@@ -23,6 +23,18 @@ client = anthropic.Anthropic(api_key=os.getenv("ANTHROPIC_API_KEY"))
 # ElevenLabs API key
 elevenlabs_api_key = os.getenv("ELEVENLABS_API_KEY")
 
+def get_voices():
+    url = "https://api.elevenlabs.io/v1/voices"
+    headers = {
+        "xi-api-key": elevenlabs_api_key
+    }
+    response = requests.get(url, headers=headers)
+    if response.status_code == 200:
+        return response.json()["voices"]
+    else:
+        print(f"Error getting voices: {response.status_code} - {response.text}")
+        return []
+
 def get_convo_list(description):
     prompt =f"Your task is to return a JSON object representing a complete conversation containing a key 'turns' with a value which is just a list of objects containing 'turn_number', an integer, and 'message', the message for that turn. Ensure you return as many turns as the user specifies, if they specify. Remember, each turn is a turn in a conversation between a phone agent (female) and a human (male). The phone agent should speak first. The conversation is described as:\n{description}.\nCritically, ensure that the human turns employ filler words (uh, uhhhh, ummmm, yeahhh, hm, hmm, etc with repeated letters to denote thinking...) and realistic language without using *sounds effects*. I repeat, do NOT use *sound effects*. Additionally, do not over-use filler words or start every human response with them. The goal is to sound realistic, not exagerrated. The AI should be conversational, employing transition phrases. The AI should always end their response with a question except when saying goodbye. Additionally, digits spaced out. For instance, the human might say: 'My phone number is 8 3 1... 5 4 8... 9 2 2 3...' instead of writing it out. They might also say 'My email is steve at gmail dot com.' where it is written out. Now provide the JSON."
     new_output = ""
@@ -259,10 +271,10 @@ def generate_conversation_video(messages, voice_ids, logo_url, male_stability, m
 
     return temp_video_path
 
-def generate_video(description, male_stability=0.65, male_style=0.35):
+def generate_video(description, female_voice_id, male_voice_id, male_stability=0.65, male_style=0.35):
     voice_ids = [
-        "cgSgspJ2msm6clMCkdW9",  # First speaker
-        "3Niy6MUaDzcs7Liw7dFs"   # Second speaker
+        female_voice_id,  # First speaker (female)
+        male_voice_id     # Second speaker (male)
     ]
     logo_url = "https://opencall.ai/images/logo-symbol.svg"
     
@@ -271,6 +283,10 @@ def generate_video(description, male_stability=0.65, male_style=0.35):
     
     return video_path
 
+# Get available voices
+available_voices = get_voices()
+voice_choices = [(voice["voice_id"], voice["name"]) for voice in available_voices]
+
 # Create Gradio interface
 iface = gr.Interface(
     fn=generate_video,
@@ -281,6 +297,18 @@ iface = gr.Interface(
             placeholder="Describe the conversation you want to generate...",
             info="You can be specific about the number of turns, tone, and content of the conversation"
         ),
+        gr.Dropdown(
+            choices=[choice[0] for choice in voice_choices],
+            value="cgSgspJ2msm6clMCkdW9",
+            label="Female Voice",
+            info="Select the voice for the female speaker"
+        ),
+        gr.Dropdown(
+            choices=[choice[0] for choice in voice_choices],
+            value="3Niy6MUaDzcs7Liw7dFs",
+            label="Male Voice", 
+            info="Select the voice for the male speaker"
+        ),
         gr.Slider(
             minimum=0.1,
             maximum=1.0,