ConseillerAugmente

Running

[email protected] commited on 12 days ago

Commit

328456e

1 Parent(s): 06eeb91

feat: Add Markdown to HTML conversion for PDF generation

- Implemented markdown_to_html function to convert Markdown text to HTML format.
- Enhanced PDF generation process to include formatted messages with appropriate styling.
- Updated config.yaml to streamline context and objective presentation.

Files changed (2) hide show

config.yaml +6 -8
utils/document.py +77 -7

config.yaml CHANGED Viewed

@@ -21,21 +21,19 @@ chapters:
       - "Quels sont les objectifs principaux ?"
     prompt_system: "ROLE : Tu es conseiller d'entreprise agricole\n
-CONTEXTE / RESSOURCES :\nTu as accès à l'entretien que le conseiller a eu avec l'exploitant agricole. Cet entretien est fourni sous forme de transcript à cette conversation (voir pièce jointe)\n
-OBJECTIF :\n
-Tu dois réaliser le paragraphe ' 1 Contexte et objectifs ' du DIAGNOSTIC PRE INSALLATION\n
-CONTRAINTE :\n
-Tu dois le présenter rigoureusement de la façon suivante :
-\n\n
 1. CONTEXTE ET OBJECTIFS\n
 * Présentation de l’exploitation à reprendre (Historique/ situation actuelle)
-[Section 01 : à compléter avec maximum 100 mots]
 * Présentation du candidat / Parcours de vie
-[Section 02 : à compléter avec maximum 100 mots]
 * Motivation et présentation du projet du candidat
 [Section 03 : à compléter avec maximum 60 mots]"

       - "Quels sont les objectifs principaux ?"
     prompt_system: "ROLE : Tu es conseiller d'entreprise agricole\n
+CONTEXTE / RESSOURCES : Tu as accès à l'entretien que le conseiller a eu avec l'exploitant agricole. Cet entretien est fourni sous forme de transcript à cette conversation (voir pièce jointe)\n
+OBJECTIF : Tu dois réaliser le paragraphe ' 1 Contexte et objectifs ' du DIAGNOSTIC PRE INSALLATION\n
+CONTRAINTE : Tu dois le présenter rigoureusement de la façon suivante :
 1. CONTEXTE ET OBJECTIFS\n
 * Présentation de l’exploitation à reprendre (Historique/ situation actuelle)
+[Section 01 : à compléter avec maximum 100 mots]\n
 * Présentation du candidat / Parcours de vie
+[Section 02 : à compléter avec maximum 100 mots]\n
 * Motivation et présentation du projet du candidat
 [Section 03 : à compléter avec maximum 60 mots]"

utils/document.py CHANGED Viewed

@@ -1,5 +1,7 @@
 import pymupdf
 import tiktoken
 from io import BytesIO
 from reportlab.lib.pagesizes import A4
@@ -113,6 +115,69 @@ def audit_descriptif_pdf(file,max_img_width=500) -> dict:
     return global_audit
 def generate_pdf(chapter_data: list):
     buffer = BytesIO()
@@ -142,20 +207,25 @@ def generate_pdf(chapter_data: list):
             for msg in chapter["messages"]:
                 if isinstance(msg, HumanMessage):
                     color = "blue"
-                    sender = "👤 Utilisateur"
                 elif isinstance(msg, AIMessage):
                     color = "green"
-                    sender = "🤖 IA"
                 elif isinstance(msg, SystemMessage):
                     color = "red"
-                    sender = "⚙️ Système"
                 else:
                     color = "black"
-                    sender = "🔹 Message"
-                print(msg.content)
-                elements.append(Paragraph(f"<b><font color='{color}'>{sender} :</font></b> {msg.content}", style_message))
-                elements.append(Spacer(1, 5))
         elements.append(Spacer(1, 15))  # Espacement entre discussions

 import pymupdf
 import tiktoken
+import markdown
+import re
 from io import BytesIO
 from reportlab.lib.pagesizes import A4
     return global_audit
+# Fonction pour convertir le Markdown en HTML pour le PDF
+def markdown_to_html(md_text):
+    return markdown.markdown(md_text, output_format='html' )
+    html = """
+    <html>
+        <head>
+            <style>
+                body { font-family: Arial, sans-serif; margin: 40px; }
+                h1 { text-align: center; color: #333; }
+                h2 { border-bottom: 2px solid #666; padding-bottom: 5px; margin-top: 30px; }
+                .message { margin-bottom: 10px; padding: 10px; border-radius: 5px; }
+                .human { background-color: #e1f5fe; }
+                .ai { background-color: #e8f5e9; }
+                .system { background-color: #ffebee; }
+            </style>
+        </head>
+    <body>
+        <h1>Conseiller augmenté CEGARA</h1>
+    """
+    # Trier les chapitres par "num"
+    chapters = sorted(chapter_data, key=lambda x: x["num"])
+    for chapter in chapters:
+        html += f"<h2>Chapitre {chapter['num']}: {chapter['title']}</h2>"
+        if len(chapter["messages"]) > 1 :
+            # Affichage des messages
+            for msg in chapter["messages"]:
+                if  isinstance(msg, HumanMessage):
+                    sender = "Utilisateur"
+                    css_class = "human"
+                elif isinstance(msg, AIMessage):
+                    sender = "IA"
+                    css_class = "ai"
+                elif isinstance(msg, SystemMessage):
+                    sender = "Système"
+                    css_class = "system"
+                else:
+                    sender = "Message"
+                    css_class = ""
+                html += f"""
+                <div class="message {css_class}">
+                    <b>{sender} :</b> {markdown_to_html(msg.content)}
+                </div>
+                """
+    html += "</body></html>"
+    return generate_pdf_from_html(html)
+# Fonction pour convertir le Markdown en texte enrichi compatible ReportLab
+def markdown_to_reportlab(text):
+    # text = text.replace("**", "<b>").replace("__", "<b>")  # Gras
+    # text = text.replace("*", "<i>").replace("_", "<i>")    # Italique
+    text = text.replace("\n", "<br/>")    # Italique
+    # text = re.sub(r"\n- (.+)", r"\n• \1", text)  # Listes à puces
+    # text = re.sub(r"^# (.+)", r"<b><font size='16'>\1</font></b>", text, flags=re.MULTILINE)  # Titre H1
+    # text = re.sub(r"^## (.+)", r"<b><font size='14'>\1</font></b>", text, flags=re.MULTILINE)  # Titre H2
+    # 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):
     buffer = BytesIO()
             for msg in chapter["messages"]:
                 if isinstance(msg, HumanMessage):
                     color = "blue"
+                    sender = "Utilisateur"
                 elif isinstance(msg, AIMessage):
                     color = "green"
+                    sender = "Conseiller augmenté CEGARA"
                 elif isinstance(msg, SystemMessage):
                     color = "red"
+                    sender = "Système"
                 else:
                     color = "black"
+                    sender = "Message"
+                elements.append(Paragraph(f"<b><font color='{color}'>{sender}</font></b>", style_message))
+                content = msg.content
+                content = markdown_to_html(content)
+                content = markdown_to_reportlab(content)
+                elements.append(Paragraph(content, style_message))
+                elements.append(Spacer(1, 10))
         elements.append(Spacer(1, 15))  # Espacement entre discussions