Update contract_builder.py

contract_builder.py

@@ -31,7 +31,8 @@ class ContractBuilder:
         styles.add(ParagraphStyle(name='Sous-titre', 
                                  fontName='Helvetica-Bold', 
                                  fontSize=12, 
-                                 spaceAfter=10))
+                                 spaceAfter=10,
+                                 alignment=TA_CENTER))
         styles.add(ParagraphStyle(name='Normal', 
                                  fontName='Helvetica', 
                                  fontSize=10, 
@@ -41,6 +42,10 @@ class ContractBuilder:
                                  fontName='Helvetica-Bold', 
                                  fontSize=11, 
                                  spaceAfter=8))
+        styles.add(ParagraphStyle(name='Sous-article', 
+                                 fontName='Helvetica-Bold', 
+                                 fontSize=10, 
+                                 spaceAfter=6))
         return styles
     
     @staticmethod
@@ -74,14 +79,24 @@ class ContractBuilder:
         # 1. Titre du contrat
         title = ContractTemplates.get_title(contract_type)
         elements.append(Paragraph(title, styles['Titre']))
-        elements.append(Spacer(1, 10))
+        elements.append(Spacer(1, 15))
         
         # 2. Préambule
         preamble_text = ContractTemplates.get_preamble_text(contract_type, author_type, author_info)
-        for paragraph in preamble_text.split('\n\n'):
+        paragraphs = preamble_text.split('\n\n')
+        for i, paragraph in enumerate(paragraphs):
             if paragraph.strip():
-                elements.append(Paragraph(paragraph, styles['Normal']))
-                elements.append(Spacer(1, 6))
+                # Si c'est un titre comme "PRÉAMBULE" ou "ENTRE LES SOUSSIGNÉS"
+                if paragraph.isupper() and len(paragraph) < 50:
+                    elements.append(Paragraph(paragraph, styles['Sous-titre']))
+                else:
+                    elements.append(Paragraph(paragraph, styles['Normal']))
+                
+                # Ajouter un espace plus grand après les titres
+                if paragraph.isupper() and len(paragraph) < 50:
+                    elements.append(Spacer(1, 10))
+                else:
+                    elements.append(Spacer(1, 6))
         
         # 3. Article 1 - Objet
         object_clause = ContractTemplates.get_object_clause(
@@ -89,8 +104,20 @@ class ContractBuilder:
             sanitize_text(work_description), 
             sanitize_text(image_description)
         )
-        elements.append(Paragraph(object_clause, styles['Article']))
-        elements.append(Spacer(1, 10))
+        
+        # Traitement de l'article objet
+        paragraphs = object_clause.split('\n\n')
+        for i, paragraph in enumerate(paragraphs):
+            if paragraph.strip():
+                # Si c'est un titre d'article (commence par "ARTICLE")
+                if paragraph.startswith("ARTICLE"):
+                    elements.append(Paragraph(paragraph, styles['Article']))
+                # Si c'est un sous-titre (comme "1.1 Œuvre concernée")
+                elif paragraph.strip().startswith("1.") and len(paragraph.split("\n")[0]) < 50:
+                    elements.append(Paragraph(paragraph, styles['Sous-article']))
+                else:
+                    elements.append(Paragraph(paragraph, styles['Normal']))
+                elements.append(Spacer(1, 6))
         
         # Compteur d'articles
         article_num = 2
@@ -98,9 +125,17 @@ class ContractBuilder:
         # 4. Article 2 - Droits cédés (pour contrat droits d'auteur)
         if "Auteur (droits d'auteur)" in contract_type:
             rights_clause = ContractTemplates.get_author_rights_clause(is_free, is_exclusive, additional_rights)
-            for paragraph in rights_clause.split('\n\n'):
+            paragraphs = rights_clause.split('\n\n')
+            for paragraph in paragraphs:
                 if paragraph.strip():
-                    elements.append(Paragraph(paragraph, styles['Normal']))
+                    # Si c'est un titre d'article (commence par "ARTICLE")
+                    if paragraph.startswith("ARTICLE"):
+                        elements.append(Paragraph(paragraph, styles['Article']))
+                    # Si c'est un sous-titre (comme "2.1 Nature de la cession")
+                    elif paragraph.strip().startswith("2.") and len(paragraph.split("\n")[0]) < 50:
+                        elements.append(Paragraph(paragraph, styles['Sous-article']))
+                    else:
+                        elements.append(Paragraph(paragraph, styles['Normal']))
                     elements.append(Spacer(1, 6))
             article_num += 1
         
@@ -108,67 +143,148 @@ class ContractBuilder:
         if "Image (droit à l'image)" in contract_type:
             image_rights_article_num = article_num if "Auteur (droits d'auteur)" not in contract_type else article_num
             image_clause = ContractTemplates.get_image_rights_clause(is_free, is_exclusive)
