Update SwitcherAI/utilities.py

SwitcherAI/utilities.py

@@ -16,7 +16,39 @@ from tqdm import tqdm
 import SwitcherAI.globals
 from SwitcherAI import wording
 
-TEMP_DIRECTORY_PATH = r"D:\Switcher\Temp\SwitcherAI\resize-vid"
+# Dynamic temp directory path - cross-platform compatible
+def get_base_temp_directory():
+    """Get base temp directory - cross-platform compatible"""
+    if hasattr(SwitcherAI.globals, 'base_temp_directory') and SwitcherAI.globals.base_temp_directory:
+        return SwitcherAI.globals.base_temp_directory
+    
+    # Default temp directory based on environment
+    if os.getenv('SPACE_ID'):  # HuggingFace Spaces
+        base_temp = os.path.join(tempfile.gettempdir(), "SwitcherAI")
+    elif platform.system().lower() == 'windows':
+        # Try to use the original Windows path if it exists and is writable
+        original_path = r"D:\Switcher\Temp\SwitcherAI"
+        if os.path.exists(os.path.dirname(original_path)):
+            try:
+                os.makedirs(original_path, exist_ok=True)
+                # Test if writable
+                test_file = os.path.join(original_path, "test_write.tmp")
+                with open(test_file, 'w') as f:
+                    f.write("test")
+                os.remove(test_file)
+                base_temp = original_path
+            except (OSError, PermissionError):
+                base_temp = os.path.join(tempfile.gettempdir(), "SwitcherAI")
+        else:
+            base_temp = os.path.join(tempfile.gettempdir(), "SwitcherAI")
+    else:  # Linux/Mac
+        base_temp = os.path.join(tempfile.gettempdir(), "SwitcherAI")
+    
+    os.makedirs(base_temp, exist_ok=True)
+    return base_temp
+
+# Use dynamic temp directory
+TEMP_DIRECTORY_PATH = get_base_temp_directory()
 TEMP_OUTPUT_NAME = 'temp.mp4'
 
 # monkey patch ssl
@@ -106,7 +138,9 @@ def get_temp_frame_paths(target_path : str) -> List[str]:
 
 def get_temp_directory_path(target_path : str) -> str:
 	target_name, _ = os.path.splitext(os.path.basename(target_path))
-	return os.path.join(TEMP_DIRECTORY_PATH, target_name)
+	# Use the dynamic temp directory instead of hardcoded path
+	base_temp = get_base_temp_directory()
+	return os.path.join(base_temp, target_name)
 
 
 def get_temp_output_path(target_path : str) -> str:
@@ -188,3 +222,24 @@ def encode_execution_providers(execution_providers : List[str]) -> List[str]:
 
 def decode_execution_providers(execution_providers : List[str]) -> List[str]:
 	return [provider for provider, encoded_execution_provider in zip(onnxruntime.get_available_providers(), encode_execution_providers(onnxruntime.get_available_providers())) if any(execution_provider in encoded_execution_provider for execution_provider in execution_providers)]
+
+
+def set_temp_directory(temp_path: str) -> None:
+	"""Set custom temp directory path"""
+	global TEMP_DIRECTORY_PATH
+	SwitcherAI.globals.base_temp_directory = temp_path
+	TEMP_DIRECTORY_PATH = temp_path
+	os.makedirs(temp_path, exist_ok=True)
+	print(f"✅ Temp directory set to: {temp_path}")
+
+
+def get_temp_directory_info() -> dict:
+	"""Get temp directory information for debugging"""
+	return {
+		'base_temp': get_base_temp_directory(),
+		'current_temp': TEMP_DIRECTORY_PATH,
+		'platform': platform.system(),
+		'is_hf_spaces': bool(os.getenv('SPACE_ID')),
+		'temp_exists': os.path.exists(TEMP_DIRECTORY_PATH),
+		'temp_writable': os.access(TEMP_DIRECTORY_PATH, os.W_OK) if os.path.exists(TEMP_DIRECTORY_PATH) else False
+	}