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)