Update app.py

app.py

@@ -7,29 +7,18 @@ 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")
 
+# st.session_state で session-specific state を作成
+if 'session_models' not in st.session_state:
+    st.session_state.session_models = {
+        'enja': fugu_translator_enja,
+        'jaen': fugu_translator_jaen,
+        'zhja': zhja_translator,
+        'jazh': jazh_translator
+    }
+
 # デフォルトの入力値
 default_model = 'enja'
 default_text = ''
@@ -40,5 +29,11 @@ text = st.text_area("入力テキスト", default_text)
 
 # 翻訳ボタンが押されたときの処理
 if st.button("翻訳する"):
-    result = translate(model, text)
-    st.write(f"翻訳結果: {result}")
+    result = st.session_state.session_models[model](text)[0]['translation_text']
+    
+    # Outputをcollumまたはcontainerに格納
+    output_col, _ = st.columns(2)
+    output_col.write(f"翻訳結果: {result}")
+
+    # Experimental rerun without re-executing the entire app
+    st.experimental_rerun([output_col])