Update app.py

app.py

@@ -49,33 +49,59 @@ with col1:
         
         if option_selected == 'Enter text input':
             st.session_state['warnings'].clear()
-            @st.fragment
-            def choose_language(i):
-                language = st.selectbox(f"Select language for input {i+1}:", 
-                     language_options, key=f'language_{i}', index=0)
-            
+        
+            # 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'] = {}
+        
             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 = choose_language(i)
-                    st.write(f"lang : {language}")
+                    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
                 with input2:
-                    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)
+                    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
+
 
         if option_selected == 'Upload a file(csv/txt)':