Update stack

app.py

@@ -27,6 +27,17 @@ def init_app():
         st.session_state["data_dict"] = config['variables']
         st.session_state["prompt_system"] = config['prompt_system']
 
+        if 'parts' in config['variables']:
+            # Flatten structure by adding part name to each field
+            st.session_state["data_dict"] = [
+                {**field, "part": part["name"]}
+                for part in config["variables"]["parts"]
+                for field in part["fields"]
+            ]
+        else:
+            # Initialize session state with single list of variables
+            st.session_state["data_dict"] = [{**field} for field in config["variables"]]
+
 
 def main():
 
@@ -63,4 +74,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()

config.yaml

@@ -7,15 +7,11 @@ prompts:
     - "Quel est le cout total avec carburant en €/h d'une moissonneuse batteuse avec une coupe de 7,5 m ?"
 
 variables:
-  - label : Mon paramètre 1
-    key : param1
-    value :
-  - label : Mon paramètre 2
-    key : param2
-    value :
-  - label : Mon paramètre 3
-    key : param3
-    value :
+  - label: "Région viticole"
+    nature: 'radio'
+    key: param1
+    options: ["Champagne", "Vin de Bordeaux", "Occitanie"]
+    value: # Valeur par défaut
 
 
 prompt_system: "
@@ -50,4 +46,4 @@ Document utilisateur : {documentContext}
 
 Voici l'historique des messages : {messages}
 La demande de l'utilisateur est : {query}
-"
+"

pages/form.py

@@ -1,14 +1,52 @@
 import streamlit as st
+from util import getYamlConfig
 
 def page():
     st.subheader("Définissez vos paramètres")
 
-    # Boucle pour créer des inputs basés sur data_dict (qui est une liste ici)
-    for param in st.session_state.data_dict:
-        # Utilisation de la clé 'label' et 'value' pour afficher et récupérer les valeurs
-        value = st.text_input(label=param['label'], value=param['value'] if param['value'] else "")
+    # Charge la configuration YAML
+    config = getYamlConfig()
+    
+    # Vérifie si la structure inclut des 'parts' ou une liste de variables
+    if 'parts' in config['variables']:
+        # Cas avec 'parts' : Trie les parts et affiche les champs par onglet
+        parts = config["variables"]["parts"]
+        parts_sorted = sorted(parts, key=lambda part: part.get('number', float('inf')))
         
-        # Mettre à jour la valeur dans le dictionnaire après la saisie utilisateur
-        param['value'] = value
+        # Création de tabs pour chaque 'part' trié
+        tabs = st.tabs([part['name'] for part in parts_sorted])
+        for part, tab in zip(parts_sorted, tabs):
+            with tab:
+                for field in part['fields']:
+                    display_field(field)
+    else:
+        # Display fields directly if no parts are defined
+        for field in st.session_state.data_dict:
+            display_field(field)
 
-page()
+
+def display_field(field):
+    """Helper function to create the correct input based on field 'nature'."""
+    if field['nature'] == 'radio':
+        value = st.radio(field['label'], field['options'], key=field['key'])
+        field['value'] = value
+    elif field['nature'] == 'selectbox':
+        value = st.selectbox(field['label'], field['options'], key=field['key'])
+        field['value'] = value
+    elif field['nature'] == 'multiselect':
+        value = st.multiselect(field['label'], field['options'], key=field['key'])
+        field['value'] = value
+    elif field['nature'] == 'date':
+        value = st.date_input(field['label'], key=field['key'])
+        field['value'] = value
+    elif field['nature'] == 'numeric':
+        value = st.number_input(field['label'], key=field['key'])
+        field['value'] = value
+    elif field['nature'] == 'text_area':
+        value = st.text_area(field['label'], value=field['value'] if 'value' in field else "", key=field['key'])
+        field['value'] = value
+    else:
+        value = st.text_input(label=field['label'], value=field['value'] if 'value' in field else "")
+        field['value'] = value
+
+page()

requirements.txt

@@ -1,4 +1,5 @@
 streamlit==1.37.0
+faiss-gpu
 streamlit_chat
 python-dotenv
 pymupdf