Update app.py

app.py

@@ -1,6 +1,5 @@
 import streamlit as st
 from transformers import MarianMTModel, MarianTokenizer
-import random
 
 # Map of language codes to full language names and flag images
 language_dict = {
@@ -51,6 +50,10 @@ st.markdown("""
     </style>
 """, unsafe_allow_html=True)
 
+# Custom Exception for Same Source and Target Language
+class SameLanguageException(Exception):
+    pass
+
 # Function to load model and tokenizer
 def load_translation_model(src_lang, tgt_lang):
     model_name = f'Helsinki-NLP/opus-mt-{src_lang}-{tgt_lang}'
@@ -60,6 +63,9 @@ def load_translation_model(src_lang, tgt_lang):
 
 # Function to translate text
 def translate_text(text, src_lang, tgt_lang):
+    if src_lang == tgt_lang:
+        raise SameLanguageException("Source and target languages must be different.")
+    
     model, tokenizer = load_translation_model(src_lang, tgt_lang)
 
     # Tokenize the input text
@@ -116,12 +122,15 @@ def run_app():
     # Button for translation
     if st.button("🔄 Translate", key="translate"):
         if text_input:
-            with st.spinner("🔄 Translating... Please wait..."):
-                # Translate the text
-                translated_text = translate_text(text_input, src_lang_code, tgt_lang_code)
-                # Display translated text directly in the success container
-                st.success(f"✅ Translation: {translated_text}")
-            st.balloons()  # Show animation when translation is complete
+            try:
+                with st.spinner("🔄 Translating... Please wait..."):
+                    # Translate the text
+                    translated_text = translate_text(text_input, src_lang_code, tgt_lang_code)
+                    # Display translated text directly in the success container
+                    st.success(f"✅ Translation: {translated_text}")
+                st.balloons()  # Show animation when translation is complete
+            except SameLanguageException as e:
+                st.error(f"❌ Error: {str(e)}")
         else:
             st.warning("⚠️ Please enter or upload some text to translate.")