Update app.py

app.py

@@ -7,9 +7,10 @@ import tempfile
 import shutil
 import re
 
-def translate_to_japanese(api_key, text):
+# Function to translate text to any language and provide pronunciation (Romaji or phonetic)
+def translate_to_language(api_key, text, language):
     """
-    Translates English text to Japanese using OpenAI's API and provides pronunciation.
+    Translates English text to the target language using OpenAI's API and provides pronunciation.
     """
     # Validate input
     if not api_key:
@@ -23,11 +24,11 @@ def translate_to_japanese(api_key, text):
     # Define the messages for the chat model
     messages_translation = [
         {"role": "system", "content": "You are a helpful translator."},
-        {"role": "user", "content": f"Translate the following English text to Japanese:\n\n{text}"}
+        {"role": "user", "content": f"Translate the following English text to {language}:\n\n{text}"}
     ]
     
     try:
-        # Call the OpenAI API to get the Japanese translation
+        # Call the OpenAI API to get the translation
         response_translation = openai.ChatCompletion.create(
             model="gpt-4o",  # Use the correct endpoint for chat models
             messages=messages_translation,
@@ -35,13 +36,13 @@ def translate_to_japanese(api_key, text):
             temperature=0.5
         )
 
-        # Extract the Japanese translation from the response
-        japanese_translation = response_translation.choices[0].message['content'].strip()
+        # Extract the translated text
+        translated_text = response_translation.choices[0].message['content'].strip()
 
-        # Define the messages for the pronunciation (Romaji) request
+        # Define the messages for the pronunciation (phonetic) request
         messages_pronunciation = [
-            {"role": "system", "content": "You are a helpful assistant who provides the Romaji (Japanese pronunciation in Latin script) of Japanese text."},
-            {"role": "user", "content": f"Provide the Romaji pronunciation for the following Japanese text:\n\n{japanese_translation}"}
+            {"role": "system", "content": f"You are a helpful assistant who provides the pronunciation in phonetic script of {language} text."},
+            {"role": "user", "content": f"Provide the pronunciation for the following {language} text:\n\n{translated_text}"}
         ]
         
         # Call the OpenAI API to get the pronunciation
@@ -52,10 +53,10 @@ def translate_to_japanese(api_key, text):
             temperature=0.5
         )
 
-        # Extract the pronunciation (Romaji) from the response
+        # Extract the pronunciation from the response
         pronunciation = response_pronunciation.choices[0].message['content'].strip()
 
-        return japanese_translation, pronunciation
+        return translated_text, pronunciation
 
     except openai.error.OpenAIError as e:
         return f"OpenAI API error: {str(e)}", None
@@ -64,21 +65,21 @@ def translate_to_japanese(api_key, text):
 
 # Function to clean pronunciation text
 def clean_pronunciation(pronunciation_text):
-    # Remove introductory phrases like "Sure! The Romaji pronunciation..."
-    pronunciation_cleaned = re.sub(r"^Sure! The Romaji pronunciation for the Japanese text.*?is[:]*", "", pronunciation_text).strip()
+    # Remove introductory phrases like "Sure! The pronunciation... is:"
+    pronunciation_cleaned = re.sub(r"^Sure! The pronunciation for the.*?text.*?is[:]*", "", pronunciation_text).strip()
     return pronunciation_cleaned
 
 # Function to generate audio file from text using gTTS
-def generate_audio_from_text(text):
-    tts = gTTS(text, lang='ja')  # 'ja' for Japanese language
+def generate_audio_from_text(text, language_code):
+    tts = gTTS(text, lang=language_code)  # Use the appropriate language code
     # Save audio to a temporary file
     temp_audio_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp3")
     tts.save(temp_audio_file.name)
     return temp_audio_file.name
 
 # Streamlit UI
