Update app.py

app.py

@@ -22,7 +22,7 @@ from typing import Dict, Any, List, Union  # Ajout des imports typing nécessair
 import logging
 
 # helpers functions
-from helpers.rapport_generator import RapportGenerator
+from helpers.rapport_generator import *
 from helpers.text_extraction import *
 from helpers.gemini_functions import *
 
@@ -70,11 +70,14 @@ def process_pdf(pdf_file):
             json.dump(extracted_data, f, ensure_ascii=False, indent=2)
 
         # Generate DOCX report
-        # docx_path = os.path.join(temp_dir, "rapport_extraction.docx")
-        # generator = RapportGenerator(json_path, docx_path)
-        # generator.generate_report()
+        try:
+            docx_path = os.path.join(temp_dir, "rapport_extraction.docx")
+            generator = RapportGenerator(json_path, docx_path)
+            generator.generate_report()
+        except Exception as e:
+            raise gr.Error(f"Error processing rapport: {str(e)}")
 
-        return text_file_path, zip_path, json_path #, docx_path
+        return text_file_path, zip_path, json_path, docx_path
 
     except Exception as e:
         raise gr.Error(f"Error processing PDF: {str(e)}")
@@ -105,7 +108,7 @@ demo = gr.Interface(
         gr.File(label="Texte extrait (TXT)"),
         gr.File(label="Images annotées (ZIP)"),
         gr.File(label="Données extraites (JSON)"),
-        # gr.File(label="Rapport généré (DOCX)")  # Nouvelle sortie
+        gr.File(label="Rapport généré (DOCX)") 
     ],
     title="Extraction de texte PDF et création d'un rapport DOCX",
     description="""