import streamlit as st
from transformers import pipeline

def main():
    st.title("Multilingual Translator")

    # Get user input
    input_text = st.text_area("Enter text to translate:")

    # Select source and target languages
    source_lang = st.selectbox("Select source language:", get_languages())
    target_lang = st.selectbox("Select target language:", get_languages())

    # Translate text
    if st.button("Translate"):
        if input_text:
            translation = translate_text(input_text, source_lang, target_lang)
            st.success(f"Translated text: {translation}")
        else:
            st.warning("Please enter text to translate.")

def get_languages():
    # You can customize this list based on the languages you want to support
    return ["en", "es", "fr", "de", "it", "ja", "ko", "zh-CN"]

def translate_text(text, source_lang, target_lang):
    translator = pipeline(task="translation", model=f"{source_lang}-to-{target_lang}")
    translation = translator(text, max_length=500)[0]['translation_text']
    return translation

if __name__ == "__main__":
    main()
streamlit run app.py