Update app.py

app.py

@@ -513,7 +513,7 @@ if "tselected_language" not in st.session_state:
 if "model_name" not in st.session_state:
     st.session_state["model_name"] = models[1]
 if "translated_text" not in st.session_state:
-    st.session_state.translated_text = ""
+    st.session_state.translation = ""
 if "message" not in st.session_state:
     st.session_state.message = ""
 
@@ -560,10 +560,12 @@ if model_name == 'Helsinki-NLP':
 
 st.write("Selected language combination:", sselected_language, " - ", tselected_language, "Selected model:", model_name)
 submit_button = st.button("Translate")
-translated_textarea = st.empty()
-message_textarea = st.empty()
-translated_textarea.text_area(":green[Translation:]", placeholder="Translation area", value='')
-message_textarea.text_area(":blue[Messages:]", placeholder="Messages area", value='')
+# translated_textarea = st.empty()
+# message_textarea = st.empty()
+# translated_textarea.text_area(":green[Translation:]", placeholder="Translation area", value='')
+# message_textarea.text_area(":blue[Messages:]", placeholder="Messages area", value='')
+st.text_area(":green[Translation:]", placeholder="Translation area", value=st.session_state.translation, key="translation_display")
+st.text_area(":blue[Messages:]", placeholder="Messages area", value=st.session_state.message, key="message_display")
 
 # Handle the submit button click
 if submit_button:
@@ -620,6 +622,8 @@ if submit_button:
     # Display the translated text
     print(translated_text)
     message = f"Translated from {sselected_language} to {tselected_language} using {model_name}."
-    translated_textarea.text_area(":green[Translation:]", value=translated_text)
-    message_textarea.text_area(":blue[Message:]", value=message)
+    # translated_textarea.text_area(":green[Translation:]", value=translated_text)
+    # message_textarea.text_area(":blue[Message:]", value=message)
+    st.session_state.translation = translated_text
+    st.session_state.message = message
     message = st.write(f"Translated text from {sselected_language} to {tselected_language} using {model_name}.")