-            image_clause = image_clause.replace("Article 2", f"Article {image_rights_article_num}")
-            for paragraph in image_clause.split('\n\n'):
+            # Remplacer le numéro d'article si nécessaire
+            if image_rights_article_num != 3:
+                image_clause = image_clause.replace("ARTICLE 3", f"ARTICLE {image_rights_article_num}")
+                # Aussi remplacer les sous-titres 3.x par le bon numéro
+                image_clause = image_clause.replace("3.1", f"{image_rights_article_num}.1")
+                image_clause = image_clause.replace("3.2", f"{image_rights_article_num}.2")
+                image_clause = image_clause.replace("3.3", f"{image_rights_article_num}.3")
+                image_clause = image_clause.replace("3.4", f"{image_rights_article_num}.4")
+            
+            paragraphs = image_clause.split('\n\n')
+            for paragraph in paragraphs:
                 if paragraph.strip():
-                    elements.append(Paragraph(paragraph, styles['Normal']))
+                    # Si c'est un titre d'article (commence par "ARTICLE")
+                    if paragraph.startswith("ARTICLE"):
+                        elements.append(Paragraph(paragraph, styles['Article']))
+                    # Si c'est un sous-titre (comme "3.1 Objet de l'autorisation")
+                    elif paragraph.strip()[0].isdigit() and "." in paragraph.strip()[:3] and len(paragraph.split("\n")[0]) < 50:
+                        elements.append(Paragraph(paragraph, styles['Sous-article']))
+                    else:
+                        elements.append(Paragraph(paragraph, styles['Normal']))
                     elements.append(Spacer(1, 6))
             article_num += 1
         
         # 6. Article - Durée et territoire
         duration_clause = ContractTemplates.get_duration_territory_clause(article_num)
-        for paragraph in duration_clause.split('\n\n'):
+        paragraphs = duration_clause.split('\n\n')
+        for paragraph in paragraphs:
             if paragraph.strip():
-                elements.append(Paragraph(paragraph, styles['Normal']))
+                # Si c'est un titre d'article (commence par "ARTICLE")
+                if paragraph.startswith("ARTICLE"):
+                    elements.append(Paragraph(paragraph, styles['Article']))
+                # Si c'est un sous-titre (comme "4.1 Durée")
+                elif paragraph.strip()[0].isdigit() and "." in paragraph.strip()[:3] and len(paragraph.split("\n")[0]) < 50:
+                    elements.append(Paragraph(paragraph, styles['Sous-article']))
+                else:
+                    elements.append(Paragraph(paragraph, styles['Normal']))
                 elements.append(Spacer(1, 6))
         article_num += 1
         
         # 7. Article - Supports d'exploitation
         final_supports = ensure_default_supports(supports)
         supports_clause = ContractTemplates.get_supports_clause(article_num, final_supports)
-        for paragraph in supports_clause.split('\n\n'):
+        paragraphs = supports_clause.split('\n\n')
+        for paragraph in paragraphs:
             if paragraph.strip():
-                elements.append(Paragraph(paragraph, styles['Normal']))
+                # Si c'est un titre d'article (commence par "ARTICLE")
+                if paragraph.startswith("ARTICLE"):
+                    elements.append(Paragraph(paragraph, styles['Article']))
+                # Si c'est un sous-titre (comme "5.1 Supports autorisés")
+                elif paragraph.strip()[0].isdigit() and "." in paragraph.strip()[:3] and len(paragraph.split("\n")[0]) < 50:
+                    elements.append(Paragraph(paragraph, styles['Sous-article']))
+                else:
+                    elements.append(Paragraph(paragraph, styles['Normal']))
                 elements.append(Spacer(1, 6))
         article_num += 1
         
         # 8. Article - Rémunération
         remuneration_clause = ContractTemplates.get_remuneration_clause(article_num, is_free, sanitize_text(remuneration))
-        for paragraph in remuneration_clause.split('\n\n'):
+        paragraphs = remuneration_clause.split('\n\n')
+        for paragraph in paragraphs:
             if paragraph.strip():
-                elements.append(Paragraph(paragraph, styles['Normal']))
+                # Si c'est un titre d'article (commence par "ARTICLE")
+                if paragraph.startswith("ARTICLE"):
+                    elements.append(Paragraph(paragraph, styles['Article']))
+                # Si c'est un sous-titre (comme "6.1 Cession à titre gratuit")
+                elif paragraph.strip()[0].isdigit() and "." in paragraph.strip()[:3] and len(paragraph.split("\n")[0]) < 50:
+                    elements.append(Paragraph(paragraph, styles['Sous-article']))
+                else:
+                    elements.append(Paragraph(paragraph, styles['Normal']))
                 elements.append(Spacer(1, 6))
         article_num += 1
         
         # 9. Article - Garanties
         warranties_clause = ContractTemplates.get_warranties_clause(article_num, contract_type)
