Update app.py

app.py

@@ -1,4 +1,9 @@
 import os
+import sys
+import subprocess
+import logging
+import warnings
+from importlib import import_module
 import cv2
 import gradio as gr
 import torch
@@ -11,25 +16,63 @@ from reportlab.pdfgen import canvas
 from reportlab.lib.units import inch
 from io import BytesIO
 import base64
-import logging
 from retrying import retry
 import uuid
 from multiprocessing import Pool, cpu_count
 from functools import partial
-import subprocess
-import sys
+
+# Handle Ultralytics config directory warning
+os.environ['YOLO_CONFIG_DIR'] = '/tmp/Ultralytics'
+os.makedirs('/tmp/Ultralytics', exist_ok=True)
 
 # ==========================
 # ByteTrack Installation & Import
 # ==========================
+warnings.filterwarnings("ignore", category=DeprecationWarning)
+
+# Setup logging before using it
+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 = 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
+    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
+# ==========================
+#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
 
 # ==========================
 # Optimized Configuration