Update app.py

app.py

@@ -7,9 +7,23 @@ import tempfile
 import shutil
 import re
 
-def translate_to_japanese(api_key, text):
+# Language mapping for gTTS
+LANGUAGE_MAP = {
+    "English": "en",
+    "Japanese": "ja",
+    "French": "fr",
+    "Spanish": "es",
+    "German": "de",
+    "Chinese": "zh",
+    "Italian": "it",
+    "Portuguese": "pt",
+    "Korean": "ko",
+    "Arabic": "ar"
+}
+
+def translate_text(api_key, text, target_language):
     """
-    Translates English text to Japanese using OpenAI's API and provides pronunciation.
+    Translates English text to the selected target language using OpenAI's API and provides pronunciation.
     """
     # Validate input
     if not api_key:
@@ -23,11 +37,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 {target_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 +49,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 translation from the response
+        translation = response_translation.choices[0].message['content'].strip()
 
-        # Define the messages for the pronunciation (Romaji) request
+        # Define the messages for the pronunciation request (Romaji or other phonetic systems)
         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": "You are a helpful assistant who provides the pronunciation of the translated text."},
+            {"role": "user", "content": f"Provide the pronunciation for the following {target_language} text:\n\n{translation}"}
         ]
         
         # Call the OpenAI API to get the pronunciation
@@ -52,10 +66,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 translation, pronunciation
 
     except openai.error.OpenAIError as e:
         return f"OpenAI API error: {str(e)}", None
@@ -65,20 +79,20 @@ 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()
+    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)  # Dynamically use the 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) using OpenAI's API.")
+st.title("Multi-language Translator with Pronunciation")
+st.markdown("Translate English text into various languages and get its pronunciation.")
 
 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,6 +103,12 @@ api_key = os.getenv("OPENAI_API_KEY")
 # Input field for the text
 english_text = st.text_area("Enter the English text to translate")
 
+# Dropdown menu for language selection
+language_option = st.selectbox(
+    "Select Target Language",
+    ["English", "Japanese", "French", "Spanish", "German", "Chinese", "Italian", "Portuguese", "Korean", "Arabic"]
+)
+
 # Button to trigger the translation
 if st.button("Translate"):
     if api_key and english_text:
@@ -101,23 +121,24 @@ if st.button("Translate"):
             progress_text.text("Translating text...")
             progress_bar.progress(33)  # Update progress bar to 33%
 
-            japanese_text, pronunciation = translate_to_japanese(api_key, english_text)
+            # Translate text and get pronunciation
+            translated_text, pronunciation = translate_text(api_key, english_text, language_option)
 
             # Step 2: Check if translation was successful
             if pronunciation:
-                progress_text.text("Generating Romaji pronunciation...")
+                progress_text.text("Generating pronunciation...")
                 progress_bar.progress(66)  # Update progress bar to 66%
 
                 # Clean pronunciation (remove unnecessary parts)
                 cleaned_pronunciation = clean_pronunciation(pronunciation)
 
                 st.markdown("### Translation Result:")
-                st.write(f"**English Text:** {english_text}")
-                st.write(f"**Japanese Output:** {japanese_text}")
+                st.write(f"**Original English Text:** {english_text}")
+                st.write(f"**Translated Text ({language_option}):** {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"Original English Text: {english_text}\n\nTranslated Text ({language_option}): {translated_text}\nPronunciation: {cleaned_pronunciation}"
 
                 # Write to a text file
                 with open("translation_result.txt", "w") as file:
@@ -136,7 +157,7 @@ if st.button("Translate"):
                 progress_text.text("Generating pronunciation audio...")
                 progress_bar.progress(100)  # Update progress bar to 100%
 
-                audio_file_path = generate_audio_from_text(cleaned_pronunciation)
+                audio_file_path = generate_audio_from_text(cleaned_pronunciation, LANGUAGE_MAP[language_option])
 
                 # Provide a button to play the pronunciation audio
                 st.audio(audio_file_path, format="audio/mp3")
@@ -145,7 +166,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)}")
@@ -153,4 +174,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.")