Update app.py

app.py

@@ -14,9 +14,6 @@ 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
 
@@ -33,15 +30,10 @@ def stitch_images_vertically(images: List[Image.Image]) -> Image.Image:
     return stitched_image
 
 def stitch_images_in_grid(images: List[Image.Image], num_columns: int) -> Image.Image:
-    """
-    Stitches images into a grid with a specified number of columns.
-    """
     if not images:
         return None
 
     num_images = len(images)
-    num_rows = math.ceil(num_images / num_columns)
-    
     columns = [images[i::num_columns] for i in range(num_columns)]
     
     stitched_columns = [stitch_images_vertically(col) for col in columns]
@@ -59,9 +51,7 @@ def stitch_images_in_grid(images: List[Image.Image], num_columns: int) -> Image.
             
     return grid_image
 
-    """
-    Processes a PDF, converts it to images, and stitches them into a single image or a grid.
-    """
+def process_pdf(pdf_file, pdf_url, dpi, num_columns, progress=gr.Progress()):
     pdf_input_source = None
     is_bytes = False
     source_name = "document"