Update app.py

app.py

@@ -3,7 +3,7 @@ import streamlit as st
 
 def translate_to_japanese(api_key, text):
     """
-    Translates English text to Japanese using OpenAI's API.
+    Translates English text to Japanese using OpenAI's API and provides pronunciation.
     """
     # Validate input
     if not api_key:
@@ -15,32 +15,49 @@ def translate_to_japanese(api_key, text):
     openai.api_key = api_key
     
     # Define the messages for the chat model
-    messages = [
+    messages_translation = [
         {"role": "system", "content": "You are a helpful translator."},
         {"role": "user", "content": f"Translate the following English text to Japanese:\n\n{text}"}
     ]
     
     try:
-        # Call the OpenAI API to get the translation
-        response = openai.ChatCompletion.create(
+        # Call the OpenAI API to get the Japanese translation
+        response_translation = openai.ChatCompletion.create(
             model="gpt-3.5-turbo",  # Use the correct endpoint for chat models
-            messages=messages,
-            max_tokens=150,  # Adjust token limit for longer outputs
+            messages=messages_translation,
+            max_tokens=150,
             temperature=0.5
         )
 
-        # Extract the translated text from the response
-        japanese_translation = response.choices[0].message['content'].strip()
-        return japanese_translation
-    
+        # Extract the Japanese translation from the response
+        japanese_translation = response_translation.choices[0].message['content'].strip()
+
+        # Define the messages for the pronunciation (Romaji) 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}"}
+        ]
+        
+        # Call the OpenAI API to get the pronunciation
+        response_pronunciation = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo",
+            messages=messages_pronunciation,
+            max_tokens=150,
+            temperature=0.5
+        )
+
+        # 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:
-        return f"OpenAI API error: {str(e)}"
+        return f"OpenAI API error: {str(e)}", None
     except Exception as e:
-        return f"An unexpected error occurred: {str(e)}"
+        return f"An unexpected error occurred: {str(e)}", None
 
 # Streamlit UI
-st.title("English to Japanese Translator")
-st.markdown("Translate English text into Japanese using OpenAI's API.")
+st.title("English to Japanese Translator with Pronunciation")
+st.markdown("Translate English text into Japanese and get its pronunciation (Romaji) using OpenAI's API.")
 
 # Input fields for the API key and text
 api_key = st.text_input("Enter your OpenAI API key", type="password")
@@ -49,8 +66,12 @@ english_text = st.text_area("Enter the English text to translate")
 # Button to trigger the translation
 if st.button("Translate"):
     if api_key and english_text:
-        japanese_text = translate_to_japanese(api_key, english_text)
-        st.markdown("### Translation Result:")
-        st.write(japanese_text)
+        japanese_text, pronunciation = translate_to_japanese(api_key, english_text)
+        if pronunciation:
+            st.markdown("### Translation Result:")
+            st.write(f"**Japanese Output:** {japanese_text}")
+            st.write(f"**Pronunciation:** {pronunciation}")
+        else:
+            st.error(japanese_text)  # Display error message if API call fails
     else:
         st.error("Please provide both an API key and text to translate.")