Update app.py

app.py

@@ -8,11 +8,11 @@ st.set_page_config(layout="wide")
 def fill_mask(sentences):
     results = {}
     warnings = []
-    for language, sentence in sentences.items():
+    for key, (language, sentence) in sentences.items():
         if "<mask>" in sentence:
             masked_sentence = sentence.replace('<mask>', unmasker.tokenizer.mask_token)
             unmasked = unmasker(masked_sentence)
-            results[language] = unmasked
+            results[key] = (language, unmasked)
         else:
             warnings.append(f"Warning: No <mask> token found in sentence: {sentence}")
     return results, warnings
@@ -31,7 +31,7 @@ if 'text_input' not in st.session_state:
 if 'warnings' not in st.session_state:
     st.session_state['warnings'] = []
 
-language_options = ['Choose language','Zulu', 'Tshivenda', 'Sepedi', 'Tswana', 'Tsonga']
+language_options = ['Choose language', 'Zulu', 'Tshivenda', 'Sepedi', 'Tswana', 'Tsonga']
 
 with col1:
     with st.container():
@@ -44,22 +44,22 @@ with col1:
             with input1:
                 language = st.selectbox(f"Select language for sentence {i+1}:", language_options, key=f'language_{i}')
             with input2:
-                disabled = True if language == "Choose language" else False
-                sentence = st.text_input(f"Enter sentence for {language} (with <mask>):", key=f'text_input_{i}', disabled=disabled)
-                if not disabled and sentence:
-                    input_sentences[language.lower()] = sentence
+                sentence = st.text_input(f"Enter sentence for {language} (with <mask>):", key=f'text_input_{i}')
+                if sentence:
+                    # Create a unique key for each sentence
+                    input_sentences[f'{language.lower()}_{i+1}'] = (language.lower(), sentence)
 
         button1, button2, _ = st.columns([2, 2, 4])
         with button1:
             if st.button("Test Example"):
-                sample_sentence = {
-                    'zulu': "Le ndoda ithi izo <mask> ukudla.",
-                    'tshivenda': "Vhana vhane vha kha ḓi bva u bebwa vha kha khombo ya u <mask> nga Listeriosis.",
-                    'tshivenda': "Rabulasi wa <mask> u khou bvelela nga u lima",
-                    'tswana': "Monna o <mask> tsamaya.",
-                    'tsonga': "N'wana wa xisati u <mask> ku tsaka."
+                sample_sentences = {
+                    'zulu_1': ('zulu', "Le ndoda ithi izo <mask> ukudla."),
+                    'tshivenda_2': ('tshivenda', "Vhana vhane vha kha ḓi bva u bebwa vha kha khombo ya u <mask> nga Listeriosis."),
+                    'tshivenda_3': ('tshivenda', "Rabulasi wa <mask> u khou bvelela nga u lima"),
+                    'tswana_4': ('tswana', "Monna o <mask> tsamaya."),
+                    'tsonga_5': ('tsonga', "N'wana wa xisati u <mask> ku tsaka.")
                 }
-                input_sentences = sample_sentence
+                input_sentences = sample_sentences
                 result, warnings = fill_mask(input_sentences)
         
         with button2:
@@ -83,8 +83,8 @@ with col1:
 with col2:
     with st.container():
         st.markdown("Output :bar_chart:")
-        if input_sentences and st.session_state.get('submit_clicked', False):
-            for language, sentence in input_sentences.items():
+        if input_sentences:
+            for key, (language, sentence) in input_sentences.items():
                 masked_sentence = sentence.replace('<mask>', unmasker.tokenizer.mask_token)
                 predictions = unmasker(masked_sentence)
 
@@ -105,8 +105,8 @@ with col2:
 
 if 'predictions' in locals():  
     if result:
-        for language, language_predictions in result.items(): 
-            original_sentence = input_sentences[language]
+        for key, (language, language_predictions) in result.items(): 
+            original_sentence = input_sentences[key][1]
             predicted_sentence = replace_mask(original_sentence, language_predictions[0]['token_str'])
             st.write(f"{language}: {predicted_sentence}\n")