make detect_disk_metrics stateless

facefusion/system.py

@@ -1,5 +1,7 @@
 import shutil
+from typing import List
 
+from facefusion import state_manager
 from facefusion.execution import detect_execution_devices
 from facefusion.types import DiskMetrics, Metrics
 
@@ -8,28 +10,33 @@ def get_metrics_set() -> Metrics:
 	return\
 	{
 		'execution_devices': detect_execution_devices(),
-		'disk': detect_disk_metrics()
+		'disks': detect_disk_metrics([ state_manager.get_temp_path() ])
 	}
 
 
-def detect_disk_metrics() -> DiskMetrics:
+def detect_disk_metrics(drive_paths : List[str]) -> List[DiskMetrics]:
-	usage = shutil.disk_usage('.')
+	disk_metrics : List[DiskMetrics] = []
 
-	return\
+	for drive_path in drive_paths:
-	{
+		disk_usage = shutil.disk_usage(drive_path)
-		'total':
+
+		disk_metrics.append(
 		{
-			'value': int(usage.total / (1024 * 1024 * 1024)),
+			'total':
-			'unit': 'GiB'
+			{
-		},
+				'value': int(disk_usage.total / (1024 * 1024 * 1024)),
-		'free':
+				'unit': 'GiB'
-		{
+			},
-			'value': int(usage.free / (1024 * 1024 * 1024)),
+			'free':
-			'unit': 'GiB'
+			{
-		},
+				'value': int(disk_usage.free / (1024 * 1024 * 1024)),
-		'utilization':
+				'unit': 'GiB'
-		{
+			},
-			'value': int(usage.used / usage.total * 100),
+			'utilization':
-			'unit': '%'
+			{
-		}
+				'value': int(disk_usage.used / disk_usage.total * 100),
-	}
+				'unit': '%'
+			}
+		})
+
+	return disk_metrics

facefusion/types.py

@@ -307,7 +307,7 @@ DiskMetrics = TypedDict('DiskMetrics',
 Metrics = TypedDict('Metrics',
 {
 	'execution_devices' : List[ExecutionDevice],
-	'disk' : DiskMetrics
+	'disks' : List[DiskMetrics]
 })
 
 DownloadProvider = Literal['github', 'huggingface']