Include pages by chapter

app.py

@@ -33,8 +33,8 @@ def init_app():
         st.session_state["chapters"] = config['chapters']
 
         for chapter in st.session_state["chapters"]:
-            st.session_state[f"chapter_{chapter['key']}"] = {
-                "num": 0,
+            st.session_state[f"chapter_{chapter['num']}"] = {
+                "num": chapter['num'],
                 "messages": [],
                 "prompt_system": chapter['prompt_system'],
                 "prompt": chapter['prompts'],
@@ -65,10 +65,13 @@ def main():
     # st.write(f"**User ID:** {userId}")
 
     documents = st.Page("pages/documents.py", title="Vos documents", icon="📂")
-    prompt_system = st.Page("pages/prompt_system.py", title="Prompt système", icon="🖊️", default=True)
     form = st.Page("pages/chapter_params.py", title="Paramètres", icon="📋")
 
-    chapters_pages = []
+    chapters_pages = [
+        st.Page(f"pages/chap_{chapter['num']}.py", title=chapter['name'], icon="📄")
+        for chapter in st.session_state["chapters"]
+    ]
+
 
     pg = st.navigation(
         {
@@ -76,7 +79,6 @@ def main():
                 documents,
             ],
             "Configurations": [
-                # prompt_system,
                 form,
             ],
             "Dialogue": chapters_pages,

config.yaml

@@ -12,50 +12,6 @@ prompts:
     - "Quelles sont les techniques de lutte biologique contre les nuisibles ?"
 
 
-# ----------------------------------------------------------
-# Gestion du formulaire dynamique
-# ----------------------------------------------------------
-# Version 1 : Liste de champs sous 'variables' (sans structure de 'parts')
-# ----------------------------------------------------------
-# Chaque champ est défini directement sous 'variables'.
-# Vous pouvez ajouter des champs en suivant ce format.
-# Pour utiliser cette version, laissez-la telle quelle et commentez la version 2 ci-dessous.
-
-# variables:
-#   - label: "Région viticole"
-#     nature: 'radio'
-#     key: param1
-#     options: ["Champagne", "Vin de Bordeaux", "Occitanie"]
-#     value: # Valeur par défaut
-
-#   - label: "Millésime"
-#     nature: 'selectbox'
-#     key: param2
-#     options: ["2019", "2020", "2021", "2022"]
-#     value: # Valeur par défaut
-
-#   - label: "Budget prévu (en €)"
-#     nature: 'numeric'
-#     key: param3
-#     value: # Valeur par défaut
-
-#   - label: "Date de récolte"
-#     nature: 'date'
-#     key: param4
-#     value: # Valeur par défaut
-
-#   - label: "Remarques sur la récolte"
-#     nature: 'text_area'
-#     key: param5
-#     value: # Valeur par défaut
-
-
-# Version 2 : Structure de 'parts' pour organiser les champs en sections
-# ----------------------------------------------------------------------
-# 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.
-
 chapters:
   - name: "Contexte et objectifs"
     num: 1
@@ -158,10 +114,6 @@ variables:
 prompt_system: "Prompt system par défaut"
 
 prompt_template: "
-Paramètre 1 : {param1}
-Paramètre 2 : {param2}
-Paramètre 3 : {param3}
-Paramètre 4 : {param4}
 
 Document de référence : {documentContext}
 

pages/chap_1.py

@@ -0,0 +1,8 @@
+import streamlit as st
+from pages.chatbot import page
+
+# Définir les paramètres à transmettre
+st.session_state["current_page"] = 1
+
+# Appeler la fonction page avec les paramètres
+page()

pages/chap_2.py

@@ -0,0 +1,8 @@
+import streamlit as st
+from pages.chatbot import page
+
+# Définir les paramètres à transmettre
+st.session_state["current_page"] = 2
+
+# Appeler la fonction page avec les paramètres
+page()

pages/chap_3.py

@@ -0,0 +1,8 @@
+import streamlit as st
+from pages.chatbot import page
+
+# Définir les paramètres à transmettre
+st.session_state["current_page"] = 3
+
+# Appeler la fonction page avec les paramètres
+page()

pages/chap_4.py

@@ -0,0 +1,8 @@
+import streamlit as st
+from pages.chatbot import page
+
+# Définir les paramètres à transmettre
+st.session_state["current_page"] = 4
+
+# Appeler la fonction page avec les paramètres
+page()

pages/chap_5.py

@@ -0,0 +1,8 @@
+import streamlit as st
+from pages.chatbot import page
+
+# Définir les paramètres à transmettre
+st.session_state["current_page"] = 5
+
+# Appeler la fonction page avec les paramètres
+page()

pages/chatbot.py

@@ -1,12 +1,22 @@
 import streamlit as st
+import uuid
 from langchain_core.messages import AIMessage, HumanMessage
 from model import selector
 from util import getYamlConfig
 from st_copy_to_clipboard import st_copy_to_clipboard
 
+chapter_num = 0
+chapter_session_key = f"chapter_{chapter_num}"
+
+def setChapter(num: int):
+    global chapter_num, chapter_session_key
+    chapter_num = num
+    chapter_session_key = f"chapter_{chapter_num}"
+
+
 def display_messages():
 
-    for i, message in enumerate(st.session_state.chat_history):
+    for i, message in enumerate(st.session_state[chapter_session_key]["messages"]):
         if isinstance(message, AIMessage):
             with st.chat_message("AI"):
                 # Display the model from the kwargs
@@ -22,46 +32,71 @@ def display_messages():
 
 def launchQuery(query: str = None):
 
+    print(f"launchQuery : {query}")
     # 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 [],
+            messages=st.session_state[chapter_session_key]["messages"] if "messages" in st.session_state[chapter_session_key] 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.session_state[chapter_session_key]["messages"].append(AIMessage(content=full_response, kwargs={"model": st.session_state["assistant"].getReadableModel()}))
     st.rerun()
 
 
-def show_prompts():
-    yaml_data = getYamlConfig()["prompts"]
+def show_prompts(prompts: list[str]):
     
     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)
+    for i, p in enumerate(prompts):
+        if expander.button(p, key=f"button_{chapter_num}_{i}"):
+            launchQuery(p)
 
 
 def page():
-    st.subheader("Posez vos questions")
+    yaml_data = getYamlConfig()["chapters"]
+    
+
+    ###############################################
+    # Récupération du chapitre depuis YAML Config #
+    ###############################################
+
+    chapter_num = st.session_state.get('current_page', 0)
+    if chapter_num == 0:
+        return
+    
+    setChapter(int(chapter_num))
+    
+    chapter = next((ch for ch in yaml_data if ch['num'] == chapter_num), None)
+    if not chapter:
+        st.text("Chapitre non trouvé")
+        return
+
+
+    #################
+    # Some controls #
+    #################
 
     if "assistant" not in st.session_state:
         st.text("Assistant non initialisé")
 
-    if "chat_history" not in st.session_state:
-        st.session_state["chat_history"] = []
+    if "messages" not in st.session_state[chapter_session_key]:
+        st.session_state[chapter_session_key]["messages"] = []
+
+
+    ############
+    #### UI ####
+    ############
+
+    st.subheader(chapter['name'])
 
     st.markdown("<style>iframe{height:50px;}</style>", unsafe_allow_html=True)
 
     # Collpase for default prompts
-    show_prompts()
+    show_prompts(chapter['prompts'])
 
     # Models selector
     selector.ModelSelector()
@@ -70,10 +105,13 @@ def page():
     display_messages()
 
 
-    user_query = st.chat_input("")
+    user_query = st.chat_input("", key=str(uuid.uuid4()))
+    
     if user_query is not None and user_query != "":
+        print(user_query)
 
-        st.session_state["chat_history"].append(HumanMessage(content=user_query))
+        st.session_state[chapter_session_key]["messages"].append(HumanMessage(content=user_query))
+        print(st.session_state[chapter_session_key]["messages"])
         
         # Stream and display response
         launchQuery(user_query)