Update app.py

app.py

@@ -30,7 +30,7 @@ if uploaded_file:
     # Scene Detection & Frame Extraction
     st.write("🔍 Detecting scene changes and extracting key frames...")
 
-    def extract_key_frames(video_path, output_folder="frames", frames_per_scene=3):
+    def extract_key_frames(video_path, output_folder="frames"):
         os.makedirs(output_folder, exist_ok=True)
         video = open_video(video_path)
         scene_manager = SceneManager()
@@ -39,16 +39,15 @@ if uploaded_file:
         scenes = scene_manager.get_scene_list()
         cap = cv2.VideoCapture(video_path)
         for i, (start, end) in enumerate(scenes):
-            start_frame = start.get_frames()
-            end_frame = end.get_frames()
-            step = (end_frame - start_frame) // (frames_per_scene + 1)
-            for j in range(frames_per_scene):
-                frame_time = start_frame + step * (j + 1)
-                cap.set(cv2.CAP_PROP_POS_FRAMES, frame_time)
-                ret, frame = cap.read()
-                if ret:
-                    frame_path = os.path.join(output_folder, f"scene_{i+1}_frame{j+1}.jpg")
-                    cv2.imwrite(frame_path, frame)
+            frame_time = start.get_frames()  # Extract frame at scene start
+            cap.set(cv2.CAP_PROP_POS_FRAMES, frame_time)
+            ret, frame = cap.read()
+            
+            if ret:
+                frame_path = os.path.join(output_folder, f"scene_{i+1}.jpg")
+                cv2.imwrite(frame_path, frame)
+                print(f"Saved: {frame_path}")
+
         cap.release()
 
     extract_key_frames(video_path)