Enhance PDF generation and update configurations: Add PDF_LOGO_PATH to .env.example, include markdown-pdf in requirements.txt, and refactor PDF generation logic to support logo integration. Introduce tests for PDF generation from Markdown input and update summarizer agent to streamline export functionality.

.env.example

@@ -18,4 +18,7 @@ LANGCHAIN_API_KEY=
 LANGCHAIN_PROJECT=routeur-ia
 
 # Voice WebRTC ICE servers (STUN/TURN pour NAT traversal sur HF Spaces)
-# DAILY_API_KEY=61efebb10fa3956006a11e194980876c4453311ef25737fb2e434bc040326deb
+# DAILY_API_KEY=61efebb10fa3956006a11e194980876c4453311ef25737fb2e434bc040326deb
+
+# PDF - logo en haut du document (section dédiée)
+PDF_LOGO_PATH=assets/logo.png

config/settings.py

@@ -108,6 +108,9 @@ class Settings(BaseSettings):
     # Voice / Daily.co (alternative transport for HF Spaces, NAT traversal)
     daily_api_key: Optional[str] = None
 
+    # PDF branding - logo displayed in top section (assets/logo.png)
+    pdf_logo_path: Optional[str] = None
+
     model_config = SettingsConfigDict(
         env_file=".env",
         env_file_encoding="utf-8",

graphs/agents/summarizer_agent.py

@@ -1,5 +1,6 @@
 """Summarizer agents: split into two nodes (LLM -> Export PDF/Upload)."""
-from typing import Callable, Optional
+from io import BytesIO
+from typing import Callable, Union
 from datetime import datetime, timezone
 
 from langchain_core.messages import SystemMessage
@@ -38,8 +39,7 @@ def summarizer_llm_node(
 
 
 def summarizer_export_node(
-    md_to_html: Callable[[str, Optional[str]], str],
-    html_to_pdf: Callable[[str], bytes],
+    markdown_to_pdf: Callable[[str], Union[bytes, BytesIO]],
     upload_pdf: Callable[[bytes, str], str],
 ):
     """Node: takes `summary_markdown` from state, builds PDF and uploads it.
@@ -54,9 +54,8 @@ def summarizer_export_node(
         messages = list(state.get("messages", []))
         summary_markdown = state.get("summary_markdown", "") or ""
 
-        # Convert to HTML, then PDF, then upload
-        html = md_to_html(summary_markdown)
-        pdf_result = html_to_pdf(html)
+        # Convert Markdown to PDF
+        pdf_result = markdown_to_pdf(summary_markdown)
 
         # Ensure pdf_bytes is of type bytes (not BytesIO)
         if hasattr(pdf_result, "read"):
@@ -105,13 +104,12 @@ def summarizer_export_node(
 
 def summarizer_node(
     llm: BaseChatModel,
-    md_to_html: Callable[[str, Optional[str]], str],
-    html_to_pdf: Callable[[str], bytes],
+    markdown_to_pdf: Callable[[str], Union[bytes, BytesIO]],
     upload_pdf: Callable[[bytes, str], str],
 ):
     """Compatibility wrapper: runs LLM summarization then export in one node."""
     llm_runner = summarizer_llm_node(llm)
-    export_runner = summarizer_export_node(md_to_html, html_to_pdf, upload_pdf)
+    export_runner = summarizer_export_node(markdown_to_pdf, upload_pdf)
 
     def _run(state: AgentState) -> AgentState:
         state_after_llm = llm_runner(state)

graphs/workflows/conversation_with_summary.py

@@ -6,8 +6,7 @@ from graphs.state import AgentState
 from graphs.nodes.retrieval import retrieve_both_types
 from graphs.agents.chat_agent import chat_node
 from graphs.agents.summarizer_agent import summarizer_llm_node, summarizer_export_node
-from tools.markdown import markdown_to_html
-from tools.pdf import html_to_pdf
+from tools.pdf import markdown_to_pdf
 from tools.storage import upload_pdf_to_supabase
 
 
@@ -21,8 +20,7 @@ def create_conversation_with_summary_graph(llm: BaseChatModel):
     workflow.add_node(
         "summarizer_export",
         summarizer_export_node(
-            md_to_html=markdown_to_html,
-            html_to_pdf=html_to_pdf,
+            markdown_to_pdf=markdown_to_pdf,
             upload_pdf=upload_pdf_to_supabase,
         ),
     )

graphs/workflows/orchestrated.py

@@ -7,8 +7,7 @@ from graphs.agents.classifier_agent import classifier_node
 from graphs.nodes.retrieval import retrieve_catalogue, retrieve_projects
 from graphs.agents.chat_agent import chat_node
 from graphs.agents.summarizer_agent import summarizer_llm_node, summarizer_export_node
-# from tools.markdown import markdown_to_html
-from tools.pdf import html_to_pdf, markdown_to_html
+from tools.pdf import markdown_to_pdf
 from tools.storage import upload_pdf_to_supabase
 
 
@@ -30,8 +29,7 @@ def create_orchestrated_graph(llm: BaseChatModel, checkpointer=None):
     workflow.add_node(
         "summarizer_export",
         summarizer_export_node(
-            md_to_html=markdown_to_html,
-            html_to_pdf=html_to_pdf,
+            markdown_to_pdf=markdown_to_pdf,
             upload_pdf=upload_pdf_to_supabase,
         ),
     )

requirements.txt

@@ -37,6 +37,10 @@ daily-python>=0.10.0 ; sys_platform == "linux"
 # Summarizer tooling
 reportlab>=4.4.4,<5
 markdown>=3.7,<4
+markdown-pdf>=1.13.0
 
 # Ecologits Carbon Footprint
 ecologits>=0.8.2,<1
+
+# Tests
+pytest>=8.0.0,<9

tests/ai_synthesis_response.md

@@ -0,0 +1,71 @@
+# **Synthèse de l'entretien avec Cédric LAMBERT**
+*Exploitant en arboriculture et productions de semences*
+
+---
+
+## **Contexte**
+- **Exploitation** : Arboriculture (espèces/variétés à préciser) et productions de semences (légumières, fourragères, etc.).
+- **Enjeux identifiés** (à confirmer lors de la visite) :
+  - Optimisation des itinéraires techniques (arboriculture et semences).
+  - Gestion des aléas climatiques (gel, sécheresse) et sanitaires (maladies, ravageurs).
+  - Rentabilité économique et diversification des débouchés.
+  - Transition agroécologique (réduction des intrants, certification HVE/bio).
+- **Objectif de la visite** : Réaliser un premier diagnostic technique et économique pour proposer des solutions adaptées (prestations, formations).
+
+---
+
+## **Objectifs de l'agriculteur**
+*(À compléter après l’entretien)*
+- **Priorités techniques** :
+  - Exemple : Améliorer la gestion du gel en verger, réduire l’usage des phytosanitaires.
+- **Priorités économiques** :
+  - Exemple : Augmenter la marge brute des cultures, diversifier les canaux de vente.
+- **Projets** :
+  - Exemple : Plantation de nouveaux vergers, développement de la vente directe de semences.
+- **Attentes vis-à-vis de la Chambre d’Agriculture** :
+  - Exemple : Accompagnement sur la transition agroenvironnementale, formation des salariés.
+
+---
+
+## **Recommandations**
+
+### **Prestations**
+| **Prestation** | **Page** | **Source** | **Contact** |
+|----------------|----------|------------|-------------|
+| Conseil en conduite de vergers (itinéraire technique, protection phytosanitaire) | p.28 | Catalogue Prestations 2024 – Chambre d’Agriculture Pays de la Loire | Service Productions Végétales – [Voir contacts p.50-51] |
+| Diagnostic et plan d’action agro-environnemental (HVE, bio, réduction des intrants) | p.28 | Catalogue Prestations 2024 – Chambre d’Agriculture Pays de la Loire | Service Environnement – [Voir contacts p.50-51] |
+| Accompagnement à la plantation et gestion du bocage (haies, agroforesterie) | p.32 | Catalogue Prestations 2024 – Chambre d’Agriculture Pays de la Loire | Aurélie PAYRAUDEAU (p.32) |
+| Étude économique et stratégie commerciale (diversification, vente directe) | p.16 | Catalogue Prestations 2024 – Chambre d’Agriculture Pays de la Loire | Service Circuits Courts – [Voir contacts p.50-51] |
+
+---
+
+### **Formations**
+| **Formation** | **Page** | **Source** | **Contact** | **Prochaine date / Message par défaut** |
+|---------------|----------|------------|-------------|----------------------------------------|
+| Réussir ses plantations : des choix stratégiques (arboriculture) | p.103 | Catalogue Formations 2024-2025 – Chambre d’Agriculture Pays de la Loire | Florent BANCTEL (conseiller Viticulture) | 05/03 et 12/03/2025 (Clisson) ou contacter pour 49. |
+| Se protéger contre le gel de printemps | p.103 | Catalogue Formations 2024-2025 – Chambre d’Agriculture Pays de la Loire | Thomas CHASSAING (conseiller Viticulture) | 04/03/2025 (44) ou 05/03/2025 (49). |
+| Renforcer l'agronomie pour plus de durabilité | p.46 | Catalogue Formations 2024-2025 – Chambre d’Agriculture Pays de la Loire | Alexandre HATET (expert Sols) | Contacter le service formation pour connaître la prochaine date. |
+
+---
+
+## **Prochaines étapes**
+1. **Après l’entretien** :
+   - Envoyer un email de remerciement à Cédric LAMBERT avec :
+     - Un résumé des points clés abordés.
+     - Les prestations/formations proposées (liens vers les pages du catalogue).
+     - Une proposition de rendez-vous pour affiner les solutions.
+   - Transmettre les documents demandés (parcellaire, registres) au service concerné pour analyse.
+
+2. **Plan d’action personnalisé** :
+   - Prioriser les prestations/formations en fonction de ses enjeux (ex : si gel = formation "Se protéger contre le gel").
+   - Proposer un calendrier (ex : diagnostic agroenvironnemental en mars, formation en avril).
+
+3. **Suivi** :
+   - Planifier un deuxième rendez-vous pour valider le plan d’action et les devis.
+   - Informer les services internes (Environnement, Productions Végétales) pour préparer les interventions.
+
+---
+**Documents joints** (à adapter) :
+- Fiche de synthèse personnalisée (cette version).
+- Extraits des catalogues prestations/formations (pages citées).
+- Modèle d’email de suivi.

tests/conftest.py

@@ -0,0 +1,6 @@
+"""Pytest configuration and fixtures."""
+import sys
+from pathlib import Path
+
+# Add project root to path so "from tools.xxx" imports work
+sys.path.insert(0, str(Path(__file__).resolve().parent.parent))

tests/test_pdf_generation.py

@@ -0,0 +1,21 @@
+# tests/test_pdf_generation.py
+"""Test PDF generation from Markdown input (markdown-pdf)."""
+from pathlib import Path
+
+from tools.pdf import markdown_to_pdf
+
+_DIR = Path(__file__).resolve().parent
+MD_INPUT_PATH = _DIR / "ai_synthesis_response.md"
+PDF_OUTPUT_PATH = _DIR / "ai_synthesis_response.pdf"
+
+
+def test_generates_pdf_from_markdown_locally():
+    """Generate a PDF from markdown input and verify it is written locally."""
+    md_input = MD_INPUT_PATH.read_text(encoding="utf-8")
+    buffer = markdown_to_pdf(md_input)
+
+    PDF_OUTPUT_PATH.write_bytes(buffer.getvalue())
+
+    assert PDF_OUTPUT_PATH.exists()
+    assert PDF_OUTPUT_PATH.stat().st_size > 0
+    assert PDF_OUTPUT_PATH.read_bytes()[:4] == b"%PDF"

tools/pdf.py

@@ -1,50 +1,51 @@
-import markdown
-import re
-
+"""PDF generation from Markdown using markdown-pdf."""
 from io import BytesIO
-from reportlab.lib.pagesizes import A4
-from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer
-from reportlab.lib.styles import getSampleStyleSheet
-from reportlab.lib.enums import TA_CENTER
-from langchain_core.messages import AIMessage, HumanMessage, SystemMessage
-
-# Fonction pour convertir le Markdown en HTML pour le PDF
-def markdown_to_html(md_text):
-    return markdown.markdown(md_text, output_format='html' )
-
-
-def html_to_pdf(html):
-    
-    html = html.replace("\n", "<br>")
-        # Nettoyage minimal du HTML pour compatibilité ReportLab
-    html = html.replace("<br>", "<br/>")
-    html = html.replace("<br />", "<br/>")
-
-    # Supprimer les balises non supportées (ul, ol, li, h1, h2, etc.)
-    html = re.sub(r"</?(ul|ol|li|h\d|blockquote|hr)>", "", html)
-    html = html.replace("**", "<b>")
-    html = html.replace("*", "<i>")
-    html = html.replace("~", "<s>")
-    html = html.replace("```markdown", "")
-    html = html.replace("```", "")
+from pathlib import Path
+from typing import Optional
+
+from markdown_pdf import MarkdownPdf, Section
+
+_PROJECT_ROOT = Path(__file__).resolve().parent.parent
+
+
+def markdown_to_pdf(
+    md_text: str,
+    toc_level: int = 2,
+    logo_path: Optional[str] = None,
+) -> BytesIO:
+    """Convert Markdown to PDF using markdown-pdf (PyMuPDF + markdown-it-py).
+
+    Supports UTF-8, tables, links, images, and TOC from headings.
+    Optionally adds a dedicated logo section at the top (centered) if logo_path is set.
+    """
+    if logo_path is None:
+        try:
+            from config.settings import settings
+            logo_path = getattr(settings, "pdf_logo_path", None)
+        except Exception:
+            logo_path = None
+
+    pdf = MarkdownPdf(toc_level=toc_level, optimize=True)
+
+    # Section 1: logo centered at top (if file exists)
+    logo_file = _PROJECT_ROOT / logo_path if logo_path else None
+    if logo_file and logo_file.is_file():
+        logo_md = f"![Logo]({logo_path})"
+        logo_css = "img { display: block; margin-left: auto; margin-right: auto; }"
+        logo_section = Section(logo_md, toc=False)
+        logo_section.root = str(_PROJECT_ROOT)
+        pdf.add_section(logo_section, user_css=logo_css)
+
+    # Section 2: main content (réduit la police)
+    content_css = (
+        "body, p, li, ul, ol, td, th { font-size: 9pt; } "
+        "h1 { font-size: 14pt; } h2 { font-size: 12pt; } h3 { font-size: 10pt; }"
+    )
+    content_section = Section(md_text)
+    content_section.root = str(_PROJECT_ROOT)
+    pdf.add_section(content_section, user_css=content_css)
 
     buffer = BytesIO()
-    doc = SimpleDocTemplate(buffer, pagesize=A4)
-
-    styles = getSampleStyleSheet()
-    style_title = styles["Title"]
-    style_title.alignment = TA_CENTER  # Centrer le titre
-    style_header = styles["Heading2"]
-    style_message = styles["BodyText"]
-
-    elements = []
-
-    # Titre principal du document
-    elements.append(Paragraph("Synthèse", style_title))
-    elements.append(Spacer(1, 20))  # Espacement après le titre
-
-    elements.append(Paragraph(html, style_message))
-
-    doc.build(elements)
+    pdf.save_bytes(buffer)
     buffer.seek(0)
     return buffer