Create pdf_tool.py

pdf_tool.py

@@ -0,0 +1,75 @@
+# pdf_tool.py
+from reportlab.lib.pagesizes import letter
+from reportlab.platypus import SimpleDocTemplate, Paragraph
+from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
+from reportlab.lib import colors
+from crewai.tools.base_tool import BaseTool
+from typing import Type
+from pydantic import BaseModel, Field
+
+class PDFToolInput(BaseModel):
+    content: str = Field(..., description="Contenu de l'exposé en format Markdown")
+    output_path: str = Field(..., description="Chemin de sortie du fichier PDF")
+
+class PDFTool(BaseTool):
+    name: str = "Generate PDF"
+    description: str = "Outil pour générer un document PDF à partir de texte en format Markdown."
+    args_schema: Type[BaseModel] = PDFToolInput
+
+    def _run(
+        self,
+        content: str,
+        output_path: str = "expose.pdf",
+        **kwargs,
+    ) -> str:
+        try:
+            doc = SimpleDocTemplate(output_path, pagesize=letter)
+            styles = getSampleStyleSheet()
+
+            # Ajout d'un style personnalisé pour le texte normal
+            styles.add(ParagraphStyle(name='CustomBodyText',
+                                      parent=styles['Normal'],
+                                      fontSize=12,
+                                      leading=14,
+                                      spaceAfter=10))
+
+            # Ajout d'un style personnalisé pour les titres de section
+            styles.add(ParagraphStyle(name='CustomHeading1',
+                                      parent=styles['Heading1'],
+                                      fontSize=18,
+                                      leading=22,
+                                      spaceBefore=20,
+                                      spaceAfter=6,
+                                      textColor=colors.HexColor("#2c3e50"))) # Couleur bleu foncé
+
+            # Ajout d'un style personnalisé pour les titres de sous-section
+            styles.add(ParagraphStyle(name='CustomHeading2',
+                                      parent=styles['Heading2'],
+                                      fontSize=14,
+                                      leading=18,
+                                      spaceBefore=10,
+                                      spaceAfter=4,
+                                      textColor=colors.HexColor("#34495e"))) # Couleur bleu-gris
+
+            # Séparer le contenu en sections en utilisant les sauts de ligne comme délimiteurs
+            sections = content.split("\n\n")
+
+            Story = []
+            for section in sections:
+                # Déterminer si la section est un titre de section, un titre de sous-section ou du texte normal
+                if section.startswith("# "):
+                    # Titre de section
+                    title = section[2:].strip()
+                    Story.append(Paragraph(title, styles["CustomHeading1"]))
+                elif section.startswith("## "):
+                    # Titre de sous-section
+                    subtitle = section[3:].strip()
+                    Story.append(Paragraph(subtitle, styles["CustomHeading2"]))
+                else:
+                    # Texte normal
+                    Story.append(Paragraph(section, styles["CustomBodyText"]))
+
+            doc.build(Story)
+            return f"Fichier PDF généré avec succès : {output_path}"
+        except Exception as e:
+            return f"Erreur lors de la génération du PDF : {e}"