Include multiple companies forms

app.py

@@ -14,6 +14,8 @@ load_dotenv()
 GROUP_NAME = os.environ.get("APP_NAME")
 LOGO = "assets/logo.png"
 
+COMPANIES = []
+
 def init_app():
 
     config = getYamlConfig()
@@ -27,19 +29,30 @@ def init_app():
         st.session_state["messages"] = []
         st.session_state["remove_undefined_value"] = True
         st.session_state["assistant"] = Rag(vectore_store=vs_manager)
-        st.session_state["data_dict"] = config['variables']
+        st.session_state["data_dict"] = {}
+        st.session_state["company_code"] = None
         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"]]
+        # Initialize data_dict for each company
+        for company in config['companies']:
+            COMPANIES.append(company)
+            company_code = company['code']
+
+            st.session_state["data_dict"][company_code] = []
+
+            if 'parts' in company['variables']:
+                # Flatten structure by adding part name to each field
+                st.session_state["data_dict"][company_code] = [
+                    {**field, "part": part["name"]}
+                    for part in company["variables"]["parts"]
+                    for field in part["fields"]
+                ]
+
+            else:
+                # Initialize session state with single list of variables
+                st.session_state["data_dict"][company_code] = [
+                    {**field} for field in company["variables"]
+                ]
 
 
 def main():
@@ -66,8 +79,7 @@ def main():
     saved_documents = st.Page("pages/persistent_documents.py", title="Communs", icon="🗃️")
     documents = st.Page("pages/documents.py", title="Vos documents", icon="📂")
     prompt_system = st.Page("pages/prompt_system.py", title="Prompt système", icon="🖊️")
-    form = st.Page("pages/form.py", title="Paramètres", icon="📋")
-    chatbot = st.Page("pages/chatbot.py", title="Chatbot", icon="🤖", default=True)
+    # chatbot = st.Page("pages/chatbot.py", title="Chatbot", icon="🤖", default=True)
 
     pg = st.navigation(
         {
@@ -77,11 +89,21 @@ def main():
             ],
             "Configurations": [
                 prompt_system,
-                form,
             ],
-            "Dialogue": [
-                chatbot
+            "Auto France Parts": [
+                st.Page("pages/afp.py", title="Paramètres", icon="📋"),
+                st.Page("pages/chatbot_afp.py", title="Dialogue", icon="🤖", default=True),
+            ],
+            "La Pyramide des Cettons": [
+                st.Page("pages/cettons.py", title="Paramètres", icon="📋"),
+                st.Page("pages/chatbot_cettons.py", title="Dialogue", icon="🤖"),
+
             ],
+            "Steel France": [
+                st.Page("pages/steel.py", title="Paramètres", icon="📋"),
+                st.Page("pages/chatbot_steel.py", title="Dialogue", icon="🤖"),
+
+            ]
         }
     )
 

config.yaml

@@ -48,113 +48,269 @@ prompts:
 # Chaque 'part' représente une section contenant un groupe de champs (par exemple : Panier, Livraison).
 # La clé 'number' permet d’ordonner les sections.
 # Pour activer cette version, commentez la Version 1 ci-dessus et décommentez cette section.
