Update app

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import os
-import glob
 import validators
+from imutils import paths
 from config import *
 from download_video import download_video
 from bg_modeling import capture_slides_bg_modeling
@@ -48,11 +48,11 @@ def process(
         output_dir_path, hash_size, hash_func, hash_queue_len, diff_threshold
     )
 
-    pdf_path = convert_slides_to_pdf(video_path, output_dir_path)
+    pdf_path = convert_slides_to_pdf(output_dir_path)
 
     # Remove unneccessary files
     os.remove(video_path)
-    for image_path in glob.glob(f"{output_dir_path}/*.jpg"):
+    for image_path in paths.list_images(output_dir_path):
         os.remove(image_path)
     return pdf_path
 

config.py

@@ -4,7 +4,7 @@ import imagehash
 
 DOWNLOAD_DIR = "downloads"
 
-FRAME_BUFFER_HISTORY = 15  # Length of the frame buffer history to model background.
+FRAME_BUFFER_HISTORY = 10  # Length of the frame buffer history to model background.
 DEC_THRESH = (
     0.75  # Threshold value, above which it is marked foreground, else background.
 )

frame_differencing.py

@@ -40,7 +40,7 @@ def capture_slides_frame_diff(
 
         screenshots_count += 1
 
-        filename = f"{screenshots_count:03}.lpg"
+        filename = f"{screenshots_count:03}.jpg"
         out_file_path = os.path.join(output_dir_path, filename)
         print(f"Saving file at: {out_file_path}")
 
@@ -68,7 +68,6 @@ def capture_slides_frame_diff(
 
             if p_non_zero >= MIN_PERCENT_THRESH and not capture_frame:
                 capture_frame = True
-
             elif capture_frame:
                 frame_elapsed += 1
 
@@ -78,11 +77,11 @@ def capture_slides_frame_diff(
 
                 screenshots_count += 1
 
-                filename = f"{screenshots_count:03}.png"
+                filename = f"{screenshots_count:03}.jpg"
                 out_file_path = os.path.join(output_dir_path, filename)
                 print(f"Saving file at: {out_file_path}")
 
-                cv2.imwrite(out_file_path, frame)
+                cv2.imwrite(out_file_path, frame, [cv2.IMWRITE_JPEG_QUALITY, 75])
 
         prev_frame = curr_frame
 

requirements.txt

@@ -4,6 +4,7 @@ Pillow
 scipy
 six
 ImageHash
+imutils
 img2pdf
 pytube
 validators

utils.py

@@ -3,7 +3,7 @@ import re
 import cv2
 import shutil
 import img2pdf
-import glob
+from imutils import paths
 
 # PIL can also be used to convert the image set into PDFs.
 # However, using PIL requires opening each of the images in the set.
@@ -46,17 +46,21 @@ def create_output_directory(video_path, output_path, type_bgsub):
     return output_dir_path
 
 
-def convert_slides_to_pdf(video_path, output_path):
-    pdf_file_name = video_path.rsplit(os.sep)[-1].split(".")[0] + ".pdf"
-    output_pdf_path = os.path.join(output_path, pdf_file_name)
+def convert_slides_to_pdf(img_dir, output_path=None):
+    if not os.path.isdir(img_dir):
+        print("The image directory doesn't exist!")
+        return
+    
+    if output_path == None:
+        pdf_file_name = os.path.basename(img_dir) + ".pdf"
+        output_path = os.path.join(img_dir, pdf_file_name)
+        print("Output PDF Path:", output_path)
 
-    print("Output PDF Path:", output_pdf_path)
     print("Converting captured slide images to PDF...")
-
-    with open(output_pdf_path, "wb") as f:
-        f.write(img2pdf.convert(sorted(glob.glob(f"{output_path}/*.jpg"))))
+    with open(output_path, "wb") as f:
+        f.write(img2pdf.convert(sorted(paths.list_images(img_dir))))
 
     print("PDF Created!")
     print("***" * 10, "\n")
 
-    return output_pdf_path
+    return output_path

video_2_slides.py

@@ -81,7 +81,7 @@ if __name__ == "__main__":
         )
         queue_len = HASH_BUFFER_HISTORY
 
-    video_path = args.video_file_path
+    video_path = args.video_path
     output_dir_path = args.out_dir
     type_bg_sub = args.type
     temp_file = False
@@ -128,7 +128,7 @@ if __name__ == "__main__":
         )
 
     if args.convert_to_pdf:
-        convert_slides_to_pdf(video_path, output_dir_path)
+        convert_slides_to_pdf(output_dir_path)
 
     # if temp_file:
     #     os.remove(video_path)