Update app.py

app.py

@@ -13,6 +13,9 @@ logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(
 logger = logging.getLogger(__name__)
 
 def stitch_images_vertically(images: List[Image.Image]) -> Image.Image:
+    """
+    Stitches a list of images together vertically.
+    """
     if not images:
         return None
 
@@ -28,7 +31,11 @@ def stitch_images_vertically(images: List[Image.Image]) -> Image.Image:
         
     return stitched_image
 
-def process_pdf(pdf_file, pdf_url, progress=gr.Progress()):
+def process_pdf(pdf_file, pdf_url, dpi, progress=gr.Progress()):
+    """
+    Processes a PDF from a file upload or a URL, converts it to images with a specified DPI,
+    and stitches them together vertically.
+    """
     pdf_input_source = None
     is_bytes = False
     source_name = "document"
@@ -54,11 +61,12 @@ def process_pdf(pdf_file, pdf_url, progress=gr.Progress()):
         raise gr.Error("Please upload a PDF file or provide a valid URL.")
 
     progress(0.3, desc="Converting PDF pages to images...")
+    logger.info(f"Using DPI: {dpi}")
     try:
         if is_bytes:
-            images = convert_from_bytes(pdf_input_source, dpi=200)
+            images = convert_from_bytes(pdf_input_source, dpi=dpi)
         else:
-            images = convert_from_path(pdf_input_source, dpi=200)
+            images = convert_from_path(pdf_input_source, dpi=dpi)
     except (PDFInfoNotInstalledError, FileNotFoundError):
         raise gr.Error("Server configuration error: Poppler dependency is missing.")
     except (PDFPageCountError, Exception) as e:
@@ -107,6 +115,15 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                         label="PDF URL",
                         placeholder="e.g., https://arxiv.org/pdf/1706.03762.pdf"
                     )
+
+            dpi_slider = gr.Slider(
+                minimum=100,
+                maximum=600,
+                step=50,
+                value=200,
+                label="Image Resolution (DPI)",
+                info="Higher DPI results in a clearer image but increases processing time and file size."
+            )
             
             submit_btn = gr.Button("Stitch PDF Pages", variant="primary")
 
@@ -125,7 +142,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
     submit_btn.click(
         fn=process_pdf,
-        inputs=[pdf_file_input, pdf_url_input],
+        inputs=[pdf_file_input, pdf_url_input, dpi_slider],
         outputs=[output_image_preview, output_image_download]
     )