Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+import io
 import time
 from pathlib import Path
 from typing import List, Dict
@@ -7,6 +8,7 @@ import streamlit as st
 import pandas as pd
 import json
 import yaml
+import zipfile
 from docling.document_converter import DocumentConverter, PdfFormatOption
 from docling.datamodel.base_models import InputFormat
 from docling.datamodel.document import ConversionStatus
@@ -244,6 +246,16 @@ def display_multimodal_result(file_path: Path):
                         except Exception as e:
                             st.error(f"Erreur d'affichage CSV : {str(e)}")
 
+def create_zip_buffer(directory: Path) -> bytes:
+    buffer = io.BytesIO()
+    with zipfile.ZipFile(buffer, 'w', zipfile.ZIP_DEFLATED) as zipf:
+        for root, _, files in os.walk(directory):
+            for file in files:
+                file_path = Path(root) / file
+                zipf.write(file_path, arcname=file_path.relative_to(directory.parent))
+    buffer.seek(0)
+    return buffer.getvalue()
+
 # Interface utilisateur
 def main():
     st.title("📊🦆 Docling Document Converter")
@@ -276,12 +288,18 @@ def main():
             results = process_files(uploaded_files, config)
             display_results(results)
             st.success("✅ Conversion terminée avec succès !")
-            st.download_button("Télécharger tous les résultats", 
-                             data=os.path.join(OUTPUT_DIR),
-                             file_name="conversion_results.zip",
-                             mime="application/zip")
-        else:
-            st.error("⚠️ Veuillez télécharger au moins un fichier")
+            
+            # Création du buffer ZIP
+            try:
+                zip_buffer = create_zip_buffer(OUTPUT_DIR)
+                st.download_button(
+                    label="📥 Télécharger tous les résultats",
+                    data=zip_buffer,
+                    file_name="conversion_results.zip",
+                    mime="application/zip"
+                )
+            except Exception as e:
+                st.error(f"Erreur lors de la création du ZIP : {str(e)}")
 
 if __name__ == "__main__":
     main()