+companies:
+  - name: "Auto France Parts"
+    code: "afp_"
+    variables:
+      parts:
+        - name: "Exploitant"
+          number: 1
+          fields:
+            - label: "Nom de l'exploitation"
+              nature: ''
+              key: exploitation_name
+              value: "Auto France Parts"
+
+            - label: "Type d'activité"
+              nature: ''
+              key: type_activite
+              value: "Commerce de gros d'équipements automobiles"
+              
+            - label: "Date de création de l'entreprise"
+              nature: 'date'
+              key: date_creation
+              value: 
+              
+            - label: "Chiffre d'affaires annuel (en €)"
+              nature: 'numeric'
+              key: ca_annuel
+              value: 72000000
+              
+            - label: "EBE (Excédent Brut d'Exploitation, en €)"
+              nature: 'numeric'
+              key: ebe
+              value: 0
+
+            - label: "Total Bilan"
+              nature: 'numeric'
+              key: total_bilan
+              value: 0
+              
+            - label: "Nombre de salarié"
+              nature: 'numeric'
+              key: nb_salaries
+              value: 20
+              
+        - name: "Type de projet"
+          number: 2
+          fields:
+            - label: "Description du projet ou événement important pour l’entreprise"
+              nature: 'text_area'
+              key: projet_description
+              value: ""
+              
+            - label: "Catégorie du projet"
+              nature: 'selectbox'
+              key: projet_categorie
+              options: ["Toutes", "Innovation", "Durabilité", "Développement", "Formation"]
+              value: "Toutes"
+
+            - label: "Budget Minimum"
+              nature: 'numeric'
+              key: budget_minimum
+              value: 1000
+
+            - label: "Budget Maximum"
+              nature: 'numeric'
+              key: budget_maximum
+              value: 6000000
+              
+        - name: "Critères de Subvention"
+          number: 3
+          fields:
+            - label: "Périmètre géographique de recherche"
+              nature: 'multiselect'
+              key: recherche_geo
+              options: [ "Tous", "Départemental", "Régional", "National", "Européen" ]
+              value: "Tous"
+
+            - label: "Type de subvention souhaitée"
+              nature: 'multiselect'
+              key: subvention_type
+              options: [ "Tous", "Avance − Prêts − Garanties", "Subvention", "Prise en charge des coûts et allègement des charges", "Autres" ]
+              value: "Tous"
+
+            - label: "Thématique de l'aide"
+              nature: 'multiselect'
+              key: subvention_thematic
+              options: [ "Tous", "Crise énergétique", "France 2030", "Plan résilience" ]
+              value: "Tous"
+
+  - name: "Steel France"
+    code: "steel_"
+    variables:
+      parts:
+        - name: "Exploitant"
+          number: 1
+          fields:
+            - label: "Nom de l'exploitation"
+              nature: ''
+              key: exploitation_name
+              value: "Steel France"
+
+            - label: "Type d'activité"
+              nature: ''
+              key: type_activite
+              value: "Commerce de gros d'équipements automobiles"
+              
+            - label: "Date de création de l'entreprise"
+              nature: 'date'
+              key: date_creation
+              value: 
+              
+            - label: "Chiffre d'affaires annuel (en €)"
+              nature: 'numeric'
+              key: ca_annuel
+              value: 
+              
+            - label: "EBE (Excédent Brut d'Exploitation, en €)"
+              nature: 'numeric'
+              key: ebe
+              value: 0
+
+            - label: "Total Bilan"
+              nature: 'numeric'
+              key: total_bilan
+              value: 0
+              
+            - label: "Nombre de salarié"
+              nature: 'numeric'
+              key: nb_salaries
+              value: 0
+              
+        - name: "Type de projet"
+          number: 2
+          fields:
+            - label: "Description du projet ou événement important pour l’entreprise"
+              nature: 'text_area'
+              key: projet_description
+              value: ""
+              
+            - label: "Catégorie du projet"
+              nature: 'selectbox'
+              key: projet_categorie
+              options: ["Toutes", "Innovation", "Durabilité", "Développement", "Formation"]
+              value: "Toutes"
+
+            - label: "Budget Minimum"
+              nature: 'numeric'
+              key: budget_minimum
+              value: 1000
+
+            - label: "Budget Maximum"
+              nature: 'numeric'
+              key: budget_maximum
+              value: 6000000
+              
+        - name: "Critères de Subvention"
+          number: 3
+          fields:
+            - label: "Périmètre géographique de recherche"
+              nature: 'multiselect'
+              key: recherche_geo
+              options: [ "Tous", "Départemental", "Régional", "National", "Européen" ]
+              value: "Tous"
+
+            - label: "Type de subvention souhaitée"
+              nature: 'multiselect'
+              key: subvention_type
+              options: [ "Tous", "Avance − Prêts − Garanties", "Subvention", "Prise en charge des coûts et allègement des charges", "Autres" ]
+              value: "Tous"
+
+            - label: "Thématique de l'aide"
+              nature: 'multiselect'
+              key: subvention_thematic
+              options: [ "Tous", "Crise énergétique", "France 2030", "Plan résilience" ]
+              value: "Tous"
+
+  - name: "La pyramide des Cettons"
+    code: "cettons_"
+    variables:
+      parts:
+        - name: "Exploitant"
+          number: 1
+          fields:
+            - label: "Nom de l'exploitation"
+              nature: ''
+              key: exploitation_name
+              value: "La pyramide des Cettons"
+
+            - label: "Type d'activité"
+              nature: ''
+              key: type_activite
+              value: "Location de terrains et d'autres biens immobiliers"
+              
+            - label: "Date de création de l'entreprise"
+              nature: 'date'
+              key: date_creation
+              value: 
+              
+            - label: "Chiffre d'affaires annuel (en €)"
+              nature: 'numeric'
+              key: ca_annuel
+              value: 0
+              
+            - label: "EBE (Excédent Brut d'Exploitation, en €)"
+              nature: 'numeric'
+              key: ebe
+              value: 0
+
+            - label: "Total Bilan"
+              nature: 'numeric'
+              key: total_bilan
+              value: 0
+              
+            - label: "Nombre de salarié"
+              nature: 'numeric'
+              key: nb_salaries
+              value: 0
+              
+        - name: "Type de projet"
+          number: 2
+          fields:
+            - label: "Description du projet ou événement important pour l’entreprise"
+              nature: 'text_area'
+              key: projet_description
+              value: ""
+              
+            - label: "Catégorie du projet"
+              nature: 'selectbox'
+              key: projet_categorie
+              options: ["Toutes", "Innovation", "Durabilité", "Développement", "Formation"]
+              value: "Toutes"
+
+            - label: "Budget Minimum"
+              nature: 'numeric'
+              key: budget_minimum
+              value: 1000
+
+            - label: "Budget Maximum"
+              nature: 'numeric'
+              key: budget_maximum
+              value: 6000000
+              
+        - name: "Critères de Subvention"
+          number: 3
+          fields:
+            - label: "Périmètre géographique de recherche"
+              nature: 'multiselect'
+              key: recherche_geo
+              options: [ "Tous", "Départemental", "Régional", "National", "Européen" ]
+              value: "Tous"
+
+            - label: "Type de subvention souhaitée"
+              nature: 'multiselect'
+              key: subvention_type
+              options: [ "Tous", "Avance − Prêts − Garanties", "Subvention", "Prise en charge des coûts et allègement des charges", "Autres" ]
+              value: "Tous"
+
+            - label: "Thématique de l'aide"
+              nature: 'multiselect'
+              key: subvention_thematic
+              options: [ "Tous", "Crise énergétique", "France 2030", "Plan résilience" ]
+              value: "Tous"
+
 
