Update app.py

app.py

@@ -1,14 +1,15 @@
 import streamlit as st
+import openai
 from kokoro import KPipeline
 import soundfile as sf
 import io
-import requests
 
 # Streamlit App UI Setup
 st.title("Text-to-Speech with Kokoro")
+
 # Expander section to display information in multiple languages
 with st.expander("Sample Prompt!"):
-    st.markdown(""" 
+    st.markdown("""
     - My name is Shukdev. (In English)
     - Mi nombre es Shukdev. (In Spanish)
     - Je m'appelle Choukdev. (In French)
@@ -77,22 +78,30 @@ speed = st.slider("Speed", min_value=0.5, max_value=2.0, value=1.0, step=0.1)
 # Initialize the TTS pipeline with user-selected language
 pipeline = KPipeline(lang_code=lang_code)
 
-# LibreTranslate API for translation
-LIBRETRANSLATE_URL = "https://libretranslate.com/translate"
+# Function to get the OpenAI API key from the user
+openai_api_key = st.text_input("Enter your OpenAI API Key:", type="password")
 
-# Translate text to English using LibreTranslate API
-def translate_to_english(text, lang_code):
-    response = requests.post(LIBRETRANSLATE_URL, data={
-        'q': text,
-        'source': lang_code,
-        'target': 'en'
-    })
+# Function to translate text to English using OpenAI's Chat API
+def translate_to_english(api_key, text, lang_code):
+    openai.api_key = api_key
+    try:
+        # Construct the prompt for translation
+        prompt = f"Translate the following text from {lang_code} to English: \n\n{text}"
 
-    if response.status_code == 200:
-        return response.json()['translatedText']
-    else:
-        st.error(f"Translation failed. Status Code: {response.status_code}")
-        return text  # Return original text in case of failure
+        response = openai.ChatCompletion.create(
+            model="gpt-4o",  # Using ChatGPT model for translation
+            messages=[
+                {"role": "system", "content": "You are a helpful assistant that translates text."},
+                {"role": "user", "content": prompt}
+            ]
+        )
+        
+        # Extract translated text from response
+        translated_text = response['choices'][0]['message']['content'].strip()
+        return translated_text
+    except Exception as e:
+        st.error(f"Error occurred during translation: {e}")
+        return text  # Fallback to original text in case of an error
 
 # Generate Audio function
 def generate_audio(text, lang_code, voice, speed):
@@ -107,7 +116,7 @@ def generate_audio(text, lang_code, voice, speed):
         return buffer
 
 # Generate and display the audio file
-if st.button('Generate Audio'):
+if st.button('Generate Audio') and openai_api_key:
     st.write("Generating speech for the original text...")
     audio_buffer = generate_audio(input_text, lang_code, voice, speed)
     
@@ -122,8 +131,8 @@ if st.button('Generate Audio'):
         mime="audio/wav"
     )
 
-    # Translate the input text to English using LibreTranslate
-    translated_text = translate_to_english(input_text, lang_code)
+    # Translate the input text to English using OpenAI
+    translated_text = translate_to_english(openai_api_key, input_text, lang_code)
 
     # Generate audio for the translated English text
     translated_audio_buffer = generate_audio(translated_text, 'a', voice, speed)