Update app.py

app.py

@@ -114,11 +114,10 @@ def process_video_for_canny(video, width, height):
     """
     print("Processing video for canny control...")
     canny_video = []
-    with imageio.get_reader(video) as reader:
-        # detect_resolution = max(video[0].size[0],video[0].size[1])
-        image_resolution = max(width, height)
-        for frame in reader: 
-            canny_video.append(canny_processor(frame, low_threshold=50, high_threshold=200, detect_resolution=image_resolution, image_resolution=image_resolution))
+    detect_resolution = max(video[0].size[0],video[0].size[1])
+    image_resolution = max(width, height)
+    for frame in video: 
+        canny_video.append(canny_processor(frame, low_threshold=50, high_threshold=200, detect_resolution=detect_resolution, image_resolution=image_resolution))
      
     return canny_video
 
@@ -132,7 +131,7 @@ def process_input_video(reference_video, width, height,
     
     try:
         # Load video into a list of PIL images
-        #video = load_video(reference_video)
+        video = load_video(reference_video)
         
         # Process video for canny edges
         processed_video = process_video_for_canny(reference_video, width, height)
@@ -150,7 +149,7 @@ def process_input_video(reference_video, width, height,
 
 def process_video_for_control(reference_video, control_type, width, height):
     """Process video based on the selected control type - now only used for non-canny types"""
-    # video = load_video(reference_video)
+    video = load_video(reference_video)
     
     if control_type == "canny":
         # This should not be called for canny since it's pre-processed