Update app.py

app.py

@@ -7,9 +7,9 @@ import tempfile
 import shutil
 import re
 
-def translate_to_japanese(api_key, text):
+def translate_to_japanese(api_key, text, tone):
     """
-    Translates English text to Japanese using OpenAI's API and provides pronunciation.
+    Translates English text to Japanese using OpenAI's API and provides pronunciation with a specified tone/style.
     """
     # Validate input
     if not api_key:
@@ -20,10 +20,18 @@ def translate_to_japanese(api_key, text):
     # Set the OpenAI API key
     openai.api_key = api_key
     
-    # Define the messages for the chat model
+    # Define the messages for the chat model with the selected tone
+    tone_prompt = ""
+    if tone == "Formal":
+        tone_prompt = "Please translate this text into formal Japanese."
+    elif tone == "Casual":
+        tone_prompt = "Please translate this text into casual Japanese."
+    elif tone == "Professional":
+        tone_prompt = "Please translate this text into professional Japanese."
+
     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"{tone_prompt}\n\n{text}"}
     ]
     
     try:
@@ -89,6 +97,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 translation tone/style
+tone_option = st.selectbox(
+    "Select Translation Tone/Style",
+    ["Formal", "Casual", "Professional"]
+)
+
 # Button to trigger the translation
 if st.button("Translate"):
     if api_key and english_text:
@@ -101,7 +115,7 @@ 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)
+            japanese_text, pronunciation = translate_to_japanese(api_key, english_text, tone_option)
 
             # Step 2: Check if translation was successful
             if pronunciation: