Update app.py

app.py

@@ -17,32 +17,77 @@ lang = st.selectbox("Select language", lang_options)
 gender_options = ['Female', 'Male']
 gender = st.selectbox("Select gender", gender_options)
 
-# Button to trigger TTS
-if st.button("Generate Speech"):
-    # API URL
-    url = "https://asr.iitm.ac.in/tts/hindi/male/"  # Change this to your API URL
-
-    # Prepare payload
-    payload = {
-        "input": text,
-        "gender": gender.lower(),
-        "lang": lang.lower(),
-        "alpha": 1,
-        "segmentwise": "True"
-    }
-
-    # Make API request
-    response = requests.post(url, json=payload)
-
-    # Handle response
-    if response.status_code == 200:
-        # Decode audio
-        audio = response.json()['audio']
-        file_name = "tts.mp3"
-        with open(file_name, 'wb') as wav_file:
-            decode_string = base64.b64decode(audio)
-            wav_file.write(decode_string)
-        
-        st.audio(file_name, format='audio/mp3')
-    else:
-        st.error("Failed to generate speech. Please check your input and try again.")
+# Dictionary mapping language and gender to API URLs
+api_urls = {
+    ('Hindi', 'Female'): "https://asr.iitm.ac.in/tts/hindi/female",
+    ('Hindi', 'Male'): "https://asr.iitm.ac.in/tts/hindi/male",
+    ('English', 'Male'): "https://asr.iitm.ac.in/tts/english/male",
+    ('English', 'Female'): "https://asr.iitm.ac.in/tts/english/female",
+    ('Malayalam', 'Male'): "https://asr.iitm.ac.in/tts/malayalam/male",
+    ('Malayalam', 'Female'): "https://asr.iitm.ac.in/tts/malayalam/female",
+    ('Tamil', 'Male'): "https://asr.iitm.ac.in/tts/tamil/male",
+    ('Tamil', 'Female'): "https://asr.iitm.ac.in/tts/tamil/female",
+    ('Telugu', 'Male'): "https://asr.iitm.ac.in/tts/telugu/male",
+    ('Telugu', 'Female'): "https://asr.iitm.ac.in/tts/telugu/female",
+    ('Punjabi', 'Male'): "https://asr.iitm.ac.in/tts/punjabi/male",
+    ('Punjabi', 'Female'): "https://asr.iitm.ac.in/tts/punjabi/female",
+    
+    ('Urdu', 'Male'): "https://asr.iitm.ac.in/tts/urdu/male",
+    ('Urdu', 'Female'): "https://asr.iitm.ac.in/tts/urdu/female",
+    ('Marathi', 'Male'): "https://asr.iitm.ac.in/tts/marathi/male",
+    ('Marathi', 'Female'): "https://asr.iitm.ac.in/tts/marathi/female",
+    ('Gujarati', 'Male'): "https://asr.iitm.ac.in/tts/gujarati/male",
+    ('Gujarati', 'Female'): "https://asr.iitm.ac.in/tts/gujarati/female",
+
+    ('Assamese', 'Male'): "https://asr.iitm.ac.in/tts/assamese/male",
+    ('Assamese', 'Female'): "https://asr.iitm.ac.in/tts/assamese/female",
+    ('Bengali', 'Male'): "https://asr.iitm.ac.in/tts/bengali/male",
+    ('Bengali', 'Female'): "https://asr.iitm.ac.in/tts/bengali/female",
+    ('Bodo', 'Male'): "https://asr.iitm.ac.in/tts/bodo/male",
+    ('Bodo', 'Female'): "https://asr.iitm.ac.in/tts/bodo/female",
+    
+    ('Kannada', 'Male'): "https://asr.iitm.ac.in/tts/kannada/male",
+    ('Kannada', 'Female'): "https://asr.iitm.ac.in/tts/kannada/female",
+    ('Manipuri', 'Male'): "https://asr.iitm.ac.in/tts/Manipuri/male",
+    ('Manipuri', 'Female'): "https://asr.iitm.ac.in/tts/Manipuri/female",
+    ('Odia', 'Male'): "https://asr.iitm.ac.in/tts/odia/male",
+    ('Odia', 'Female'): "https://asr.iitm.ac.in/tts/odia/female",
+    
+    ('Rajasthani', 'Male'): "https://asr.iitm.ac.in/tts/rajasthani/male",
+    ('Rajasthani', 'Female'): "https://asr.iitm.ac.in/tts/rajasthani/female",
+
+    # Add more combinations as needed
+}
+
+# Get the URL based on language and gender selection
+selected_url = api_urls.get((lang, gender))
+
+if selected_url:
+    # Button to trigger TTS
+    if st.button("Generate Speech"):
+        # Prepare payload
+        payload = {
+            "input": text,
+            "gender": gender.lower(),
+            "lang": lang.lower(),
+            "alpha": 1,
+            "segmentwise": "True"
+        }
+
+        # Make API request
+        response = requests.post(selected_url, json=payload)
+
+        # Handle response
+        if response.status_code == 200:
+            # Decode audio
+            audio = response.json()['audio']
+            file_name = "tts.mp3"
+            with open(file_name, 'wb') as wav_file:
+                decode_string = base64.b64decode(audio)
+                wav_file.write(decode_string)
+            
+            st.audio(file_name, format='audio/mp3')
+        else:
+            st.error("Failed to generate speech. Please check your input and try again.")
+else:
+    st.error("Selected language and gender combination is not supported.")