Update app.py

app.py

@@ -1,3 +1,4 @@
+
 import cv2
 import numpy as np
 import torch
@@ -99,7 +100,7 @@ def estimate_trajectory(ball_positions, detection_frames, frames):
         return None, None, None, None, None, None, f"Error in trajectory interpolation: {str(e)}"
 
     vis_trajectory = list(zip(x_coords, y_coords))
-    t_full = np.linspace(times[0], times[-1] + 0.5, len(times) + 5)  # Reduced points
+    t_full = np.linspace(times[0], times[-1] + 0.5, len(times) + 5)
     x_full = fx(t_full)
     y_full = fy(t_full)
     full_trajectory = list(zip(x_full, y_full))
@@ -144,7 +145,7 @@ def lbw_decision(ball_positions, full_trajectory, frames, pitch_point, impact_po
         return f"Out (Ball hits stumps, Pitch at x: {pitch_x:.1f}, y: {pitch_y:.1f}, Impact at x: {impact_x:.1f}, y: {impact_y:.1f})", full_trajectory, pitch_point, impact_point
     return f"Not Out (Missing stumps, Pitch at x: {pitch_x:.1f}, y: {pitch_y:.1f}, Impact at x: {impact_x:.1f}, y: {impact_y:.1f})", full_trajectory, pitch_point, impact_point
 
-def generate_slow_motion(frames, vis_trajectory, pitch_point, pitch_frame, impact_point, impact_frame, detection_frames, output_path):
+def generate_slow_motion(frames, vis_trajectory, pitch_point, pitch_frame, impact_point, impact_frame, detection_frames, output_path, decision):
     if not frames:
         return None
     frame_height, frame_width = frames[0].shape[:2]
@@ -183,7 +184,7 @@ def generate_slow_motion(frames, vis_trajectory, pitch_point, pitch_frame, impac
             cv2.putText(frame, "Impact Factor", (int(x) + 10, int(y) + 20),
                         cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 0, 255), 1)
 
-        if impact_point and i == impact_frame and "Out" in lbw_decision(ball_positions, full_trajectory, frames, pitch_point, impact_point)[0]:
+        if impact_point and i == impact_frame and "Out" in decision:
             cv2.putText(frame, "Wicket Factor", (int(stumps_x) - 50, int(stumps_y) - 20),
                         cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 165, 255), 1)
 
@@ -200,7 +201,7 @@ def drs_review(video):
     decision, full_trajectory, pitch_point, impact_point = lbw_decision(ball_positions, full_trajectory, frames, pitch_point, impact_point)
 
     output_path = f"output_{uuid.uuid4()}.mp4"
-    slow_motion_path = generate_slow_motion(frames, vis_trajectory, pitch_point, pitch_frame, impact_point, impact_frame, detection_frames, output_path)
+    slow_motion_path = generate_slow_motion(frames, vis_trajectory, pitch_point, pitch_frame, impact_point, impact_frame, detection_frames, output_path, decision)
 
     debug_output = f"{debug_log}\n{trajectory_log}"
     return f"DRS Decision: {decision}\nDebug Log:\n{debug_output}", slow_motion_path
@@ -218,4 +219,4 @@ iface = gr.Interface(
 )
 
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch()