Add application file

app.py

@@ -1,22 +1,50 @@
 import streamlit as st
-from transformers import pipeline, AutoModelForSeq2SeqLM, AutoTokenizer
+from transformers import AutoModelForSeq2SeqLM, AutoTokenizer, TranslationPipeline
 
-st.title("Translation App (Kazakh to English)")
-st.write("Powered by Hugging Face and Streamlit")
+st.title("Multilingual Translation App")
+st.write("### Powered by Hugging Face and Streamlit")
 
-# Load model and tokenizer
-st.write("Loading the model, please wait...")
-model = AutoModelForSeq2SeqLM.from_pretrained('issai/tilmash')
-tokenizer = AutoTokenizer.from_pretrained("issai/tilmash")
+@st.cache_resource
+def load_tilmash():
+    # Загрузка модели и токенайзера
+    model = AutoModelForSeq2SeqLM.from_pretrained('issai/tilmash')
+    tokenizer = AutoTokenizer.from_pretrained("issai/tilmash")
+    
+    # Инициализация пайплайна перевода
+    return TranslationPipeline(
+        model=model, 
+        tokenizer=tokenizer, 
+        max_length=1000
+    )
 
-# Initialize translation pipeline
-tilmash = pipeline("translation", model=model, tokenizer=tokenizer, src_lang="kaz_Cyrl", tgt_lang="eng_Latn", max_length=1000)
+tilmash = load_tilmash()
+
+# Список поддерживаемых языков
+languages = {
+    "Kazakh (Cyrillic)": "kaz_Cyrl",
+    "Russian": "rus_Cyrl",
+    "English": "eng_Latn",
+    "Turkish": "tur_Latn"
+}
+
+# Выбор исходного и целевого языка
+src_lang = st.selectbox("Choose the source language:", list(languages.keys()))
+tgt_lang = st.selectbox("Choose the target language:", list(languages.keys()))
+
+# Ввод текста пользователем
+user_input = st.text_area("Enter text to translate:")
 
-# Input text
-input_text = st.text_area("Enter text in Kazakh:")
 if st.button("Translate"):
-    if input_text.strip():
-        translation = tilmash(input_text)
-        st.write("Translation:", translation[0]['translation_text'])
+    if src_lang == tgt_lang:
+        st.write("Source and target languages must be different.")
+    elif user_input.strip():
+        # Выполнение перевода
+        translation = tilmash(
+            user_input, 
+            src_lang=languages[src_lang], 
+            tgt_lang=languages[tgt_lang]
+        )
+        st.write(f"Translation ({src_lang} → {tgt_lang}):")
+        st.write(translation[0]["translation_text"])
     else:
         st.write("Please enter some text to translate.")