Update app.py

app.py

@@ -133,6 +133,25 @@ def load_frames(video_file, max_num_frames=64, long_edge=448):
         imgs.append(img)
 
     return imgs, frame_timestamps
+
+def pdf_to_multi_image(local_pdf):
+    import fitz, io
+    doc_ = fitz.open(local_pdf)
+    
+    all_input_images = []
+    
+    for page_num in range(len(doc_)):
+        page = doc.load_page(page_num)
+        
+        pixmap = page.get_pixmap(dpi=96)
+        
+        image = Image.open(io.BytesIO(pixmap.tobytes("png")))
+        
+        all_input_images.append(image)
+
+    
+    doc_.close()
+    return all_input_images
     
 @wrap_gen_fn
 @spaces.GPU(duration=30)