docker compose, fix_symlinks

.dockerignore

@@ -0,0 +1,3 @@
+.python-version
+.backup
+backup

.gitignore

@@ -26,6 +26,8 @@ share/python-wheels/
 *.egg
 MANIFEST
 
+backup/
+
 # PyInstaller
 #  Usually these files are written by a python script from a template
 #  before PyInstaller builds the exe, so as to inject date/other infos into it.

.python-version

@@ -1 +1 @@
-3.10
+tts

README.md

@@ -24,3 +24,26 @@ pip install torch==2.4.0+cu124 torchaudio==2.4.0+cu124 torchvision==0.19.0 --ext
 sudo apt-get update
 sudo apt-get install sox ffmpeg
 ```
+
+
+## Backup and Restore Volume
+
+p/s: Need to mount backup folder to container
+
+### Backup
+```bash
+tar -czvf /backup/data_cache_backup.tar.gz /home/user/.cache
+tar -czvf /backup/data_gfpgan_backup.tar.gz /home/user/app/gfpgan
+```
+
+### Restore
+
+```bash
+mkdir -p /home/user/.cache
+cd /home/user/.cache
+tar xzvf /backup/data_cache_backup.tar.gz --strip 1
+
+mkdir -p /home/user/app/gfpgan
+cd /home/user/app/gfpgan
+tar xzvf /backup/data_gfpgan_backup.tar.gz --strip 1
+```

docker-compose.yaml

@@ -0,0 +1,20 @@
+services:
+  test:
+    build: .
+    ports:
+      - "7860:7860"
+    stdin_open: true   # equivalent to -it
+    tty: true          # equivalent to -it
+    restart: "no"      # equivalent to --rm (don’t restart container automatically)
+    # environment:
+      # - HF_ENDPOINT=http://172.16.15.118:9557/repository/atalink-hf-models
+    volumes:
+      - test_data_cache:/home/user/.cache
+      - test_data_gfpgan:/home/user/app/gfpgan
+      # - ./backup:/backup
+
+volumes:
+  test_data_cache:
+    external: true
+  test_data_gfpgan:
+    external: true

fix_symlinks.py

@@ -0,0 +1,19 @@
+#!/usr/bin/env python3
+import os
+# cd checkpoints
+# Root folder where your symlinks are located
+folder = "."
+old_prefix = "/Users/cuongpc/"
+new_prefix = "/home/user/"
+
+old_prefix_1 = "../../../../../root/"
+
+for entry in os.listdir(folder):
+    path = os.path.join(folder, entry)
+    if os.path.islink(path):
+        target = os.readlink(path)
+        if target.startswith(old_prefix) or target.startswith(old_prefix_1):
+            new_target = target.replace(old_prefix, new_prefix, 1).replace(old_prefix_1, new_prefix, 1)
+            print(f"Updating: {path}\n  {target} -> {new_target}")
+            os.remove(path)  # remove old symlink
+            os.symlink(new_target, path)

prepare_environment.py

@@ -0,0 +1,11 @@
+
+# help me write a python script to download a file from remote server and storage in /tmp
+
+# if the file has already exist in /tmp, just skip download step
+
+# the next step, create the volume, and extract the downloaded data into that volume
+
+# the final step, cleanup /tmp
+
+# http://172.16.15.118:9557/repository/atalink-hf-models/backups/data_cache_backup.tar.gz
+# http://172.16.15.118:9557/repository/atalink-hf-models/backups/data_gfpgan_backup.tar.gz