Update app.py

app.py

@@ -147,10 +147,18 @@ def create_heatmap(frame_scores, current_frame):
     
     return tmp_file.name
 
-def load_sample_frame(video_path):
+def load_sample_frame(video_path, frame_number=None):
     cap = cv2.VideoCapture(video_path)
     if not cap.isOpened():
         return None
+    
+    total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
+    
+    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)
+    
     ret, frame = cap.read()
     cap.release()
     if not ret:
@@ -175,7 +183,7 @@ 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"), 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", 100), label="Drone Video of African Wildlife Wild Botswan by wildimagesonline.com - Sample Video Frame")
         use_sample_button = gr.Button("Use Sample Video")
         progress_bar = gr.Progress()