Update app.py

app.py

@@ -12,6 +12,7 @@ import os
 import datetime
 from urllib.parse import urlparse
 from bs4 import BeautifulSoup
+import tempfile
 
 # Configure logging
 logging.basicConfig(level=logging.INFO, 
@@ -212,26 +213,26 @@ def process_urls(url_input, bulk_toggle, action_radio, max_urls, crawl_depth, pr
         # Update progress
         progress((idx + 1) / total_urls)
     
-    # Create ZIP file in memory
-    memory_file = io.BytesIO()
-    with zipfile.ZipFile(memory_file, 'w', zipfile.ZIP_DEFLATED) as zipf:
-        # Add screenshots to ZIP
-        for screenshot_url, screenshot_data in screenshots:
-            sanitized_screenshot_url = sanitize_filename(screenshot_url)
-            filename = f"{sanitized_screenshot_url}.png"
-            zipf.writestr(filename, screenshot_data)
+    # Create a temporary file to store the ZIP
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".zip") as tmp_file:
+        with zipfile.ZipFile(tmp_file, 'w', zipfile.ZIP_DEFLATED) as zipf:
+            # Add screenshots to ZIP
+            for screenshot_url, screenshot_data in screenshots:
+                sanitized_screenshot_url = sanitize_filename(screenshot_url)
+                filename = f"{sanitized_screenshot_url}.png"
+                zipf.writestr(filename, screenshot_data)
+            
+            # Add scraped data and changes log to ZIP
+            if scraped_data:
+                data_to_save = {
+                    'scraped_data': scraped_data,
+                    'changes_log': changes_log,
+                    'timestamp': datetime.datetime.now().isoformat()
+                }
+                zipf.writestr('data.json', json.dumps(data_to_save, indent=2))
         
-        # Add scraped data and changes log to ZIP
-        if scraped_data:
-            data_to_save = {
-                'scraped_data': scraped_data,
-                'changes_log': changes_log,
-                'timestamp': datetime.datetime.now().isoformat()
-            }
-            zipf.writestr('data.json', json.dumps(data_to_save, indent=2))
-    
-    # Prepare return values
-    memory_file.seek(0)
+        # Get the path to the temporary file
+        zip_file_path = tmp_file.name
     
     # Prepare display data
     display_data = {
@@ -240,8 +241,8 @@ def process_urls(url_input, bulk_toggle, action_radio, max_urls, crawl_depth, pr
         'changes_detected': changes_log
     }
     
-    # Return ZIP bytes and display data
-    return memory_file, json.dumps(display_data, indent=2)
+    # Return the path to the temporary ZIP file and display data
+    return zip_file_path, json.dumps(display_data, indent=2)
     
 def create_interface():
     """Create the Gradio interface."""