Update app.py

app.py

@@ -147,17 +147,16 @@ def create_heatmap(frame_scores, current_frame):
     
     return tmp_file.name
 
-def load_sample_frame(video_path, frame_number=None):
+def load_sample_frame(video_path, target_frame=87, original_fps=30, processing_fps=1):
     cap = cv2.VideoCapture(video_path)
     if not cap.isOpened():
         return None
     
-    total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
+    # Calculate the corresponding frame number in the original video
+    original_frame_number = int(target_frame * (original_fps / processing_fps))
     
-    if frame_number is None or frame_number >= total_frames:
-        frame_number = total_frames // 2  # Default to middle frame
-    
-    cap.set(cv2.CAP_PROP_POS_FRAMES, frame_number)
+    # Set the frame position
+    cap.set(cv2.CAP_PROP_POS_FRAMES, original_frame_number)
     
     ret, frame = cap.read()
     cap.release()
@@ -183,7 +182,10 @@ def gradio_app():
         heatmap_output = gr.Image(label="Detection Heatmap")
         output_image = gr.Image(label="Processed Frame")
         error_output = gr.Textbox(label="Error Messages", visible=False)
-        sample_video_frame = gr.Image(value=load_sample_frame("Drone Video of African Wildlife Wild Botswan.mp4", 100), label="Drone Video of African Wildlife Wild Botswan by wildimagesonline.com - Sample Video Frame")
+        sample_video_frame = gr.Image(
+            value=load_sample_frame("Drone Video of African Wildlife Wild Botswan.mp4", target_frame=87),
+            label="Drone Video of African Wildlife Wild Botswan by wildimagesonline.com - Sample Video Frame (Frame 87 at 1 FPS)"
+        )
         use_sample_button = gr.Button("Use Sample Video")
         progress_bar = gr.Progress()