Update app.py

app.py

@@ -7,37 +7,41 @@ import uuid
 import shutil
 from tqdm import tqdm
 import threading
-import psutil, os
-
-def print_open_files(tag=""):
-    pid = os.getpid()
-    p = psutil.Process(pid)
-    print(f"[DEBUG] {tag} Open file descriptors:", p.num_fds())
-
-def print_process_count(tag=""):
-    proc_count = len(psutil.pids())
-    print(f"[DEBUG] {tag} Process count:", proc_count)
-
-def print_thread_count(tag=""):
-    print(f"[DEBUG] {tag} Thread count:", len(threading.enumerate()))
+import os
+import psutil
 
-def print_ffmpeg_processes(tag=""):
-    count = 0
-    for p in psutil.process_iter(['name', 'cmdline']):
-        if 'ffmpeg' in (p.info['name'] or '') or \
-           any('ffmpeg' in (arg or '') for arg in (p.info['cmdline'] or [])):
-            count += 1
-    print(f"[DEBUG] {tag} FFmpeg running:", count)
+PROC = psutil.Process(os.getpid())
 
 def debug_resources(tag=""):
+    # Threads in this process
+    thread_count = PROC.num_threads()
+
+    # Children processes spawned by this process (recursive = include grandchildren)
+    children = PROC.children(recursive=True)
+    child_count = len(children)
+
+    # Group children by name to see what’s being spawned (ffmpeg, python, etc.)
+    name_counts = {}
+    for c in children:
+        try:
+            name = c.name()
+        except psutil.Error:
+            name = "unknown"
+        name_counts[name] = name_counts.get(name, 0) + 1
+
+    # File descriptors (Linux/macOS; on Windows use len(PROC.open_files()) instead)
+    try:
+        fd_count = PROC.num_fds()
+    except AttributeError:
+        fd_count = len(PROC.open_files())
+
     print("========== RESOURCE DEBUG", tag, "==========")
-    print_thread_count()
-    print_process_count()
-    print_ffmpeg_processes()
-    print_open_files()
+    print("[DEBUG]", "Threads in this process:", thread_count)
+    print("[DEBUG]", "Child processes (recursive):", child_count)
+    print("[DEBUG]", "Child processes by name:", name_counts)
+    print("[DEBUG]", "Open file descriptors:", fd_count)
     print("============================================")
 
-
 from huggingface_hub import snapshot_download, list_repo_files, hf_hub_download
 import importlib, site