Update app.py

app.py

@@ -97,7 +97,19 @@ with col1:
                 stringio = StringIO(uploaded_file.getvalue().decode("utf-8"))
                 string_data = stringio.read()
                 
-                input_sentences = string_data.split("\n")
+                sentences = string_data.split("\n")
+
+                i = 0
+                for sentence in sentences:
+                    i += 1
+                    if ":" in sentence:
+                        splitted = sentence.split(":")
+                        language = splitted[0]
+                        sentence_mask = splitted[1]
+                        input_sentences[f'{i}'] = (language.lower(), sentence)
+                        
+                    else:
+                        warnings.append(f"Warning: No ':' token found in sentence: {sentence} in line {i}")
     
                 if st.button("Submit",use_container_width=True):
                     if st.session_state['warnings']: