[DEV] update ui app.py, update prepare_env.py

app.py

@@ -114,35 +114,22 @@ def sadtalker_demo():
             f"""
     ![logo](https://vietnam.atalink.com/favicon.ico)
 
-    ## F5-TTS & SadTalker
+    ## Atalink TTS_Talker
 
-    Nhập text, upload sample voice và ảnh để tạo video nói chuyện. Chủ đạo Atalink Blue.
+    Nhập text, upload sample voice và ảnh để tạo video nói chuyện.
         """
         )
-        with gr.Tab("Lịch sử video"):
-            with gr.Row(elem_classes="gr-row"):
-                video_list = gr.Dropdown(
-                    choices=list_videos(),
-                    label="Chọn video để xem",
-                    interactive=True,
-                    scale=1,
-                )
-                video_player = gr.Video(
-                    height=180, width=180, label="Video lịch sử", scale=1
-                )
-            video_list.change(lambda x: x, inputs=video_list, outputs=video_player)
-
         with gr.Tab("Tạo video mới"):
             with gr.Row(elem_classes="gr-row"):
                 ref_audio = gr.Audio(label="🔊 Sample Voice", type="filepath")
                 ref_text = gr.Textbox(
-                    label="📝 Reference Transcript (optional)",
+                    label="📝 Nội dung tham khảo (tùy chọn)",
                     placeholder="Nhập transcript tiếng Việt cho sample voice nếu có...",
                     lines=2,
                 )
             with gr.Row(elem_classes="gr-row"):
                 gen_text = gr.Textbox(
-                    label="📝 Text",
+                    label="📝 Nội dung cần tạo",
                     placeholder="Nhập nội dung để tạo giọng nói...",
                     lines=3,
                 )
@@ -214,12 +201,28 @@ def sadtalker_demo():
                     use_idle_mode = gr.Checkbox(label="Idle Animation")
                     length_of_audio = gr.Number(value=5, label="Độ dài video (giây)")
                     blink_every = gr.Checkbox(label="Chớp mắt", value=True)
-            btn_generate = gr.Button("🔥 Tạo giọng nói & video", elem_id="btn-generate")
+            btn_generate = gr.Button(
+                "🔥 Tạo giọng nói & video", elem_id="btn-generate", interactive=False
+            )
             with gr.Row(elem_classes="gr-row"):
                 output_audio = gr.Audio(label="🎧 Audio đã tạo", type="filepath")
                 gen_video = gr.Video(
                     label="Video đã tạo", format="mp4", scale=1, height=180, width=180
                 )
+
+            def enable_generate(audio, text, image):
+                return gr.update(interactive=bool(audio and text and image))
+
+            ref_audio.change(
+                enable_generate, [ref_audio, gen_text, source_image], btn_generate
+            )
+            gen_text.change(
+                enable_generate, [ref_audio, gen_text, source_image], btn_generate
+            )
+            source_image.change(
+                enable_generate, [ref_audio, gen_text, source_image], btn_generate
+            )
+
             btn_generate.click(
                 generate_voice_and_video,
                 inputs=[
@@ -245,6 +248,19 @@ def sadtalker_demo():
                 ],
                 outputs=[output_audio, gen_video],
             )
+        with gr.Tab("Lịch sử video"):
+            with gr.Row(elem_classes="gr-row"):
+                video_list = gr.Dropdown(
+                    choices=list_videos(),
+                    label="Chọn video để xem",
+                    interactive=True,
+                    scale=1,
+                )
+                video_player = gr.Video(
+                    height=180, width=180, label="Video lịch sử", scale=1
+                )
+            video_list.change(lambda x: x, inputs=video_list, outputs=video_player)
+
     return sadtalker_interface
 
 

utils/prepare_environment.py

@@ -1,4 +1,3 @@
-
 # 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
@@ -10,70 +9,92 @@
 # 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
 
-
-#https://drive.usercontent.google.com/download?id=1Kuh-kZApPvI6EIBBy5fK1SRy8fGBHg4z&export=download&authuser=0&confirm=t&uuid=302429b1-d1d4-4aba-8320-ec04d80da509&at=AN8xHoqDn4UCrIap5o_SrGrqaD9F:1758299190305
 import os
 import requests
 import shutil
 import subprocess
 import tarfile
+from tqdm import tqdm
 
 DOWNLOADS = [
-    # "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"
-    "https://huggingface.co/hynt/F5-TTS-Vietnamese-ViVoice/resolve/main/config.json"
+    "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"
+    # "https://huggingface.co/hynt/F5-TTS-Vietnamese-ViVoice/resolve/main/config.json"
 ]
 
 
 TMP_DIR = "/tmp"
-VOLUME_PREFIX = "test_data_"
+VOLUME_PREFIX = "test_"
+
 
 def download_file(url, dest_folder):
     filename = os.path.basename(url)
     dest_path = os.path.join(dest_folder, filename)
     if os.path.exists(dest_path):