-variables:
-  parts:
-    - name: "Exploitant"
-      number: 1
-      fields:
-        - label: "Nom de l'exploitation"
-          nature: ''
-          key: exploitation_name
-          value: "Auto France Parts"
-
-        # - label: "Localisation"
-        #   nature: 'selectbox'
-        #   key: localisation
-        #   options: ["Nouvelle-Aquitaine"]
-        #   value: "Nouvelle-Aquitaine"
-
-        - label: "Type d'activité"
-          nature: ''
-          key: type_activite
-          value: "Commerce de gros d'équipements automobiles"
-
-        # - label: "Spécificités et/ou Certifications"
-        #   nature: 'multiselect'
-        #   key: specificite_certifications
-        #   options: ["Bio Oui Partielle", "Bio Oui Totale", "HVE (Haute Valeur Environnementale)", "TCS (Techniques Culturales Simplifiées)"]
-        #   value: 
-          
-        - label: "Date de création de l'entreprise"
-          nature: 'date'
-          key: date_creation
-          value: 
-          
-        - label: "Chiffre d'affaires annuel (en €)"
-          nature: 'numeric'
-          key: ca_annuel
-          value: 72000000
-          
-        - label: "EBE (Excédent Brut d'Exploitation, en €)"
-          nature: 'numeric'
-          key: ebe
-          value: 0
-
-        - label: "Total Bilan"
-          nature: 'numeric'
-          key: total_bilan
-          value: 0
-          
-        - label: "Nombre de salarié"
-          nature: 'numeric'
-          key: nb_salaries
-          value: 20
-          
-    - name: "Type de projet"
-      number: 2
-      fields:
-        - label: "Description du projet ou événement important pour l’entreprise"
-          nature: 'text_area'
-          key: projet_description
-          value: ""
-          
-        - label: "Catégorie du projet"
-          nature: 'selectbox'
-          key: projet_categorie
-          options: ["Toutes", "Innovation", "Durabilité", "Développement", "Formation"]
-          value: "Toutes"
-
-        - label: "Budget Minimum"
-          nature: 'numeric'
-          key: budget_minimum
-          value: 1000
-
-        - label: "Budget Maximum"
-          nature: 'numeric'
-          key: budget_maximum
-          value: 6000000
-
-        # - label: "Budget total estimé"
-        #   nature: 'slider'
-        #   key: projet_budget
-        #   value: [] # Valeur par défaut
-        #   min: 0
-        #   max: 50000
-        #   step: 500
-
-          
-    - name: "Critères de Subvention"
-      number: 3
-      fields:
-        - label: "Périmètre géographique de recherche"
-          nature: 'multiselect'
-          key: recherche_geo
-          options: [ "Tous", "Départemental", "Régional", "National", "Européen" ]
-          value: "Tous"
-
-        - label: "Type de subvention souhaitée"
-          nature: 'multiselect'
-          key: subvention_type
-          options: [ "Tous", "Avance − Prêts − Garanties", "Subvention", "Prise en charge des coûts et allègement des charges", "Autres" ]
-          value: "Tous"
-
-        - label: "Thématique de l'aide"
-          nature: 'multiselect'
-          key: subvention_thematic
-          options: [ "Tous", "Crise énergétique", "France 2030", "Plan résilience" ]
-          value: "Tous"
-          
 
 prompt_system: "
 Informations générales de l'exploitation agricole :

