Update app.py

app.py

@@ -20,26 +20,39 @@ st.title("Multi-Language Translator")
 
 st.sidebar.header("Translation Settings")
 
-# Input source and target languages
-src_lang = st.sidebar.text_input("Enter source language code (e.g., 'en' for English):", "en")
-tgt_lang = st.sidebar.text_input("Enter target language code (e.g., 'ur' for Urdu):", "ur")
-
-# Validate language codes
-if not src_lang.strip() or not tgt_lang.strip():
-    st.sidebar.error("Please enter valid source and target language codes.")
-else:
-    # Load the selected model
-    try:
-        st.sidebar.write(f"Model: `{src_lang}-{tgt_lang}`")
-        model, tokenizer = load_model(src_lang.strip(), tgt_lang.strip())
-
-        # Input text
-        text_to_translate = st.text_area("Enter text to translate:")
-
-        # Translate and display output
-        if text_to_translate.strip():
+# Dropdown menus for source and target languages
+language_options = {
+    "English": "en",
+    "Urdu": "ur",
+    "German": "de",
+    "French": "fr",
+    "Spanish": "es",
+    "Italian": "it",
+}
+
+src_lang = st.sidebar.selectbox("Select Source Language", list(language_options.keys()))
+tgt_lang = st.sidebar.selectbox("Select Target Language", list(language_options.keys()))
+
+# Get language codes
+src_lang_code = language_options[src_lang]
+tgt_lang_code = language_options[tgt_lang]
+
+# Input text
+text_to_translate = st.text_area("Enter text to translate:")
+
+# Translate button
+if st.button("Translate"):
+    if not text_to_translate.strip():
+        st.error("Please enter some text to translate.")
+    elif src_lang_code == tgt_lang_code:
+        st.error("Source and target languages must be different.")
+    else:
+        try:
             with st.spinner("Translating..."):
+                # Load the model
+                model, tokenizer = load_model(src_lang_code, tgt_lang_code)
+                # Translate the text
                 translated_text = translate_text(text_to_translate, model, tokenizer)
             st.markdown(f"### Translated Text:\n{translated_text}")
-    except Exception as e:
-        st.error(f"Error loading model for `{src_lang}-{tgt_lang}`: {str(e)}")
+        except Exception as e:
+            st.error(f"Error during translation: {str(e)}")