Update app.py

app.py

@@ -146,13 +146,23 @@ def estimate_trajectory(ball_positions, frames, detection_frames):
 
     trajectory_3d = [pixel_to_3d(x, y, frame_height, frame_width) for x, y in trajectory_2d]
     detections_3d = [pixel_to_3d(x, y, frame_height, frame_width) for x, y in filtered_positions]
+    
+    # Handle missing pitch and impact points gracefully
     pitch_point_3d = pixel_to_3d(pitch_point[0], pitch_point[1], frame_height, frame_width) if pitch_point else None
     impact_point_3d = pixel_to_3d(impact_point[0], impact_point[1], frame_height, frame_width) if impact_point else None
 
+    # Handle cases where no pitch/impact point is found
+    if pitch_point is None:
+        pitch_frame = "N/A"
+        pitch_point_3d = None  # No 3D coordinates for pitch point
+    if impact_point is None:
+        impact_frame = "N/A"
+        impact_point_3d = None  # No 3D coordinates for impact point
+
     debug_log = (
         f"Trajectory estimated successfully\n"
-        f"Pitch point at frame {pitch_frame + 1}: ({pitch_point[0]:.1f}, {pitch_point[1]:.1f})\n"
-        f"Impact point at frame {impact_frame + 1}: ({impact_point[0]:.1f}, {impact_point[1]:.1f})\n"
+        f"Pitch point at frame {pitch_frame + 1 if pitch_frame != 'N/A' else 'N/A'}: {pitch_point if pitch_point else 'Not detected'}\n"
+        f"Impact point at frame {impact_frame + 1 if impact_frame != 'N/A' else 'N/A'}: {impact_point if impact_point else 'Not detected'}\n"
         f"Detections in frames: {filtered_frames}"
     )
     return trajectory_2d, pitch_point, impact_point, pitch_frame, impact_frame, detections_3d, trajectory_3d, pitch_point_3d, impact_point_3d, debug_log
@@ -328,4 +338,4 @@ iface = gr.Interface(
 )
 
 if __name__ == "__main__":
-    iface.launch()  
+    iface.launch()