Projet : extraction de factures vers un fichier vérifié

from pydantic import BaseModel, Field

class Ligne(BaseModel):
    libelle: str
    montant_ht: float

class Facture(BaseModel):
    fournisseur: str
    numero: str
    date: str = Field(description="Date au format AAAA-MM-JJ ; chaîne vide si la facture n'en porte pas")
    lignes: list[Ligne]
    total_ht: float
    taux_tva: float = Field(description="Taux de TVA en pour cent, par exemple 20 ou 5.5")
    total_tva: float
    total_ttc: float

def controler(f, deja_vues, ecart=0.01):
    """Les règles d'une facture recevable ; le modèle lit, le programme vérifie."""
    regles = [
        ("date présente", f.date != ""),
        ("hors taxe = somme des lignes", abs(sum(l.montant_ht for l in f.lignes) - f.total_ht) <= ecart),
        ("TVA = hors taxe × taux", abs(f.total_ht * f.taux_tva / 100 - f.total_tva) <= ecart),
        ("total = hors taxe + TVA", abs(f.total_ht + f.total_tva - f.total_ttc) <= ecart),
        ("pas déjà reçue", (f.fournisseur, f.numero) not in deja_vues),
    ]
    return [nom for nom, passe in regles if not passe]

import json
import re
from pathlib import Path

from dotenv import load_dotenv
load_dotenv()

from openai import OpenAI

from facture import Facture

client = OpenAI()
texte = Path("lot-de-factures.md").read_text(encoding="utf-8")
factures = re.split(r"\n## Facture \d+\n", texte)[1:]   # une facture par section

lectures = []
for t in factures:
    r = client.responses.parse(
        model="gpt-6-luna",
        input=[
            {"role": "system", "content": "Extrais les informations de la facture. "
             "Recopie les montants tels qu'ils sont écrits ; ne calcule rien."},
            {"role": "user", "content": t},
        ],
        text_format=Facture,
    )
    lectures.append(r.output_parsed.model_dump())

Path("lectures.json").write_text(json.dumps(lectures, ensure_ascii=False, indent=1), encoding="utf-8")
print(len(lectures), "factures lues")

import csv
import json

from facture import Facture, controler

lectures = [Facture(**x) for x in json.load(open("lectures.json", encoding="utf-8"))]
deja_vues, acceptees, de_cote = set(), [], []
for n, f in enumerate(lectures, 1):
    echecs = controler(f, deja_vues)
    deja_vues.add((f.fournisseur, f.numero))
    (de_cote if echecs else acceptees).append((n, f, echecs))
    print(f"Facture {n} ({f.numero}) :", "acceptée" if not echecs else "de côté, règle en échec : " + " ; ".join(echecs))

with open("acceptees.csv", "w", newline="", encoding="utf-8") as sortie:
    w = csv.writer(sortie, delimiter=";")
    w.writerow(["fournisseur", "numero", "date", "total_ht", "total_tva", "total_ttc"])
    for _, f, _ in acceptees:
        w.writerow([f.fournisseur, f.numero, f.date, f.total_ht, f.total_tva, f.total_ttc])
print(f"{len(acceptees)} acceptées, écrites dans acceptees.csv ; {len(de_cote)} mises de côté")