Update app.py

app.py

@@ -5,6 +5,7 @@ from PIL import Image
 from gtts import gTTS
 import tempfile
 import shutil
+import re
 
 def translate_to_japanese(api_key, text):
     """
@@ -53,6 +54,7 @@ def translate_to_japanese(api_key, text):
 
         # Extract the pronunciation (Romaji) from the response
         pronunciation = response_pronunciation.choices[0].message['content'].strip()
+
         return japanese_translation, pronunciation
 
     except openai.error.OpenAIError as e:
@@ -60,6 +62,12 @@ def translate_to_japanese(api_key, text):
     except Exception as e:
         return f"An unexpected error occurred: {str(e)}", None
 
+# 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()
+    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
@@ -86,13 +94,16 @@ if st.button("Translate"):
     if api_key and english_text:
         japanese_text, pronunciation = translate_to_japanese(api_key, english_text)
         if pronunciation:
+            # 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"**Pronunciation:** {pronunciation}")
+            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: {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:
@@ -108,7 +119,7 @@ if st.button("Translate"):
                 )
 
             # Generate audio for pronunciation
-            audio_file_path = generate_audio_from_text(pronunciation)
+            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")