Update gully_drs_core/replay_utils.py

gully_drs_core/replay_utils.py

@@ -1,49 +1,49 @@
 import cv2
-import streamlit as st
 import tempfile
 import numpy as np
 
-def save_replay_clip(video_path, ball_path, stump_zone):
+def save_replay_clip(video_path, ball_path_3d, stump_zone, impact_frame=None):
     cap = cv2.VideoCapture(video_path)
-    fps = cap.get(cv2.CAP_PROP_FPS)
+    fps = int(cap.get(cv2.CAP_PROP_FPS))
     frame_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
     frame_height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
-
+    
     temp_output = tempfile.NamedTemporaryFile(delete=False, suffix='.mp4')
-    out = cv2.VideoWriter(temp_output.name, cv2.VideoWriter_fourcc(*'mp4v'), fps, (frame_width, frame_height))
-
-    zoomed_frame_index = -1
-    if ball_path:
-        zoomed_frame_index = len(ball_path) - 1  # Use last ball position frame for zoom
-
+    out = cv2.VideoWriter(
+        temp_output.name,
+        cv2.VideoWriter_fourcc(*'mp4v'),
+        fps,
+        (frame_width, frame_height)
+    )
+    
     frame_count = 0
-    cap.set(cv2.CAP_PROP_POS_FRAMES, 0)
-
+    zoomed_frame_index = impact_frame - 1 if impact_frame else -1
+    
+    # Extract 2D coordinates for video overlay
+    ball_path_2d = [(x, y) for x, y, z in ball_path_3d]
+    
     while True:
         ret, frame = cap.read()
         if not ret:
             break
-
-        # Add trajectory
-        for i in range(1, len(ball_path)):
-            cv2.line(frame, ball_path[i-1], ball_path[i], (0, 255, 0), 2)
-
+        
+        # Draw 2D trajectory on video
+        for i in range(1, len(ball_path_2d)):
+            cv2.line(frame, ball_path_2d[i-1], ball_path_2d[i], (0, 255, 0), 2)
+        
         # Draw stump zone
         x1, y1, x2, y2 = stump_zone
         cv2.rectangle(frame, (x1, y1), (x2, y2), (0, 0, 255), 2)
-
+        
         # Auto-zoom on impact frame
-        if frame_count == zoomed_frame_index:
-            zoom_frame = zoom_on_point(frame, ball_path[-1])
-            out.write(zoom_frame)
-        else:
-            out.write(frame)
-
+        if frame_count == zoomed_frame_index and ball_path_2d:
+            frame = zoom_on_point(frame, ball_path_2d[-1])
+        
+        out.write(frame)
         frame_count += 1
-
+    
     cap.release()
     out.release()
-
     return temp_output.name
 
 def zoom_on_point(frame, point, zoom_factor=2.0):
@@ -54,16 +54,9 @@ def zoom_on_point(frame, point, zoom_factor=2.0):
     y1 = max(0, cy - box_h // 2)
     x2 = min(w, cx + box_w // 2)
     y2 = min(h, cy + box_h // 2)
-
+    
     cropped = frame[y1:y2, x1:x2]
     zoomed = cv2.resize(cropped, (w, h), interpolation=cv2.INTER_LINEAR)
     return zoomed
 
-def download_button(file_path, filename):
-    with open(file_path, "rb") as file:
-        btn = st.download_button(
-            label="💾 Download Replay Clip",
-            data=file,
-            file_name=filename,
-            mime="video/mp4"
-        )
+# Removed download_button (handled in live_match.py with Gradio)