pages/__init__.py

@@ -0,0 +1 @@
+# ...no code needed...

pages/afp.py

@@ -0,0 +1,131 @@
+import datetime
+import streamlit as st
+from util import getYamlConfig
+
+company_code = "afp_"
+
+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"][st.session_state["company_code"]]:
+        if item['key'] == key:
+            item['value'] = new_value
+            break
+
+
+def page():
+    st.subheader("Définissez vos paramètres")
+    st.text("L'ensemble des critères sont facultatifs.")
+
+    st.session_state["company_code"] = company_code
+    company_config = None
+
+    # Charge la configuration YAML
+    config = getYamlConfig()
+
+    for company in config['companies']:
+        if company['code'] == st.session_state["company_code"]:
+            company_config = company
+
+
+    # Vérifie si la structure inclut des 'parts' ou une liste de variables
+    if 'parts' in company_config['variables']:
+        # Cas avec 'parts' : Trie les parts et affiche les champs par onglet
+        parts = company_config["variables"]["parts"]
+        parts_sorted = sorted(parts, key=lambda part: part.get('number', float('inf')))
+        
+        # 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']:
+                    for field_session in st.session_state["data_dict"][st.session_state["company_code"]]:
+                        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"][st.session_state["company_code"]]:
+            display_field(field)
+
+
+def display_field(field):
+    """Helper function to create the correct input based on field 'nature'."""
+    key = 'form_' + field['key']
+    if field['nature'] == 'radio':
+        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':
+        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':
+        default_values = field.get('value', [])
+
+        st.multiselect(
+            field['label'], 
+            field['options'], 
+            default=default_values if default_values else None, 
+            key=key, 
+            on_change=update_session_state,
+            args=(key,)
+        )
+    elif field['nature'] == 'date':
+        st.date_input(
+            field['label'], 
+            value=datetime.date.fromisoformat("2011-09-20"),
+            key=key, 
+            on_change=update_session_state,
+            args=(key,)
+        )
+
+    elif field['nature'] == 'numeric':
+        st.number_input(field['label'], 
+            value=field.get('value', 0), 
+            key=key, 
+            on_change=update_session_state,
+            args=(key,)
+        )
+
+    elif field['nature'] == 'text_area':
+        st.text_area(field['label'], 
+            value=field.get('value', ""), 
+            key=key, 
+            on_change=update_session_state,
+            args=(key,)
+        )
+
+    elif field['nature'] == 'slider':
+        st.slider(field['label'], 
+            value=field.get('value', [0, 1000]), 
+            min_value=field.get('min', 0),        # Valeur minimale du slider
+            max_value=field.get('max', 5000),        # Valeur maximale du slider
+            step=field.get('step', 500),
+            key=key, 
+            on_change=update_session_state,
+            args=(key,)
+        )
+
+    else:
+        st.text_input(label=field['label'], 
+            value=field.get('value', ""), 
+            key=key, 
+            on_change=update_session_state,
+            args=(key,)
+        )
+
+page()

pages/cettons.py

