Fix form value to session storage

pages/form.py

@@ -1,6 +1,17 @@
 import streamlit as st
 from util import getYamlConfig
 
+def update_session_state(key,):
+    # Get new value from session state and change key for save it in params
+    new_value = st.session_state[key]
+    key = key[5:]
+    
+    for item in st.session_state.data_dict:
+        if item['key'] == key:
+            item['value'] = new_value
+            break
+
+
 def page():
     st.subheader("Définissez vos paramètres")
 
@@ -18,7 +29,9 @@ def page():
         for part, tab in zip(parts_sorted, tabs):
             with tab:
                 for field in part['fields']:
-                    display_field(field)
+                    for field_session in st.session_state.data_dict:
+                        if field['key'] == field_session['key']:
+                            display_field(field_session)
     else:
         # Display fields directly if no parts are defined
         for field in st.session_state.data_dict:
@@ -27,26 +40,67 @@ def page():
 
 def display_field(field):
     """Helper function to create the correct input based on field 'nature'."""
+    key = 'form_' + field['key']
     if field['nature'] == 'radio':
-        value = st.radio(field['label'], field['options'], key=field['key'])
-        field['value'] = value
+        st.radio(
+            field['label'],
+            field['options'],
+            index=field['options'].index(field.get('value')) if field.get('value') in field['options'] else 0,
+            key=key,
+            on_change=update_session_state,
+            args=(key,)
+        )
+
     elif field['nature'] == 'selectbox':
-        value = st.selectbox(field['label'], field['options'], key=field['key'])
-        field['value'] = value
+        st.selectbox(
+            field['label'], 
+            field['options'], 
+            index=field['options'].index(field.get('value')) if field.get('value') in field['options'] else 0, 
+            key=key, 
+            on_change=update_session_state,
+            args=(key,)
+        )
+
     elif field['nature'] == 'multiselect':
-        value = st.multiselect(field['label'], field['options'], key=field['key'])
-        field['value'] = value
+        st.multiselect(
+            field['label'], 
+            field['options'], 
+            default=[field['options'].index(value) for value in field.get('value') if value in field['options']], 
+            key=key, 
+            on_change=update_session_state,
+            args=(key,)
+        )
     elif field['nature'] == 'date':
-        value = st.date_input(field['label'], key=field['key'])
-        field['value'] = value
+        st.date_input(
+            field['label'], 
+            value=field.get('value', None),
+            key=key, 
+            on_change=update_session_state,
+            args=(key,)
+        )
+
     elif field['nature'] == 'numeric':
-        value = st.number_input(field['label'], key=field['key'])
-        field['value'] = value
+        st.number_input(field['label'], 
+            value=field.get('value', 0), 
+            key=key, 
+            on_change=update_session_state,
+            args=(key,)
+        )
+
     elif field['nature'] == 'text_area':
-        value = st.text_area(field['label'], value=field['value'] if field['value'] else "", key=field['key'])
-        field['value'] = value
+        st.text_area(field['label'], 
+            value=field.get('value', ""), 
+            key=key, 
+            on_change=update_session_state,
+            args=(key,)
+        )
+
     else:
-        value = st.text_input(label=field['label'], value=field['value'] if field['value'] else "")
-        field['value'] = value
+        st.text_input(label=field['label'], 
+            value=field.get('value', ""), 
+            key=key, 
+            on_change=update_session_state,
+            args=(key,)
+        )
 
 page()