-st.title("English to Japanese Translator with Pronunciation")
-st.markdown("Translate English text into Japanese and get its pronunciation (Romaji).")
+st.title("English to Multiple Language Translator with Pronunciation")
+st.markdown("Translate English text into Japanese, Spanish, Italian, and German and get their pronunciation (phonetic).")
 
 translateimg = Image.open("Untitled.png")  # Ensure the file is in the correct directory
 st.image(translateimg, use_container_width=True)  # Adjust the size as per preference
@@ -89,24 +90,36 @@ api_key = os.getenv("OPENAI_API_KEY")
 # Input field for the text
 english_text = st.text_area("Enter the English text to translate")
 
+# Language selection dropdown
+languages = ["Japanese", "Spanish", "Italian", "German"]
+selected_language = st.selectbox("Select the target language", languages)
+
 # Initialize the progress bar and progress text above the translate button
 progress_bar = st.progress(0)
 progress_text = st.empty()  # To show the progress text
 
+# Mapping of languages to their corresponding language codes for gTTS
+language_codes = {
+    "Japanese": "ja",
+    "Spanish": "es",
+    "Italian": "it",
+    "German": "de"
+}
+
 # Button to trigger the translation
 if st.button("Translate"):
     if api_key and english_text:
-
         try:
             # Step 1: Request translation
-            progress_text.text("Translating text...")
+            progress_text.text(f"Translating text to {selected_language}...")
             progress_bar.progress(33)  # Update progress bar to 33%
 
-            japanese_text, pronunciation = translate_to_japanese(api_key, english_text)
+            # Translate based on the selected language
+            translated_text, pronunciation = translate_to_language(api_key, english_text, selected_language)
 
             # Step 2: Check if translation was successful
             if pronunciation:
-                progress_text.text("Generating Romaji pronunciation...")
+                progress_text.text(f"Generating {selected_language} pronunciation...")
                 progress_bar.progress(66)  # Update progress bar to 66%
 
                 # Clean pronunciation (remove unnecessary parts)
@@ -114,11 +127,11 @@ if st.button("Translate"):
 
                 st.markdown("### Translation Result:")
                 st.write(f"**English Text:** {english_text}")
-                st.write(f"**Japanese Output:** {japanese_text}")
+                st.write(f"**{selected_language} Translation:** {translated_text}")
                 st.write(f"**Pronunciation:** {cleaned_pronunciation}")
 
                 # Save the result in a text file
-                result_text = f"English Text: {english_text}\n\nJapanese Translation: {japanese_text}\nPronunciation: {cleaned_pronunciation}"
+                result_text = f"English Text: {english_text}\n\n{selected_language} Translation: {translated_text}\nPronunciation: {cleaned_pronunciation}"
 
                 # Write to a text file
                 with open("translation_result.txt", "w") as file:
@@ -134,10 +147,11 @@ if st.button("Translate"):
                     )
 
                 # Step 3: Generate audio for pronunciation
-                progress_text.text("Generating pronunciation audio...")
+                progress_text.text(f"Generating pronunciation audio for {selected_language}...")
                 progress_bar.progress(100)  # Update progress bar to 100%
 
-                audio_file_path = generate_audio_from_text(cleaned_pronunciation)
+                # Generate audio for the cleaned pronunciation in the selected language
+                audio_file_path = generate_audio_from_text(cleaned_pronunciation, language_codes[selected_language])
 
                 # Provide a button to play the pronunciation audio
                 st.audio(audio_file_path, format="audio/mp3")
@@ -146,7 +160,7 @@ if st.button("Translate"):
                 st.image(translateimg2, width=150)  # Adjust the size as per preference
 
             else:
-                st.error(japanese_text)  # Display error message if API call fails
+                st.error(translated_text)  # Display error message if API call fails
 
         except Exception as e:
             st.error(f"An error occurred: {str(e)}")
@@ -154,4 +168,4 @@ if st.button("Translate"):
         if not api_key:
             st.error("API key is missing. Please add it as a secret in Hugging Face Settings.")
         else:
-            st.error("Please provide text to translate.")
+            st.error("Please provide text to translate.")