feat: Add option to display system prompt in PDF generation

app.py

@@ -75,6 +75,8 @@ def main():
         }
     )
 
+    st.session_state["displaySystemOnPDF"] = st.sidebar.checkbox("Afficher le prompt système", value=True)
+
     if st.sidebar.button("Générer le PDF"):
 
         chapters = []
@@ -82,7 +84,7 @@ def main():
         for chapter in st.session_state["chapters"]:
             chapters.append(st.session_state[f"chapter_{chapter['num']}"])
 
-        pdf_buffer = generate_pdf(chapters)
+        pdf_buffer = generate_pdf(chapters, displayPromptSystem=st.session_state["displaySystemOnPDF"])
 
         st.sidebar.download_button(
             label="📥 Télécharger le PDF",

utils/document.py

@@ -179,7 +179,7 @@ def markdown_to_reportlab(text):
     # text = re.sub(r"^### (.+)", r"<b><font size='12'>\1</font></b>", text, flags=re.MULTILINE)  # Titre H3
     return text
 
-def generate_pdf(chapter_data: list):
+def generate_pdf(chapter_data: list, displayPromptSystem = True):
     buffer = BytesIO()
     doc = SimpleDocTemplate(buffer, pagesize=A4)
 
@@ -211,7 +211,7 @@ def generate_pdf(chapter_data: list):
                 elif isinstance(msg, AIMessage):
                     color = "green"
                     sender = "Conseiller augmenté CEGARA"
-                elif isinstance(msg, SystemMessage):
+                elif displayPromptSystem and isinstance(msg, SystemMessage):
                     color = "red"
                     sender = "Système"
                 else: