Update app.py

app.py

@@ -49,59 +49,33 @@ with col1:
         
         if option_selected == 'Enter text input':
             st.session_state['warnings'].clear()
-        
-            # Initialize session state to preserve language and sentence inputs between reruns
-            if 'input_sentences' not in st.session_state:
-                st.session_state['input_sentences'] = {}
-            if 'languages_selected' not in st.session_state:
-                st.session_state['languages_selected'] = {}
-        
+            @st.fragment
+            def choose_language(i):
+                language = st.selectbox(f"Select language for input {i+1}:", 
+                     language_options, key=f'language_{i}', index=0)
+            
             input1, input2 = st.columns(2)
-        
             for i in range(5):
-                # Get the previously selected language and sentence, if available
-                previous_language = st.session_state['languages_selected'].get(f'language_{i}', 'Choose language')
-                previous_sentence = st.session_state['input_sentences'].get(f'text_input_{i}', '')
-        
-                # Select language in column 1
                 with input1:
-                    language = st.selectbox(
-                        f"Select language for input {i+1}:",
-                        language_options,
-                        key=f'language_{i}',
-                        index=language_options.index(previous_language) if previous_language in language_options else 0
-                    )
-                    # Store selected language in session state
-                    st.session_state['languages_selected'][f'language_{i}'] = language
-        
-                # Enter sentence in column 2
+                    language = choose_language(i)
+                    st.write(f"lang : {language}")
                 with input2:
-                    sentence = st.text_input(
-                        f"Enter sentence for input {i+1} (with <mask>):", 
-                key=f'text_input_{i}', 
-                value=previous_sentence
-            )
-            # Store input sentence in session state
-            st.session_state['input_sentences'][f'text_input_{i}'] = sentence
-
-            if sentence:
-                if language and language != 'Choose language':
-                    # Add valid language and sentence to input_sentences
-                    st.session_state['input_sentences'][f'{language.lower()}_{i+1}'] = sentence
-                else:
-                    st.session_state['warnings'].append(f"Warning: Choose the language for input {i+1}")
-
-            # Submit button
-            if st.button("Submit"):
-                if st.session_state['warnings']:
-                    # Show warnings if any
-                    for warning in st.session_state['warnings']:
-                        st.warning(warning)
-                else:
-                    # Process the sentences if no warnings
-                    result, warnings = fill_mask(st.session_state['input_sentences'])
-                    st.session_state['warnings'] = warnings
-
+                    sentence = st.text_input(f"Enter sentence for input {i+1} (with <mask>):", key=f'text_input_{i}')
+                    if sentence:
+                        if language:
+                            input_sentences[f'{language.lower()}_{i+1}'] = (language.lower(), sentence)
+                        else:
+                            warnings = []
+                            warnings.append(f"Warning: Choose the language for input {i+1}")
+                            st.session_state['warnings'] = warnings 
+
+            if st.button("Submit",use_container_width=True):
+                result, warnings = fill_mask(input_sentences)
+                st.session_state['warnings'] = warnings
+                            
+            if st.session_state['warnings']:
+                for warning in st.session_state['warnings']:
+                    st.warning(warning)
 
         if option_selected == 'Upload a file(csv/txt)':