Update app.py

app.py

@@ -2,8 +2,8 @@ import streamlit as st
 from transformers import MarianMTModel, MarianTokenizer
 
 # Function to perform translation
-def translate(text, src_lang, tgt_lang):
-    model_name = f'Helsinki-NLP/opus-mt-{src_lang}-{tgt_lang}'
+def translate(text, src_lang_code, tgt_lang_code):
+    model_name = f'Helsinki-NLP/opus-mt-{src_lang_code}-{tgt_lang_code}'
     tokenizer = MarianTokenizer.from_pretrained(model_name)
     model = MarianMTModel.from_pretrained(model_name)
 
@@ -14,10 +14,6 @@ def translate(text, src_lang, tgt_lang):
 
     return translated_text
 
-# Streamlit app layout
-st.title("Language Translator App")
-
-
 # Language mapping
 language_map = {
     "English": "en",
@@ -30,11 +26,15 @@ language_map = {
     "Hindi": "hi"
 }
 
+# Streamlit app layout
+st.title("Language Translator App")
+
 # Input and output language options
 languages_full = list(language_map.keys())
 
-src_lang = st.selectbox("Select input language:", languages_full, index=0)
-tgt_lang = st.selectbox("Select output language:", languages_full, index=1)
+# Language selection
+src_lang_full = st.selectbox("Select input language:", languages_full, index=0)
+tgt_lang_full = st.selectbox("Select output language:", languages_full, index=1)
 
 # Map full names to language codes
 src_lang_code = language_map[src_lang_full]
@@ -45,7 +45,7 @@ text_to_translate = st.text_area("Enter text to translate:")
 
 # Translate button
 if st.button("Translate"):
-    if src_lang == tgt_lang:
+    if src_lang_code == tgt_lang_code:
         st.warning("Input and output languages must be different.")
     elif text_to_translate.strip() == "":
         st.warning("Please enter text to translate.")