@@ -0,0 +1,131 @@
+import datetime
+import streamlit as st
+from util import getYamlConfig
+
+company_code = "cettons_"
+
+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"][st.session_state["company_code"]]:
+        if item['key'] == key:
+            item['value'] = new_value
+            break
+
+
+def page():
+    st.subheader("Définissez vos paramètres")
+    st.text("L'ensemble des critères sont facultatifs.")
+
+    st.session_state["company_code"] = company_code
+    company_config = None
+
+    # Charge la configuration YAML
+    config = getYamlConfig()
+
+    for company in config['companies']:
+        if company['code'] == st.session_state["company_code"]:
+            company_config = company
+
+
+    # Vérifie si la structure inclut des 'parts' ou une liste de variables
+    if 'parts' in company_config['variables']:
+        # Cas avec 'parts' : Trie les parts et affiche les champs par onglet
+        parts = company_config["variables"]["parts"]
+        parts_sorted = sorted(parts, key=lambda part: part.get('number', float('inf')))
+        
+        # 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']:
+                    for field_session in st.session_state["data_dict"][st.session_state["company_code"]]:
+                        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"][st.session_state["company_code"]]:
+            display_field(field)
+
+
+def display_field(field):
+    """Helper function to create the correct input based on field 'nature'."""
+    key = 'form_' + field['key']
+    if field['nature'] == 'radio':
+        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':
+        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':
+        default_values = field.get('value', [])
+
+        st.multiselect(
+            field['label'], 
+            field['options'], 
+            default=default_values if default_values else None, 
+            key=key, 
+            on_change=update_session_state,
+            args=(key,)
+        )
+    elif field['nature'] == 'date':
+        st.date_input(
+            field['label'], 
+            value=datetime.date.fromisoformat("2021-03-29"),
+            key=key, 
+            on_change=update_session_state,
+            args=(key,)
+        )
+
+    elif field['nature'] == 'numeric':
+        st.number_input(field['label'], 
+            value=field.get('value', 0), 
+            key=key, 
+            on_change=update_session_state,
+            args=(key,)
+        )
+
+    elif field['nature'] == 'text_area':
+        st.text_area(field['label'], 
+            value=field.get('value', ""), 
+            key=key, 
+            on_change=update_session_state,
+            args=(key,)
+        )
+
+    elif field['nature'] == 'slider':
+        st.slider(field['label'], 
+            value=field.get('value', [0, 1000]), 
+            min_value=field.get('min', 0),        # Valeur minimale du slider
+            max_value=field.get('max', 5000),        # Valeur maximale du slider
+            step=field.get('step', 500),
+            key=key, 
+            on_change=update_session_state,
+            args=(key,)
+        )
+
+    else:
+        st.text_input(label=field['label'], 
+            value=field.get('value', ""), 
+            key=key, 
+            on_change=update_session_state,
+            args=(key,)
+        )
+
+page()

pages/chatbot_afp.py

@@ -0,0 +1,147 @@
+import streamlit as st
+from langchain_core.messages import AIMessage, HumanMessage, SystemMessage
+from langchain.prompts import PromptTemplate
+from model import selector
+from util import getYamlConfig
+from st_copy_to_clipboard import st_copy_to_clipboard
+from subscription.index import is_usage_limit_reached, daily_limit_enabled
+
+def display_messages():
+
+    for i, message in enumerate(st.session_state.chat_history):
+        if isinstance(message, AIMessage):
+            with st.chat_message("AI"):
+                # Display the model from the kwargs
+                model = message.kwargs.get("model", "Unknown Model")  # Get the model, default to "Unknown Model"
+                st.write(f"**Model :** {model}")
+                st.markdown(message.content)
+                st_copy_to_clipboard(message.content,key=f"message_{i}")
+                # show_retrieved_documents(st.session_state.chat_history[i-1].content)
+        
+        elif isinstance(message, HumanMessage):
+            with st.chat_message("Moi"):
+                st.write(message.content)
+
+        elif isinstance(message, SystemMessage):
+            with st.chat_message("System"):
+                st.write(message.content)
+
+def show_retrieved_documents(query: str = ''):
+    if query == '':
+        return
+    
+    # Créer l'expander pour les documents trouvés
+    expander = st.expander("Documents trouvés")
+    
+    # Boucler à travers les documents récupérés
+    for item in st.session_state.get("retrived_documents", []):
+        if 'query' in item:
+            if item["query"] == query:
+                for doc in item.get("documents", []):
+                    expander.write(doc["metadata"]["source"])
+
+
+def launchQuery(query: str = None):
+
+    # Initialize the assistant's response
+    full_response = st.write_stream(
+        st.session_state["assistant"].ask(
+            query,
+            # prompt_system=st.session_state.prompt_system,
+            messages=st.session_state["chat_history"] if "chat_history" in st.session_state else [],
+            variables=st.session_state["data_dict"]
+        ))
+
+    # Temporary placeholder AI message in chat history
+    st.session_state["chat_history"].append(AIMessage(content=full_response, kwargs={"model": st.session_state["assistant"].getReadableModel()}))
+    st.rerun()
+
+
+def show_prompts():
+    yaml_data = getYamlConfig()["prompts"]
+    
+    expander = st.expander("Prompts pré-définis")
+    
+    for categroy in yaml_data:
+        expander.write(categroy.capitalize())
+
+        for item in yaml_data[categroy]:
+            if expander.button(item, key=f"button_{item}"):
+                launchQuery(item)
+
+def remplir_texte(texte: str, variables: dict, remove_line_if_unspecified: bool = False) -> str:
+    # Convertir les valeurs en chaînes de caractères pour éviter les erreurs avec format()
+    variables_str = {
+        key: (', '.join(value) if isinstance(value, list) and len(value) else value if value else 'Non spécifié')
+        for key, value in variables.items()
+    }
+
+    # Remplacer les variables dynamiques dans le texte
+    try:
+        texte_rempli = texte.format(**variables_str)
+    except KeyError as e:
+        raise ValueError(f"Clé manquante dans le dictionnaire : {e}")
+    
+    # Supprimer les lignes contenant "Non spécifié" si l'option est activée
+    if remove_line_if_unspecified:
+        lignes = texte_rempli.split('\n')
+        lignes = [ligne for ligne in lignes if 'Non spécifié' not in ligne]
+        texte_rempli = '\n'.join(lignes)
+    
+    return texte_rempli
+
+def page():
+    st.subheader("Auto France Parts")
+
+    company_code = "afp_"
+
+    if "chat_history" not in st.session_state or len(st.session_state["chat_history"]) <= 2:
+        # print("Initialisation de l'historique")
+        if st.session_state["data_dict"][company_code] is not None:
+            # Convertir la liste en dictionnaire avec 'key' comme clé et 'value' comme valeur
+            vars = {item['key']: item['value'] for item in st.session_state["data_dict"][company_code] if 'key' in item and 'value' in item}
+
+        system_template = st.session_state.prompt_system
+        full = remplir_texte(system_template, vars, st.session_state["remove_undefined_value"])
+
+        st.session_state["chat_history"] = [
+            SystemMessage(content=full),
+        ]
+
+    if "assistant" not in st.session_state:
+        st.text("Assistant non initialisé")
+
+    st.markdown("<style>iframe{height:50px;}</style>", unsafe_allow_html=True)
+
+    # Collpase for default prompts
+    show_prompts()
+
+    # Models selector
+    selector.ModelSelector()
+
+    if(len(st.session_state["chat_history"])):
+        if st.button("Effacer l'historique"):
+            st.session_state["chat_history"] = []
+
+    # Displaying messages
+    display_messages()
+
+    message = ""
+    reached = False
+
+    if daily_limit_enabled():
+        reached = is_usage_limit_reached()
+        if reached:
+            message = "Vous avez atteint la limite quotidienne de messages."
+
+    user_query = st.chat_input(message, disabled=reached)
+
+    if user_query is not None and user_query != "":
+
+        st.session_state["chat_history"].append(HumanMessage(content=user_query))
+        
+        # Stream and display response
+        launchQuery(user_query)
+
+# DOnnes moi la liste des 3 meilleures subventions. Uniquement le nom des subventions
+page()

