Update app.py

app.py

@@ -13,21 +13,24 @@ def compress_pdf(input_file, url, strength):
     if input_file is not None and url and url.strip() != "":
         return None, "Please provide either a file or a URL, not both."
     
-    if url and url.strip() != "":
-        try:
+    try:
+        if url and url.strip() != "":
             response = requests.get(url)
             response.raise_for_status()
             pdf_content = io.BytesIO(response.content)
-        except requests.RequestException as e:
-            return None, f"Error downloading PDF: {str(e)}"
-    else:
-        pdf_content = input_file
-
-    try:
-        # Get the initial file size
-        pdf_content.seek(0, os.SEEK_END)
-        initial_size = pdf_content.tell()
-        pdf_content.seek(0)
+            initial_size = len(response.content)
+        else:
+            if hasattr(input_file, 'name'):
+                # If input_file is a file path
+                with open(input_file.name, 'rb') as file:
+                    pdf_content = io.BytesIO(file.read())
+                initial_size = os.path.getsize(input_file.name)
+            else:
+                # If input_file is file-like object
+                pdf_content = io.BytesIO(input_file.read())
+                pdf_content.seek(0, io.SEEK_END)
+                initial_size = pdf_content.tell()
+                pdf_content.seek(0)
 
         reader = PyPDF2.PdfReader(pdf_content)
         writer = PyPDF2.PdfWriter()
@@ -85,6 +88,8 @@ def compress_pdf(input_file, url, strength):
     except Exception as e:
         return None, f"Error compressing PDF: {str(e)}"
 
+# The rest of the code remains the same
+
 def process_and_compress(input_file, url, strength):
     sys.setrecursionlimit(10000)