Update app.py

app.py

@@ -503,7 +503,7 @@ def translate_text(model_name: str, s_language: str, t_language: str, input_text
 
 # App layout
 st.header("Text Machine Translation")
-input_text = st.text_input("Enter text to translate:")
+input_text = st.text_input("Enter text to translate:", placeholder="Enter text to translate, maximum 512 characters!")
 
 # Initialize session state if not already set
 if "sselected_language" not in st.session_state:
@@ -512,6 +512,10 @@ if "tselected_language" not in st.session_state:
     st.session_state["tselected_language"] = options[1]
 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 = ""
+if "message" not in st.session_state:
+    st.session_state.message = ""
 
 # Model selection FIRST
 model_name = st.selectbox("Select a model:", models, 
@@ -553,14 +557,13 @@ if model_name == 'Helsinki-NLP':
         tokenizer = AutoTokenizer.from_pretrained(model_name)
         model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
         pipe = pipeline("translation", model=model, tokenizer=tokenizer)
-if model_name.startswith('t5'):
-    tokenizer = T5Tokenizer.from_pretrained(model_name)
-    model = T5ForConditionalGeneration.from_pretrained(model_name).to(device)
 
 st.write("Selected language combination:", sselected_language, " - ", tselected_language, "Selected model:", model_name)
 submit_button = st.button("Translate")
-translated_textarea = st.text_area(":blue[Translation:]", placeholder="Translation area", value='')
-message_textarea = st.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='')
 
 # Handle the submit button click
 if submit_button:
@@ -617,6 +620,6 @@ if submit_button:
     # Display the translated text
     print(translated_text)
     message = f"Translated from {sselected_language} to {tselected_language} using {model_name}."
-    translated_textarea = st.text_area(":blue[Translation:]", value=translated_text)
-    message_textarea = st.text_area(":blue[Message:]", value=message)
+    translated_textarea.text_area(":green[Translation:]", value=translated_text)
+    message_textarea.text_area(":blue[Message:]", value=message)
     message = st.write(f"Translated text from {sselected_language} to {tselected_language} using {model_name}.")