pages/chatbot_cettons.py

@@ -0,0 +1,147 @@
+import streamlit as st
+from langchain_core.messages import AIMessage, HumanMessage, SystemMessage
+from langchain.prompts import PromptTemplate
+from model import selector
+from util import getYamlConfig
+from st_copy_to_clipboard import st_copy_to_clipboard
+from subscription.index import is_usage_limit_reached, daily_limit_enabled
+
+def display_messages():
+
+    for i, message in enumerate(st.session_state.chat_history):
+        if isinstance(message, AIMessage):
+            with st.chat_message("AI"):
+                # Display the model from the kwargs
+                model = message.kwargs.get("model", "Unknown Model")  # Get the model, default to "Unknown Model"
+                st.write(f"**Model :** {model}")
+                st.markdown(message.content)
+                st_copy_to_clipboard(message.content,key=f"message_{i}")
+                # show_retrieved_documents(st.session_state.chat_history[i-1].content)
+        
+        elif isinstance(message, HumanMessage):
+            with st.chat_message("Moi"):
+                st.write(message.content)
+
+        elif isinstance(message, SystemMessage):
+            with st.chat_message("System"):
+                st.write(message.content)
+
+def show_retrieved_documents(query: str = ''):
+    if query == '':
+        return
+    
+    # Créer l'expander pour les documents trouvés
+    expander = st.expander("Documents trouvés")
+    
+    # Boucler à travers les documents récupérés
+    for item in st.session_state.get("retrived_documents", []):
+        if 'query' in item:
+            if item["query"] == query:
+                for doc in item.get("documents", []):
+                    expander.write(doc["metadata"]["source"])
+
+
+def launchQuery(query: str = None):
+
+    # Initialize the assistant's response
+    full_response = st.write_stream(
+        st.session_state["assistant"].ask(
+            query,
+            # prompt_system=st.session_state.prompt_system,
+            messages=st.session_state["chat_history"] if "chat_history" in st.session_state else [],
+            variables=st.session_state["data_dict"]
+        ))
+
+    # Temporary placeholder AI message in chat history
+    st.session_state["chat_history"].append(AIMessage(content=full_response, kwargs={"model": st.session_state["assistant"].getReadableModel()}))
+    st.rerun()
+
+
+def show_prompts():
+    yaml_data = getYamlConfig()["prompts"]
+    
+    expander = st.expander("Prompts pré-définis")
+    
+    for categroy in yaml_data:
+        expander.write(categroy.capitalize())
+
+        for item in yaml_data[categroy]:
+            if expander.button(item, key=f"button_{item}"):
+                launchQuery(item)
+
+def remplir_texte(texte: str, variables: dict, remove_line_if_unspecified: bool = False) -> str:
+    # Convertir les valeurs en chaînes de caractères pour éviter les erreurs avec format()
+    variables_str = {
+        key: (', '.join(value) if isinstance(value, list) and len(value) else value if value else 'Non spécifié')
+        for key, value in variables.items()
+    }
+
+    # Remplacer les variables dynamiques dans le texte
+    try:
+        texte_rempli = texte.format(**variables_str)
+    except KeyError as e:
+        raise ValueError(f"Clé manquante dans le dictionnaire : {e}")
+    
+    # Supprimer les lignes contenant "Non spécifié" si l'option est activée
+    if remove_line_if_unspecified:
+        lignes = texte_rempli.split('\n')
+        lignes = [ligne for ligne in lignes if 'Non spécifié' not in ligne]
+        texte_rempli = '\n'.join(lignes)
+    
+    return texte_rempli
+
+def page():
+    st.subheader("La Pyramide des Cettons")
+
+    company_code = "cettons_"
+
+    if "chat_history" not in st.session_state or len(st.session_state["chat_history"]) <= 2:
+        # print("Initialisation de l'historique")
+        if st.session_state["data_dict"][company_code] is not None:
+            # Convertir la liste en dictionnaire avec 'key' comme clé et 'value' comme valeur
+            vars = {item['key']: item['value'] for item in st.session_state["data_dict"][company_code] if 'key' in item and 'value' in item}
+
+        system_template = st.session_state.prompt_system
+        full = remplir_texte(system_template, vars, st.session_state["remove_undefined_value"])
+
+        st.session_state["chat_history"] = [
+            SystemMessage(content=full),
+        ]
+
+    if "assistant" not in st.session_state:
+        st.text("Assistant non initialisé")
+
+    st.markdown("<style>iframe{height:50px;}</style>", unsafe_allow_html=True)
+
+    # Collpase for default prompts
+    show_prompts()
+
+    # Models selector
+    selector.ModelSelector()
+
+    if(len(st.session_state["chat_history"])):
+        if st.button("Effacer l'historique"):
+            st.session_state["chat_history"] = []
+
+    # Displaying messages
+    display_messages()
+
+    message = ""
+    reached = False
+
+    if daily_limit_enabled():
+        reached = is_usage_limit_reached()
+        if reached:
+            message = "Vous avez atteint la limite quotidienne de messages."
+
+    user_query = st.chat_input(message, disabled=reached)
+
+    if user_query is not None and user_query != "":
+
+        st.session_state["chat_history"].append(HumanMessage(content=user_query))
+        
+        # Stream and display response
+        launchQuery(user_query)
+
+# DOnnes moi la liste des 3 meilleures subventions. Uniquement le nom des subventions
+page()

