Update app.py

app.py

@@ -6,9 +6,14 @@ import gradio as gr
 from scipy.interpolate import interp1d
 import uuid
 import os
-from OpenGL.GL import *
-from OpenGL.GLU import *
-from pygame import display, event, QUIT
+try:
+    from OpenGL.GL import *
+    from OpenGL.GLU import *
+    from pygame import display, event, QUIT
+    HAS_OPENGL = True
+except ImportError:
+    print("Warning: PyOpenGL or Pygame not found. 3D visualization will be disabled. Install with 'pip install PyOpenGL PyOpenGL_accelerate pygame'.")
+    HAS_OPENGL = False
 
 # Load the trained YOLOv8n model
 model = YOLO("best.pt")
@@ -64,15 +69,14 @@ def estimate_trajectory_3d(ball_positions, detection_frames, frames):
         return None, None, None, None, None, None, "Error: Fewer than 2 valid single-ball detections"
     frame_height, frame_width = frames[0].shape[:2]
 
-    # Simple 2D to 3D mapping (approximation)
     x_coords = np.array([pos[0] for pos in ball_positions]) / frame_width * PITCH_LENGTH
     y_coords = np.array([frame_height - pos[1] for pos in ball_positions]) / frame_height * STUMPS_HEIGHT * 2
-    z_coords = np.zeros_like(x_coords)  # Depth estimation needed (simplified as 0 for now)
+    z_coords = np.zeros_like(x_coords)  # Placeholder for depth
     times = np.array([i / FRAME_RATE for i in range(len(ball_positions))])
 
     pitch_idx = 0
     for i, y in enumerate(y_coords):
-        if y < STUMPS_HEIGHT:  # Pitch near ground
+        if y < STUMPS_HEIGHT:
             pitch_idx = i
             break
     pitch_point = (x_coords[pitch_idx], y_coords[pitch_idx], 0)
@@ -89,7 +93,6 @@ def estimate_trajectory_3d(ball_positions, detection_frames, frames):
     impact_point = (x_coords[impact_idx], y_coords[impact_idx], 0)
     impact_frame = detection_frames[impact_idx]
 
-    # Interpolate 3D trajectory
     fx = interp1d(times[:impact_idx + 1], x_coords[:impact_idx + 1], kind='linear', fill_value="extrapolate")
     fy = interp1d(times[:impact_idx + 1], y_coords[:impact_idx + 1], kind='quadratic', fill_value="extrapolate")
     fz = interp1d(times[:impact_idx + 1], z_coords[:impact_idx + 1], kind='linear', fill_value="extrapolate")
@@ -104,7 +107,7 @@ def lbw_decision(ball_positions, full_trajectory, frames, pitch_point, impact_po
         return "Error: No data", None, None, None
     frame_height, frame_width = frames[0].shape[:2]
     stumps_x = PITCH_LENGTH / 2
-    stumps_y = 0  # Ground level
+    stumps_y = 0
     stumps_width = STUMPS_WIDTH
 
     pitch_x, pitch_y, _ = pitch_point
@@ -130,14 +133,9 @@ 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 init_3d_window(width, height):
-    pygame.init()
-    display.set_mode((width, height), DOUBLEBUF | OPENGL)
-    gluPerspective(45, (width / height), 0.1, 50.0)
-    glTranslatef(0.0, -5.0, -30)
-    glEnable(GL_DEPTH_TEST)
-
 def draw_3d_scene(trajectory, pitch_point, impact_point, decision):
+    if not HAS_OPENGL:
+        return
     glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT)
     glBegin(GL_LINES)
     for i in range(len(trajectory) - 1):
@@ -146,7 +144,6 @@ def draw_3d_scene(trajectory, pitch_point, impact_point, decision):
         glVertex3f(trajectory[i + 1][0], trajectory[i + 1][1], trajectory[i + 1][2])
     glEnd()
 
-    # Draw pitch and stumps
     glColor3f(0, 1, 0)  # Green pitch
     glBegin(GL_QUADS)
     glVertex3f(0, 0, 0)
@@ -163,7 +160,6 @@ def draw_3d_scene(trajectory, pitch_point, impact_point, decision):
     glVertex3f(PITCH_LENGTH / 2 + STUMPS_WIDTH / 2, STUMPS_HEIGHT, 0)
     glEnd()
 
-    # Draw pitching and impact points
     if pitch_point:
         glColor3f(0, 1, 0)  # Green
         glPushMatrix()
@@ -178,7 +174,6 @@ def draw_3d_scene(trajectory, pitch_point, impact_point, decision):
         glutSolidSphere(0.1, 20, 20)
         glPopMatrix()
 
-    # Display decision (simplified text)
     if "Out" in decision:
         glColor3f(1, 0.65, 0)  # Orange
         glRasterPos3f(PITCH_LENGTH / 2, STUMPS_HEIGHT, 0)
@@ -187,6 +182,15 @@ def draw_3d_scene(trajectory, pitch_point, impact_point, decision):
 
     display.flip()
 
+def init_3d_window(width, height):
+    if not HAS_OPENGL:
+        return
+    pygame.init()
+    display.set_mode((width, height), DOUBLEBUF | OPENGL)
+    gluPerspective(45, (width / height), 0.1, 50.0)
+    glTranslatef(0.0, -5.0, -30)
+    glEnable(GL_DEPTH_TEST)
+
 def drs_review(video):
     frames, ball_positions, detection_frames, debug_log = process_video(video)
     if not frames:
@@ -194,28 +198,27 @@ 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)
 
-    # Generate 3D visualization (separate window)
-    init_3d_window(800, 600)
-    from OpenGL.GLUT import glutInit, glutSolidSphere
-    glutInit()
-    for _ in range(100):  # Limited frames for demo
-        draw_3d_scene(full_trajectory, pitch_point, impact_point, decision)
-        event.pump()
+    if HAS_OPENGL:
+        init_3d_window(800, 600)
+        from OpenGL.GLUT import glutInit, glutSolidSphere
+        glutInit()
+        for _ in range(100):  # Limited frames for demo
+            draw_3d_scene(full_trajectory, pitch_point, impact_point, decision)
+            event.pump()
 
     output_path = f"output_{uuid.uuid4()}.mp4"
-    # Note: 3D rendering is separate; 2D video output is placeholder
     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
 
-# Gradio interface (placeholder for 3D)
+# Gradio interface
 iface = gr.Interface(
     fn=drs_review,
     inputs=gr.Video(label="Upload Video Clip"),
     outputs=[
         gr.Textbox(label="DRS Decision and Debug Log"),
-        gr.Video(label="3D Visualization (Separate Window)")
+        gr.Video(label="3D Visualization (Separate Window if enabled)")
     ],
     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."