Update app.py

app.py

@@ -7,17 +7,9 @@ import tempfile
 import shutil
 import re
 
-# Language mapping for gTTS
-LANGUAGE_MAP = {
-    "English": "en",
-    "Japanese": "ja",
-    "German": "de",
-    "Hindi": "hi"
-}
-
-def translate_text(api_key, text, target_language):
+def translate_to_japanese(api_key, text):
     """
-    Translates English text to the selected target language using OpenAI's API and provides pronunciation.
+    Translates English text to Japanese using OpenAI's API and provides pronunciation.
     """
     # Validate input
     if not api_key:
@@ -31,11 +23,11 @@ def translate_text(api_key, text, target_language):
     # 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 {target_language}:\n\n{text}"}
+        {"role": "user", "content": f"Translate the following English text to Japanese:\n\n{text}"}
     ]
     
     try:
-        # Call the OpenAI API to get the translation
+        # Call the OpenAI API to get the Japanese translation
         response_translation = openai.ChatCompletion.create(
             model="gpt-4o",  # Use the correct endpoint for chat models
             messages=messages_translation,
@@ -43,13 +35,13 @@ def translate_text(api_key, text, target_language):
             temperature=0.5
         )
 
-        # Extract the translation from the response
-        translation = response_translation.choices[0].message['content'].strip()
+        # Extract the Japanese translation from the response
+        japanese_translation = response_translation.choices[0].message['content'].strip()
 
-        # Define the messages for the pronunciation request (Romaji or other phonetic systems)
+        # Define the messages for the pronunciation (Romaji) request
         messages_pronunciation = [
-            {"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}"}
+            {"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}"}
         ]
         
         # Call the OpenAI API to get the pronunciation
@@ -60,10 +52,10 @@ def translate_text(api_key, text, target_language):
             temperature=0.5
         )
 
-        # Extract the pronunciation from the response
+        # Extract the pronunciation (Romaji) from the response
         pronunciation = response_pronunciation.choices[0].message['content'].strip()
 
-        return translation, pronunciation
+        return japanese_translation, pronunciation
 
     except openai.error.OpenAIError as e:
         return f"OpenAI API error: {str(e)}", None
@@ -72,21 +64,21 @@ def translate_text(api_key, text, target_language):
 
 # Function to clean pronunciation text
 def clean_pronunciation(pronunciation_text):
-    # Remove introductory phrases like "Certainly! The pronunciation for the Arabic text..."
-    pronunciation_cleaned = re.sub(r"^(Certainly!|Sure!|The pronunciation for the .+? text.*?is[:]*\s*)", "", pronunciation_text).strip()
+    # 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()
     return pronunciation_cleaned
 
 # Function to generate audio file from text using gTTS
-def generate_audio_from_text(text, language_code):
-    tts = gTTS(text, lang=language_code)  # Dynamically use the language code
+def generate_audio_from_text(text):
+    tts = gTTS(text, lang='ja')  # 'ja' for Japanese language
     # 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("Multi-language Translator with Pronunciation")
-st.markdown("Translate English text into various languages and get its pronunciation.")
+st.title("English to Japanese Translator with Pronunciation")
+st.markdown("Translate English text into Japanese and get its pronunciation (Romaji) using OpenAI's API.")
 
 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
@@ -97,12 +89,6 @@ 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", "German", "Hindi"]
-)
-
 # Button to trigger the translation
 if st.button("Translate"):
     if api_key and english_text:
@@ -115,24 +101,23 @@ if st.button("Translate"):
             progress_text.text("Translating text...")
             progress_bar.progress(33)  # Update progress bar to 33%
 
-            # Translate text and get pronunciation
-            translated_text, pronunciation = translate_text(api_key, english_text, language_option)
+            japanese_text, pronunciation = translate_to_japanese(api_key, english_text)
 
             # Step 2: Check if translation was successful
             if pronunciation:
-                progress_text.text("Generating pronunciation...")
+                progress_text.text("Generating Romaji 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"**Original English Text:** {english_text}")
-                st.write(f"**Translated Text ({language_option}):** {translated_text}")
+                st.write(f"**English Text:** {english_text}")
+                st.write(f"**Japanese Output:** {japanese_text}")
                 st.write(f"**Pronunciation:** {cleaned_pronunciation}")
 
                 # Save the result in a text file
-                result_text = f"Original English Text: {english_text}\n\nTranslated Text ({language_option}): {translated_text}\nPronunciation: {cleaned_pronunciation}"
+                result_text = f"English Text: {english_text}\n\nJapanese Translation: {japanese_text}\nPronunciation: {cleaned_pronunciation}"
 
                 # Write to a text file
                 with open("translation_result.txt", "w") as file:
@@ -151,7 +136,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, LANGUAGE_MAP[language_option])
+                audio_file_path = generate_audio_from_text(cleaned_pronunciation)
 
                 # Provide a button to play the pronunciation audio
                 st.audio(audio_file_path, format="audio/mp3")
@@ -160,7 +145,7 @@ if st.button("Translate"):
                 st.image(translateimg2, width=150)  # Adjust the size as per preference
 
             else:
-                st.error(translated_text)  # Display error message if API call fails
+                st.error(japanese_text)  # Display error message if API call fails
 
         except Exception as e:
             st.error(f"An error occurred: {str(e)}")
@@ -168,4 +153,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.")