Update app.py

app.py

@@ -4,6 +4,7 @@ import requests
 import io
 import os
 import tempfile
+import sys
 
 def compress_pdf(input_file, url, quality):
     if input_file is None and (url is None or url.strip() == ""):
@@ -26,14 +27,19 @@ def compress_pdf(input_file, url, quality):
         reader = PyPDF2.PdfReader(pdf_content)
         writer = PyPDF2.PdfWriter()
 
-        for page in reader.pages:
-            if quality == "High":
-                page.compress_content_streams()  # This is the lowest compression
-            elif quality == "Medium":
-                page.compress_content_streams(level=2)
-            else:  # Low quality
-                page.compress_content_streams(level=3)
-            writer.add_page(page)
+        for i, page in enumerate(reader.pages):
+            try:
+                if quality == "High":
+                    page.compress_content_streams()  # This is the lowest compression
+                elif quality == "Medium":
+                    page.compress_content_streams(level=2)
+                else:  # Low quality
+                    page.compress_content_streams(level=3)
+                writer.add_page(page)
+            except RecursionError:
+                return None, f"Error: PDF structure too complex. Compression stopped at page {i+1}."
+            except Exception as e:
+                return None, f"Error compressing page {i+1}: {str(e)}"
 
         with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as temp_file:
             writer.write(temp_file)
@@ -44,6 +50,9 @@ def compress_pdf(input_file, url, quality):
         return None, f"Error compressing PDF: {str(e)}"
 
 def process_and_compress(input_file, url, quality):
+    # Increase recursion limit
+    sys.setrecursionlimit(10000)
+    
     output_file, message = compress_pdf(input_file, url, quality)
     if output_file:
         return output_file, message