Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-import fitz
+import fitz  # PyMuPDF
 import cv2
 from pdf2image import convert_from_path
 import pytesseract
@@ -78,6 +78,8 @@ def generate_comparison_pdf(original_pdf, edited_pdf):
 
 # Gradio interface function
 def pdf_comparison(original_pdf, edited_pdf):
+    if original_pdf.size > 50 * 1024 * 1024 or edited_pdf.size > 50 * 1024 * 1024:  # 50 MB limit
+        return "Error: File size exceeds 50 MB. Please upload smaller files."
     result_path = generate_comparison_pdf(original_pdf.name, edited_pdf.name)
     return result_path
 
@@ -91,5 +93,4 @@ interface = gr.Interface(
 )
 
 if __name__ == "__main__":
-    interface.launch(timeout=300)
-
+    interface.launch()