Update ins2.txt

ins2.txt

@@ -13,9 +13,9 @@ def translate_to_japanese(api_key, text):
     """
     # Validate input
     if not api_key:
-        return "Error: API key is missing."
+        return "Error: API key is missing.", None
     if not text:
-        return "Error: Input text is empty."
+        return "Error: Input text is empty.", None
 
     # Set the OpenAI API key
     openai.api_key = api_key
@@ -92,42 +92,63 @@ 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, 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:** {cleaned_pronunciation}")
-            
-            # Save the result in a text file
-            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:
-                file.write(result_text)
-
-            # Create a download button for the user to download the file
-            with open("translation_result.txt", "rb") as file:
-                st.download_button(
-                    label="Download Translation Result",
-                    data=file,
-                    file_name="translation_result.txt",
-                    mime="text/plain"
-                )
-
-            # Generate audio for 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")
-
-            translateimg2 = Image.open("v3.png")  # Ensure the file is in the correct directory
-            st.image(translateimg2, width=150)  # Adjust the size as per preference
-        else:
-            st.error(japanese_text)  # Display error message if API call fails
+        # Initialize the progress bar
+        progress_bar = st.progress(0)
+        progress_text = st.empty()  # To show the progress text
+
+        try:
+            # Step 1: Request translation
+            progress_text.text("Translating text...")
+            progress_bar.progress(33)  # Update progress bar to 33%
+
+            japanese_text, pronunciation = translate_to_japanese(api_key, english_text)
+
+            # Step 2: Check if translation was successful
+            if 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"**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"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:
+                    file.write(result_text)
+
+                # Create a download button for the user to download the file
+                with open("translation_result.txt", "rb") as file:
+                    st.download_button(
+                        label="Download Translation Result",
+                        data=file,
+                        file_name="translation_result.txt",
+                        mime="text/plain"
+                    )
+
+                # Step 3: Generate audio for pronunciation
+                progress_text.text("Generating pronunciation audio...")
+                progress_bar.progress(100)  # Update progress bar to 100%
+
+                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")
+
+                translateimg2 = Image.open("v3.png")  # Ensure the file is in the correct directory
+                st.image(translateimg2, width=150)  # Adjust the size as per preference
+
+            else:
+                st.error(japanese_text)  # Display error message if API call fails
+
+        except Exception as e:
+            st.error(f"An error occurred: {str(e)}")
     else:
         if not api_key:
             st.error("API key is missing. Please add it as a secret in Hugging Face Settings.")