Update app.py

app.py

@@ -133,6 +133,59 @@ def lbw_decision(ball_positions, full_trajectory, frames, pitch_point, impact_po
         return f"Out (Ball hits stumps)", full_trajectory, pitch_point, impact_point
     return f"Not Out (Missing stumps)", 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, decision, frame_width, frame_height):
+    if not frames:
+        return None
+    fourcc = cv2.VideoWriter_fourcc(*'mp4v')
+    out = cv2.VideoWriter(output_path, fourcc, FRAME_RATE / SLOW_MOTION_FACTOR, (frame_width, frame_height))
+
+    trajectory_points = np.array([[p[0] * frame_width / PITCH_LENGTH, frame_height - (p[1] * frame_height / (STUMPS_HEIGHT * 2))] for p in vis_trajectory], dtype=np.int32).reshape((-1, 1, 2))
+
+    for i, frame in enumerate(frames):
+        # Draw stumps outline (scaled back to pixel coordinates)
+        stumps_x = frame_width / 2
+        stumps_y = frame_height * 0.8
+        stumps_width_pixels = frame_width * (STUMPS_WIDTH / PITCH_LENGTH)
+        stumps_height_pixels = frame_height * (STUMPS_HEIGHT / (STUMPS_HEIGHT * 2))
+        cv2.line(frame, (int(stumps_x - stumps_width_pixels / 2), int(stumps_y)),
+                 (int(stumps_x + stumps_width_pixels / 2), int(stumps_y)), (255, 255, 255), 2)
+        cv2.line(frame, (int(stumps_x - stumps_width_pixels / 2), int(stumps_y - stumps_height_pixels)),
+                 (int(stumps_x - stumps_width_pixels / 2), int(stumps_y)), (255, 255, 255), 2)
+        cv2.line(frame, (int(stumps_x + stumps_width_pixels / 2), int(stumps_y - stumps_height_pixels)),
+                 (int(stumps_x + stumps_width_pixels / 2), int(stumps_y)), (255, 255, 255), 2)
+
+        # Draw crease line
+        cv2.line(frame, (int(stumps_x - stumps_width_pixels / 2), int(stumps_y)),
+                 (int(stumps_x + stumps_width_pixels / 2), int(stumps_y)), (255, 255, 0), 2)
+
+        if i in detection_frames and trajectory_points.size > 0:
+            idx = detection_frames.index(i) + 1
+            if idx <= len(trajectory_points):
+                cv2.polylines(frame, [trajectory_points[:idx]], False, (0, 0, 255), 2)  # Blue trajectory
+
+        if pitch_point and i == pitch_frame:
+            x = pitch_point[0] * frame_width / PITCH_LENGTH
+            y = frame_height - (pitch_point[1] * frame_height / (STUMPS_HEIGHT * 2))
+            cv2.circle(frame, (int(x), int(y)), 8, (0, 255, 0), -1)  # Green for pitching
+            cv2.putText(frame, "Pitching", (int(x) + 10, int(y) - 10),
+                        cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 255, 0), 1)
+
+        if impact_point and i == impact_frame:
+            x = impact_point[0] * frame_width / PITCH_LENGTH
+            y = frame_height - (impact_point[1] * frame_height / (STUMPS_HEIGHT * 2))
+            cv2.circle(frame, (int(x), int(y)), 8, (0, 0, 255), -1)  # Red for impact
+            cv2.putText(frame, "Impact", (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 decision:
+            cv2.putText(frame, "Wickets", (int(stumps_x) - 50, int(stumps_y) - 20),
+                        cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 165, 255), 1)  # Orange for wickets
+
+        for _ in range(SLOW_MOTION_FACTOR):
+            out.write(frame)
+    out.release()
+    return output_path
+
 def draw_3d_scene(trajectory, pitch_point, impact_point, decision):
     if not HAS_OPENGL:
         return
@@ -198,6 +251,10 @@ def drs_review(video):
     full_trajectory, vis_trajectory, pitch_point, pitch_frame, impact_point, impact_frame, trajectory_log = estimate_trajectory_3d(ball_positions, detection_frames, frames)
     decision, full_trajectory, pitch_point, impact_point = lbw_decision(ball_positions, full_trajectory, frames, pitch_point, impact_point)
 
+    frame_height, frame_width = frames[0].shape[:2]
+    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, decision, frame_width, frame_height)
+
     if HAS_OPENGL:
         init_3d_window(800, 600)
         from OpenGL.GLUT import glutInit, glutSolidSphere
@@ -206,9 +263,6 @@ def drs_review(video):
             draw_3d_scene(full_trajectory, pitch_point, impact_point, decision)
             event.pump()
 
-    output_path = f"output_{uuid.uuid4()}.mp4"
-    slow_motion_path = None  # To be enhanced with 3D export
-
     debug_output = f"{debug_log}\n{trajectory_log}"
     return f"DRS Decision: {decision}\nDebug Log:\n{debug_output}", slow_motion_path
 
@@ -218,10 +272,10 @@ iface = gr.Interface(
     inputs=gr.Video(label="Upload Video Clip"),
     outputs=[
         gr.Textbox(label="DRS Decision and Debug Log"),
-        gr.Video(label="3D Visualization (Separate Window if enabled)")
+        gr.Video(label="Slow-Motion Replay with 2D Annotations")
     ],
     title="AI-Powered 3D DRS for LBW",
-    description="Upload a video clip for 3D DRS analysis with pitching (green), impact (red), and wickets (orange) visualization."
+    description="Upload a video clip for 3D DRS analysis with pitching (green), impact (red), and wickets (orange) visualization, and 2D annotated video output."
 )
 
 if __name__ == "__main__":