Update app.py

app.py

@@ -86,10 +86,13 @@ def extract_frames(video_path, output_folder, fps):
         f'{output_folder}/frame_%04d.jpg'
     ]
     try:
-        subprocess.run(command, check=True, capture_output=True, text=True)
+        result = subprocess.run(command, check=True, capture_output=True, text=True)
+        print(f"FFmpeg stdout: {result.stdout}")
+        print(f"FFmpeg stderr: {result.stderr}")
     except subprocess.CalledProcessError as e:
         print(f"Error extracting frames: {e}")
-        print(f"FFmpeg output: {e.output}")
+        print(f"FFmpeg stdout: {e.stdout}")
+        print(f"FFmpeg stderr: {e.stderr}")
         raise
 
 import fractions
@@ -111,13 +114,45 @@ def extract_and_align_faces_from_video(video_path, aligned_faces_folder, desired
         '-of', 'csv=p=0',
         video_path
     ]
-    ffprobe_output = subprocess.check_output(ffprobe_command, universal_newlines=True).strip().split(',')
-    frame_count = int(ffprobe_output[0])
-    
-    # Convert fractional frame rate to float
-    frac = fractions.Fraction(ffprobe_output[1])
-    original_fps = float(frac.numerator) / float(frac.denominator)
-    
+    try:
+        ffprobe_output = subprocess.check_output(ffprobe_command, universal_newlines=True).strip().split(',')
+        print(f"FFprobe output: {ffprobe_output}")  # Debugging output
+        
+        if len(ffprobe_output) != 2:
+            raise ValueError(f"Unexpected FFprobe output format: {ffprobe_output}")
+        
+        frame_count = ffprobe_output[0]
+        frame_rate = ffprobe_output[1]
+        
+        print(f"Frame count (raw): {frame_count}")  # Debugging output
+        print(f"Frame rate (raw): {frame_rate}")  # Debugging output
+        
+        # Convert frame count to int
+        try:
+            frame_count = int(frame_count)
+        except ValueError:
+            print(f"Warning: Could not convert frame count '{frame_count}' to int. Using fallback method.")
+            frame_count = len([f for f in os.listdir(frames_folder) if f.endswith('.jpg')])
+        
+        # Convert fractional frame rate to float
+        try:
+            frac = fractions.Fraction(frame_rate)
+            original_fps = float(frac.numerator) / float(frac.denominator)
+        except (ValueError, ZeroDivisionError):
+            print(f"Warning: Could not convert frame rate '{frame_rate}' to float. Using fallback method.")
+            # Fallback: Count frames and divide by video duration
+            frame_count = len([f for f in os.listdir(frames_folder) if f.endswith('.jpg')])
+            duration_command = ['ffprobe', '-v', 'error', '-show_entries', 'format=duration', '-of', 'default=noprint_wrappers=1:nokey=1', video_path]
+            duration = float(subprocess.check_output(duration_command, universal_newlines=True).strip())
+            original_fps = frame_count / duration
+        
+    except subprocess.CalledProcessError as e:
+        print(f"Error running FFprobe: {e}")
+        raise
+    except Exception as e:
+        print(f"Unexpected error processing video info: {e}")
+        raise
+
     print(f"Total frames: {frame_count}, Original FPS: {original_fps}, Desired FPS: {desired_fps}")
 
     embeddings_by_frame = {}