pages/{chatbot.py → chatbot_steel.py}

@@ -91,16 +91,15 @@ def remplir_texte(texte: str, variables: dict, remove_line_if_unspecified: bool
     return texte_rempli
 
 def page():
-    st.subheader("Posez vos questions")
+    st.subheader("Steel France")
 
-    if "assistant" not in st.session_state:
-        st.text("Assistant non initialisé")
+    company_code = "steel_"
 
-    if "chat_history" not in st.session_state or len(st.session_state["chat_history"]) < 2:
-
-        if st.session_state["data_dict"] is not None:
+    if "chat_history" not in st.session_state or len(st.session_state["chat_history"]) <= 2:
+        # print("Initialisation de l'historique")
+        if st.session_state["data_dict"][company_code] is not None:
             # Convertir la liste en dictionnaire avec 'key' comme clé et 'value' comme valeur
-            vars = {item['key']: item['value'] for item in st.session_state["data_dict"] if 'key' in item and 'value' in item}
+            vars = {item['key']: item['value'] for item in st.session_state["data_dict"][company_code] if 'key' in item and 'value' in item}
 
         system_template = st.session_state.prompt_system
         full = remplir_texte(system_template, vars, st.session_state["remove_undefined_value"])
@@ -109,6 +108,9 @@ def page():
             SystemMessage(content=full),
         ]
 
+    if "assistant" not in st.session_state:
+        st.text("Assistant non initialisé")
+
     st.markdown("<style>iframe{height:50px;}</style>", unsafe_allow_html=True)
 
     # Collpase for default prompts
@@ -141,5 +143,5 @@ def page():
         # Stream and display response
         launchQuery(user_query)
 
-
+# DOnnes moi la liste des 3 meilleures subventions. Uniquement le nom des subventions
 page()

