Update src/components/faceswap.py

src/components/faceswap.py

@@ -33,10 +33,10 @@ class FaceSwapper:
         # Set up the gdown cache directory
         gdown_cache_dir = "/tmp/.cache/gdown"
         os.makedirs(gdown_cache_dir, exist_ok=True)
-        os.environ['XDG_CACHE_HOME'] = gdown_cache_dir
+        os.environ['XDG_CACHE_HOME'] = "/tmp/.cache"
 
         # Log the cache directory and cookie path for debugging
-        logger.info(f"Setting XDG_CACHE_HOME to: {gdown_cache_dir}")
+        logger.info(f"Setting XDG_CACHE_HOME to: {os.environ['XDG_CACHE_HOME']}")
 
         # Ensure cookies.txt exists
         cookie_path = os.path.join(gdown_cache_dir, "cookies.txt")
@@ -46,6 +46,9 @@ class FaceSwapper:
                 pass
         logger.info(f"Ensured cookies.txt exists at: {cookie_path}")
 
+        # Set gdown's home directory explicitly
+        gdown.config.home = gdown_cache_dir
+
         # Check if model path exists, and download if not
         if not os.path.exists(model_path):
             os.makedirs(os.path.dirname(model_path), exist_ok=True)