Update app.py

app.py

@@ -5,8 +5,8 @@ from gtts import gTTS
 from io import BytesIO
 
 # Title and Introduction
-st.title("Language Learning App")
-st.write("Improve your language skills with translation, grammar correction, and pronunciation practice!")
+st.title("Advanced Language Learning App")
+st.write("Enhance your language skills with multilingual translation, grammar correction, and pronunciation practice!")
 
 # Sidebar Options
 st.sidebar.header("Choose an Operation")
@@ -16,33 +16,55 @@ operation = st.sidebar.selectbox("Operation", options)
 # Input Text
 text = st.text_area("Enter your text here:")
 
+# Supported Languages
+languages = {
+    "English": "en", "French": "fr", "Spanish": "es", "German": "de", "Chinese": "zh", "Japanese": "ja",
+    "Arabic": "ar", "Russian": "ru", "Hindi": "hi", "Urdu": "ur", "Portuguese": "pt", "Italian": "it",
+    "Dutch": "nl", "Korean": "ko", "Swedish": "sv", "Turkish": "tr", "Polish": "pl", "Danish": "da",
+    "Finnish": "fi", "Norwegian": "no", "Greek": "el", "Hebrew": "he", "Thai": "th", "Vietnamese": "vi",
+    "Czech": "cs", "Hungarian": "hu", "Indonesian": "id", "Malay": "ms", "Filipino": "tl", "Swahili": "sw"
+}
+
 # Translation Functionality
 if operation == "Translate Text":
-    target_language = st.selectbox("Select Target Language", ["French", "Spanish", "German", "Chinese", "Japanese"])
+    source_language = st.selectbox("Select Source Language", list(languages.keys()))
+    target_language = st.selectbox("Select Target Language", list(languages.keys()))
 
     if st.button("Translate"):
-        translator = pipeline("translation_en_to_" + target_language.lower())
-        translation = translator(text, max_length=100)[0]['translation_text']
-        st.write("### Translation:")
-        st.write(translation)
+        try:
+            translator = pipeline("translation", model="Helsinki-NLP/opus-mt-{src}-{tgt}".format(
+                src=languages[source_language], tgt=languages[target_language]))
+            translation = translator(text, max_length=400)[0]['translation_text']
+            st.write("### Translation:")
+            st.write(translation)
+        except Exception as e:
+            st.error(f"Translation failed: {e}")
 
 # Grammar Correction Functionality
 elif operation == "Correct Grammar":
     if st.button("Correct Grammar"):
-        corrector = pipeline("text2text-generation", model="t5-small", task="grammar-correction")
-        corrected_text = corrector(text, max_length=100)[0]['generated_text']
-        st.write("### Corrected Text:")
-        st.write(corrected_text)
+        try:
+            corrector = pipeline("text2text-generation", model="prithivida/grammar-error-correction")
+            corrected_text = corrector(text, max_length=400)[0]['generated_text']
+            st.write("### Corrected Text:")
+            st.write(corrected_text)
+        except Exception as e:
+            st.error(f"Grammar correction failed: {e}")
 
 # Pronunciation Practice Functionality
 elif operation == "Practice Pronunciation":
+    language = st.selectbox("Select Pronunciation Language", list(languages.keys()))
+
     if st.button("Generate Audio"):
-        tts = gTTS(text=text, lang="en")
-        audio_file = BytesIO()
-        tts.write_to_fp(audio_file)
-        audio_file.seek(0)
-        st.audio(audio_file, format="audio/mp3")
-        st.write("### Listen to the pronunciation!")
+        try:
+            tts = gTTS(text=text, lang=languages[language])
+            audio_file = BytesIO()
+            tts.write_to_fp(audio_file)
+            audio_file.seek(0)
+            st.audio(audio_file, format="audio/mp3")
+            st.write("### Listen to the pronunciation!")
+        except Exception as e:
+            st.error(f"Pronunciation generation failed: {e}")
 
 # Prompt for Text Input
 else: