app.py

import streamlit as st
from transformers import pipeline

# transformers パイプラインのインポート
fugu_translator_enja = pipeline('translation', model='staka/fugumt-en-ja')
fugu_translator_jaen = pipeline('translation', model='staka/fugumt-ja-en')
zhja_translator = pipeline("translation", model="Helsinki-NLP/opus-mt-tc-big-zh-ja")
jazh_translator = pipeline(model="larryvrh/mt5-translation-ja_zh")

# デフォルトのモデル
current_model = 'enja'

# 関数: 翻訳を行う
def translate(model, text):
    global current_model
    
    # モデルが変更された場合、変更を反映
    if model != current_model:
        current_model = model
    
    if current_model == 'enja':
        return fugu_translator_enja(text)[0]['translation_text']
    elif current_model == 'jaen':
        return fugu_translator_jaen(text)[0]['translation_text']
    elif current_model == 'zhja':
        return zhja_translator(text)[0]['translation_text']
    elif current_model == 'jazh':
        return jazh_translator(text)

# Streamlit アプリケーション
st.title("Multi-Language Translator")

# デフォルトの入力値
default_model = 'enja'
default_text = ''

# ユーザー入力の取得
model = st.selectbox("モデル", ['enja', 'jaen', 'zhja', 'jazh'], index=0, key='model')
text = st.text_area("入力テキスト", default_text)

# 翻訳ボタンが押されたときの処理
if st.button("翻訳する"):
    result = translate(model, text)
    st.write(f"翻訳結果: {result}")