pierreguillou commited on
Commit
4483a9e
·
verified ·
1 Parent(s): 6e159ac

Update helpers/rapport_generator.py

Browse files
Files changed (1) hide show
  1. helpers/rapport_generator.py +4 -4
helpers/rapport_generator.py CHANGED
@@ -11,11 +11,12 @@ from typing import Dict, Any, List, Union # Ajout des imports typing nécessair
11
  import logging
12
 
13
  class RapportGenerator:
14
- def __init__(self, json_file: str):
15
  self.doc = Document()
16
  self.logger = logging.getLogger(__name__)
17
  logging.basicConfig(level=logging.INFO)
18
  self.json_data = self.load_json(json_file)
 
19
  self._setup_document()
20
 
21
  def load_json(self, json_file: str) -> Dict:
@@ -379,9 +380,8 @@ class RapportGenerator:
379
  self._process_special_fields("MISSION", self.json_data["MISSION"])
380
 
381
  # Sauvegarde du document
382
- output_file = "rapport_extraction.docx"
383
- self.doc.save(output_file)
384
- self.logger.info(f"Rapport généré avec succès: {output_file}")
385
 
386
  return output_file
387
 
 
11
  import logging
12
 
13
  class RapportGenerator:
14
+ def __init__(self, json_file: str, docx_path: str):
15
  self.doc = Document()
16
  self.logger = logging.getLogger(__name__)
17
  logging.basicConfig(level=logging.INFO)
18
  self.json_data = self.load_json(json_file)
19
+ self.docx_path = docx_path
20
  self._setup_document()
21
 
22
  def load_json(self, json_file: str) -> Dict:
 
380
  self._process_special_fields("MISSION", self.json_data["MISSION"])
381
 
382
  # Sauvegarde du document
383
+ self.doc.save(self.docx_path)
384
+ self.logger.info(f"Rapport généré avec succès: {self.docx_path}")
 
385
 
386
  return output_file
387