pages/form.py

@@ -7,7 +7,7 @@ def update_session_state(key,):
     new_value = st.session_state[key]
     key = key[5:]
     
-    for item in st.session_state.data_dict:
+    for item in st.session_state["data_dict"][st.session_state["company_code"]]:
         if item['key'] == key:
             item['value'] = new_value
             break
@@ -15,16 +15,26 @@ def update_session_state(key,):
 
 def page():
     st.subheader("Définissez vos paramètres")
-
     st.text("L'ensemble des critères sont facultatifs.")
 
+    st.session_state["company_code"] = "steel_"
+    company_config = None
+
     # Charge la configuration YAML
     config = getYamlConfig()
+
+    for company in config['companies']:
+        print(f"{company['code']} = {st.session_state["company_code"]}")
+        if company['code'] == st.session_state["company_code"]:
+            print("ok")
+            company_config = company
     
+    print(company_config)
+
     # Vérifie si la structure inclut des 'parts' ou une liste de variables
-    if 'parts' in config['variables']:
+    if 'parts' in company_config['variables']:
         # Cas avec 'parts' : Trie les parts et affiche les champs par onglet
-        parts = config["variables"]["parts"]
+        parts = company_config["variables"]["parts"]
         parts_sorted = sorted(parts, key=lambda part: part.get('number', float('inf')))
         
         # Création de tabs pour chaque 'part' trié
@@ -32,12 +42,12 @@ def page():
         for part, tab in zip(parts_sorted, tabs):
             with tab:
                 for field in part['fields']:
-                    for field_session in st.session_state.data_dict:
+                    for field_session in st.session_state["data_dict"][st.session_state["company_code"]]:
                         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:
+        for field in st.session_state["data_dict"][st.session_state["company_code"]]:
             display_field(field)
 
 

pages/steel.py

@@ -0,0 +1,131 @@
+import datetime
+import streamlit as st
+from util import getYamlConfig
+
+company_code = "steel_"
+
+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"][st.session_state["company_code"]]:
+        if item['key'] == key:
+            item['value'] = new_value
+            break
+
+
+def page():
+    st.subheader("Définissez vos paramètres")
+    st.text("L'ensemble des critères sont facultatifs.")
+
+    st.session_state["company_code"] = company_code
+    company_config = None
+
+    # Charge la configuration YAML
+    config = getYamlConfig()
+
+    for company in config['companies']:
+        if company['code'] == st.session_state["company_code"]:
+            company_config = company
+            
+
+    # Vérifie si la structure inclut des 'parts' ou une liste de variables
+    if 'parts' in company_config['variables']:
+        # Cas avec 'parts' : Trie les parts et affiche les champs par onglet
+        parts = company_config["variables"]["parts"]
+        parts_sorted = sorted(parts, key=lambda part: part.get('number', float('inf')))
+        
+        # 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']:
+                    for field_session in st.session_state["data_dict"][st.session_state["company_code"]]:
+                        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"][st.session_state["company_code"]]:
+            display_field(field)
+
+
+def display_field(field):
+    """Helper function to create the correct input based on field 'nature'."""
+    key = 'form_' + field['key']
+    if field['nature'] == 'radio':
+        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':
+        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':
+        default_values = field.get('value', [])
+
+        st.multiselect(
+            field['label'], 
+            field['options'], 
+            default=default_values if default_values else None, 
+            key=key, 
+            on_change=update_session_state,
+            args=(key,)
+        )
+    elif field['nature'] == 'date':
+        st.date_input(
+            field['label'], 
+            value=datetime.date.fromisoformat("2011-09-20"),
+            key=key, 
+            on_change=update_session_state,
+            args=(key,)
+        )
+
+    elif field['nature'] == 'numeric':
+        st.number_input(field['label'], 
+            value=field.get('value', 0), 
+            key=key, 
+            on_change=update_session_state,
+            args=(key,)
+        )
+
+    elif field['nature'] == 'text_area':
+        st.text_area(field['label'], 
+            value=field.get('value', ""), 
+            key=key, 
+            on_change=update_session_state,
+            args=(key,)
+        )
+
+    elif field['nature'] == 'slider':
+        st.slider(field['label'], 
+            value=field.get('value', [0, 1000]), 
+            min_value=field.get('min', 0),        # Valeur minimale du slider
+            max_value=field.get('max', 5000),        # Valeur maximale du slider
+            step=field.get('step', 500),
+            key=key, 
+            on_change=update_session_state,
+            args=(key,)
+        )
+
+    else:
+        st.text_input(label=field['label'], 
+            value=field.get('value', ""), 
+            key=key, 
+            on_change=update_session_state,
+            args=(key,)
+        )
+
+page()