Back to temp PDf storage

app.py

@@ -2,7 +2,9 @@ import fitz  # PyMuPDF
 import gradio as gr
 import zipfile
 import io
+import os
 from datetime import datetime
+import tempfile
 
 # Helper function to find color areas
 def find_color_areas(page, target_color, tolerance=30):
@@ -139,12 +141,15 @@ def process_pdf_files(input_pdfs, selected_color_comment_indices, tolerance, cus
     # Move pointer to the beginning of the ZIP file BytesIO object
     zip_in_memory.seek(0)
 
-    # Set the filename for the ZIP file
+    # Generate the current date string
     current_date = datetime.now().strftime("%Y-%m-%d")
-    zip_filename = f"CoordinationPDFS_{current_date}.zip"
+
+    # Create a named temporary file to store the ZIP
+    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=f"_CoordinationPDFS_{current_date}.zip")
+    with open(temp_file.name, 'wb') as tmp_file:
+        tmp_file.write(zip_in_memory.read())
     
-    # Return the in-memory ZIP file and its filename for Gradio to download
-    return (zip_filename, zip_in_memory)
+    return temp_file.name  # Return the path to the temp file for Gradio to handle the download
 
 # Define the Gradio interface
 interface = gr.Interface(