-        print(f"{filename} already exists at ```{dest_path}```, skipping download.")
+        print(f"✅ [SKIP] {filename} already exists at \033[96m{dest_path}\033[0m, skipping download.")
         return dest_path
-    print(f"Downloading {filename} ...")
+    print(f"\n📥 [START] Downloading: \033[1;36m{filename}\033[0m → \033[96m{dest_path}\033[0m")
     with requests.get(url, stream=True) as r:
         r.raise_for_status()
-        with open(dest_path, 'wb') as f:
-            shutil.copyfileobj(r.raw, f)
-    print(f"Downloaded {filename} at {dest_path}")
+        total = int(r.headers.get("content-length", 0))
+        chunk_size = 1024 * 1024  # 1MB
+        with open(dest_path, "wb") as f, tqdm(
+            total=total, unit="B", unit_scale=True, desc=filename
+        ) as pbar:
+            for chunk in r.iter_content(chunk_size=chunk_size):
+                if chunk:
+                    f.write(chunk)
+                    pbar.update(len(chunk))
+    print(f"✅ [DONE] Downloaded: \033[1;32m{filename}\033[0m → \033[96m{dest_path}\033[0m\n")
     return dest_path
 
+
 def create_volume_and_extract(tar_path, volume_name):
     # Create Docker volume
     subprocess.run(["docker", "volume", "create", volume_name], check=True)
     # Extract tar.gz into the volume using a temporary container
-    print(f"Extracting {tar_path} into volume {volume_name} ...")
-    subprocess.run([
-        "docker", "run", "--rm",                       # Chạy container tạm thời, tự xóa sau khi xong
-        "-v", f"{volume_name}:/data",                  # Mount Docker volume (volume_name) vào thư mục /data trong container
-        "-v", f"{os.path.dirname(tar_path)}:/tmpdata", # Mount thư mục chứa file tar.gz trên host vào /tmpdata trong container
-        "busybox",                                # Image dùng để chạy container (ở đây là Ubuntu 22.04)
-        "bash", "-c",                                  # Chạy lệnh bash trong container
-        f"tar -xzf /tmpdata/{os.path.basename(tar_path)} -C /data" # Lệnh giải nén file tar.gz từ /tmpdata vào /data
-    ], check=True)
-    print(f"Extracted {tar_path} into volume {volume_name}")
+    print(f"\n📦 [EXTRACT] Extracting \033[1;36m{os.path.basename(tar_path)}\033[0m into Docker volume \033[1;33m{volume_name}\033[0m ...")
+    subprocess.run(
+        [
+            "docker",
+            "run",
+            "--rm",  # Chạy container tạm thời, tự xóa sau khi xong
+            "-v",
+            f"{volume_name}:/data",  # Mount Docker volume (volume_name) vào thư mục /data trong container
+            "-v",
+            f"{os.path.dirname(tar_path)}:/tmpdata",  # Mount thư mục chứa file tar.gz trên host vào /tmpdata trong container
+            "busybox",  # Image dùng để chạy container (ở đây là Ubuntu 22.04)
+            "bash",
+            "-c",  # Chạy lệnh bash trong container
+            f"tar -xzf /tmpdata/{os.path.basename(tar_path)} -C /data",  # Lệnh giải nén file tar.gz từ /tmpdata vào /data
+        ],
+        check=True,
+    )
+    print(f"✅ [DONE] Extracted \033[1;36m{os.path.basename(tar_path)}\033[0m into volume \033[1;33m{volume_name}\033[0m\n")
+
 
 def cleanup_tmp(files):
     for f in files:
         try:
             os.remove(f)
-            print(f"Removed {f}")
+            print(f"🧹 [CLEANUP] Removed \033[96m{f}\033[0m")
         except Exception as e:
-            print(f"Could not remove {f}: {e}")
+            print(f"⚠️ [CLEANUP] Could not remove \033[96m{f}\033[0m: {e}")
+
 
 def main():
     downloaded_files = []
     for url in DOWNLOADS:
         tar_path = download_file(url, TMP_DIR)
         downloaded_files.append(tar_path)
-        volume_name = VOLUME_PREFIX + os.path.splitext(os.path.splitext(os.path.basename(tar_path))[0])[0]
-        print("🚀 ~ volume_name:", volume_name)
+        volume_name = (
+            VOLUME_PREFIX
+            + os.path.splitext(os.path.splitext(os.path.basename(tar_path))[0])[0]
+        )
+    print(f"🚀 [VOLUME] Name: \033[1;33m{volume_name}\033[0m")
         # create_volume_and_extract(tar_path, volume_name)
     # cleanup_tmp(downloaded_files)
 
+
 if __name__ == "__main__":
     main()