-        for paragraph in warranties_clause.split('\n\n'):
+        paragraphs = warranties_clause.split('\n\n')
+        for paragraph in paragraphs:
             if paragraph.strip():
-                elements.append(Paragraph(paragraph, styles['Normal']))
+                # Si c'est un titre d'article (commence par "ARTICLE")
+                if paragraph.startswith("ARTICLE"):
+                    elements.append(Paragraph(paragraph, styles['Article']))
+                # Si c'est un sous-titre (comme "7.1 Garanties de l'Auteur")
+                elif paragraph.strip()[0].isdigit() and "." in paragraph.strip()[:3] and len(paragraph.split("\n")[0]) < 50:
+                    elements.append(Paragraph(paragraph, styles['Sous-article']))
+                else:
+                    elements.append(Paragraph(paragraph, styles['Normal']))
                 elements.append(Spacer(1, 6))
         article_num += 1
         
         # 10. Article - Résiliation
         termination_clause = ContractTemplates.get_termination_clause(article_num)
-        for paragraph in termination_clause.split('\n\n'):
+        paragraphs = termination_clause.split('\n\n')
+        for paragraph in paragraphs:
             if paragraph.strip():
-                elements.append(Paragraph(paragraph, styles['Normal']))
+                # Si c'est un titre d'article (commence par "ARTICLE")
+                if paragraph.startswith("ARTICLE"):
+                    elements.append(Paragraph(paragraph, styles['Article']))
+                # Si c'est un sous-titre (comme "8.1 Résiliation pour inexécution")
+                elif paragraph.strip()[0].isdigit() and "." in paragraph.strip()[:3] and len(paragraph.split("\n")[0]) < 50:
+                    elements.append(Paragraph(paragraph, styles['Sous-article']))
+                else:
+                    elements.append(Paragraph(paragraph, styles['Normal']))
                 elements.append(Spacer(1, 6))
         article_num += 1
         
-        # 11. Article - Loi applicable
+        # 11. Article - Dispositions diverses
+        other_clauses = ContractTemplates.get_other_clauses(article_num, contract_type)
+        paragraphs = other_clauses.split('\n\n')
+        for paragraph in paragraphs:
+            if paragraph.strip():
+                # Si c'est un titre d'article (commence par "ARTICLE")
+                if paragraph.startswith("ARTICLE"):
+                    elements.append(Paragraph(paragraph, styles['Article']))
+                # Si c'est un sous-titre (comme "9.1 Clause de non-dénigrement")
+                elif paragraph.strip()[0].isdigit() and "." in paragraph.strip()[:3] and len(paragraph.split("\n")[0]) < 50:
+                    elements.append(Paragraph(paragraph, styles['Sous-article']))
+                else:
+                    elements.append(Paragraph(paragraph, styles['Normal']))
+                elements.append(Spacer(1, 6))
+        article_num += 1
+        
+        # 12. Article - Loi applicable
         law_clause = ContractTemplates.get_applicable_law_clause(article_num)
-        for paragraph in law_clause.split('\n\n'):
+        paragraphs = law_clause.split('\n\n')
+        for paragraph in paragraphs:
             if paragraph.strip():
-                elements.append(Paragraph(paragraph, styles['Normal']))
+                # Si c'est un titre d'article (commence par "ARTICLE")
+                if paragraph.startswith("ARTICLE"):
+                    elements.append(Paragraph(paragraph, styles['Article']))
+                # Si c'est un sous-titre (comme "10.1 Loi applicable")
+                elif paragraph.strip()[0].isdigit() and "." in paragraph.strip()[:3] and len(paragraph.split("\n")[0]) < 50:
+                    elements.append(Paragraph(paragraph, styles['Sous-article']))
+                else:
+                    elements.append(Paragraph(paragraph, styles['Normal']))
                 elements.append(Spacer(1, 6))
         
-        # 12. Signatures
-        elements.append(Spacer(1, 20))
+        # 13. Signatures
+        elements.append(Spacer(1, 30))
         signatures = ContractTemplates.get_signatures_template(contract_type)
-        for paragraph in signatures.split('\n\n'):
+        paragraphs = signatures.split('\n\n')
+        for paragraph in paragraphs:
             if paragraph.strip():
                 elements.append(Paragraph(paragraph, styles['Normal']))
                 elements.append(Spacer(1, 8))
         
-        return elements
+        return elements