Update app.py

app.py

@@ -32,13 +32,13 @@ class BallTrackerLinkedIn:
         self.smooth_window = 5
         self.max_interpolate_gap = 30
 
-    def process_batch(self, frames, progress=gr.Progress()):
+    def process_batch(self, frames, progress_callback=None):
         """Process un batch de frames pour le tracking"""
         positions = []
 
         for i, frame in enumerate(frames):
-            if progress:
-                progress((i + 1) / len(frames), desc=f"Detecting ball... {i+1}/{len(frames)}")
+            if progress_callback:
+                progress_callback((i + 1) / len(frames), desc=f"Detecting ball... {i+1}/{len(frames)}")
             
             self.frame_idx = i
             results = self.ball_model.track(
@@ -168,7 +168,7 @@ class VideoProcessorLinkedIn:
                 cv2.circle(frame, (x, y), 4, (255, 255, 255), 1, cv2.LINE_AA)
 
     def process_video(self, video_path, player1_name="PLAYER 1", player2_name="PLAYER 2", 
-                     max_duration=30, progress=gr.Progress()):
+                     max_duration=30, progress=gr.Progress(track_tqdm=True)):
         """Traiter la vidéo et retourner la version annotée"""
         
         if video_path is None:
@@ -207,7 +207,7 @@ class VideoProcessorLinkedIn:
 
             # Phase 1: Tracking de la balle
             progress(0.1, desc="Tracking ball...")
-            positions = self.tracker.process_batch(frames, progress)
+            positions = self.tracker.process_batch(frames, progress_callback=progress)
             
             # Phase 2: Interpolation
             progress(0.4, desc="Interpolating missing positions...")
@@ -233,9 +233,8 @@ class VideoProcessorLinkedIn:
             trail_positions = []
 
             for frame_idx, (_, ball_pos) in enumerate(positions):
-                if progress:
-                    progress(0.6 + 0.3 * (frame_idx / len(frames)), 
-                            desc=f"Rendering... {frame_idx+1}/{len(frames)}")
+                progress(0.6 + 0.3 * (frame_idx / len(frames)), 
+                        desc=f"Rendering... {frame_idx+1}/{len(frames)}")
                 
                 annotated = frames[frame_idx].copy()
 
@@ -318,32 +317,30 @@ class VideoProcessorLinkedIn:
             return None, f"❌ Erreur: {str(e)}"
 
 
-# Télécharger les modèles au démarrage
-def download_models():
-    """Télécharge les modèles YOLO nécessaires"""
-    logger.info("Downloading YOLO models...")
-    
-    # Modèles de base (téléchargés automatiquement par ultralytics)
-    _ = YOLO('yolov8m.pt')
-    _ = YOLO('yolov8m-pose.pt')
-    
-    logger.info("✅ Models ready!")
-
+# Variable globale pour le processeur (initialisé paresseusement)
+processor = None
 
-# Initialiser le processeur
-logger.info("Initializing app...")
-download_models()
-
-# Note: Pour le modèle de balle de tennis personnalisé, vous devrez l'uploader
-# Pour l'instant, on utilise le modèle YOLO standard
-ball_model_path = 'yolov8m.pt'  # Remplacer par le chemin de votre modèle custom
-processor = VideoProcessorLinkedIn(ball_model_path)
+def get_processor():
+    """Initialise le processeur de manière paresseuse"""
+    global processor
+    if processor is None:
+        logger.info("Initializing processor...")
+        # Télécharger les modèles
+        logger.info("Downloading YOLO models...")
+        _ = YOLO('yolov8m.pt')
+        _ = YOLO('yolov8m-pose.pt')
+        logger.info("✅ Models ready!")
+        
+        ball_model_path = 'yolov8m.pt'
+        processor = VideoProcessorLinkedIn(ball_model_path)
+    return processor
 
 
 # Interface Gradio
-def process_video_gradio(video, player1, player2, max_duration):
+def process_video_gradio(video, player1, player2, max_duration, progress=gr.Progress(track_tqdm=True)):
     """Wrapper pour Gradio"""
-    return processor.process_video(video, player1, player2, max_duration)
+    proc = get_processor()
+    return proc.process_video(video, player1, player2, max_duration, progress)
 
 
 # Créer l'interface
@@ -415,8 +412,4 @@ with gr.Blocks(title="🎾 CourtSide-CV - Tennis Analysis", theme=gr.themes.Soft
 
 # Lancer l'application
 if __name__ == "__main__":
-    demo.launch(
-        server_name="0.0.0.0",  # Écoute sur toutes les interfaces
-        server_port=7860,        # Port par défaut HF Spaces
-        share=False              # Pas besoin de share sur HF
-    )
+    demo.launch()