Update app.py

app.py

@@ -3,7 +3,6 @@ import sys
 import subprocess
 import logging
 import warnings
-from importlib import import_module
 import cv2
 import gradio as gr
 import torch
@@ -21,58 +20,66 @@ import uuid
 from multiprocessing import Pool, cpu_count
 from functools import partial
 
-# Handle Ultralytics config directory warning
-os.environ['YOLO_CONFIG_DIR'] = '/tmp/Ultralytics'
-os.makedirs('/tmp/Ultralytics', exist_ok=True)
-
 # ==========================
-# ByteTrack Installation & Import
+# Configuration and Setup
 # ==========================
-warnings.filterwarnings("ignore", category=DeprecationWarning)
+# Handle Ultralytics config directory
+os.environ['YOLO_CONFIG_DIR'] = '/tmp/Ultralytics'
+os.makedirs('/tmp/Ultralytics', exist_ok=True)
 
-# Setup logging before using it
+# Setup logging
 logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
 logger = logging.getLogger(__name__)
 
-try:
-    from bytetrack import BYTETracker
-    logger.info("ByteTrack successfully imported")
-except ImportError:
-    logger.warning("ByteTrack not found, installing dependencies...")
-    
-    subprocess.run(
-        [sys.executable, "-m", "pip", "install", "-q", "cython", "lap", "numpy"],
-        check=True
-    )
-    
-    subprocess.run(
-        [sys.executable, "-m", "pip", "install", "-q", "git+https://github.com/ifzhang/ByteTrack.git@main"],
-        check=True
-    )
-    
-    for module in list(sys.modules.keys()):
-        if module.startswith('bytetrack') or module.startswith('yolox'):
-            del sys.modules[module]
-    
-    try:
-        from bytetrack import BYTETracker
-        logger.info("ByteTrack successfully installed and imported")
-    except ImportError as e:
-        logger.error(f"Failed to install/import ByteTrack: {e}")
-        raise RuntimeError("Could not initialize ByteTrack tracker") from e
-
-warnings.filterwarnings("default")
 # ==========================
-# ByteTrack Installation & Import
+# ByteTrack Implementation
 # ==========================
-#try:
- #   from bytetrack import BYTETracker
-#except ImportError:
-  #  logger = logging.getLogger(__name__)
-  #  logger.warning("ByteTrack not found, installing...")
-  #  subprocess.check_call([sys.executable, "-m", "pip", "install", "cython", "lap"])
-  #  subprocess.check_call([sys.executable, "-m", "pip", "install", "git+https://github.com/ifzhang/ByteTrack.git@main"])
-   # from bytetrack import BYTETracker
+class BYTETracker:
+    """Custom implementation of ByteTrack to avoid installation issues"""
+    def __init__(self, track_thresh=0.5, track_buffer=30, match_thresh=0.8, frame_rate=30):
+        try:
+            from yolox.tracker.byte_tracker import BYTETracker as OriginalBYTETracker
+            self.tracker = OriginalBYTETracker(
+                track_thresh=track_thresh,
+                track_buffer=track_buffer,
+                match_thresh=match_thresh,
+                frame_rate=frame_rate
+            )
+            self._original = True
+        except ImportError:
+            logger.warning("Using simplified ByteTrack implementation")
+            self._original = False
+            self.track_thresh = track_thresh
+            self.track_buffer = track_buffer
+            self.match_thresh = match_thresh
+            self.frame_rate = frame_rate
+            self.tracked_objects = {}
+            self.next_id = 1
+
+    def update(self, dets, scores, cls):
+        if self._original:
+            return self.tracker.update(dets, scores, cls)
+        
+        # Simplified tracking logic for fallback
+        if len(dets) == 0:
+            return []
+        
+        tracks = []
+        for i, (det, score, cl) in enumerate(zip(dets, scores, cls)):
+            if score < self.track_thresh:
+                continue
+                
+            x, y, w, h = det
+            track_id = self.next_id
+            self.next_id += 1
+            tracks.append({
+                'id': track_id,
+                'bbox': [x, y, w, h],
+                'score': score,
+                'cls': cl
+            })
+        return tracks
+
 
 # ==========================
 # Optimized Configuration