dock avatar12

Dockerfile

@@ -52,16 +52,16 @@ RUN pip install --no-cache-dir \
 RUN git clone --depth 1 https://github.com/OpenTalker/SadTalker.git /app/SadTalker
 
 # ---------- Download SadTalker Checkpoints ----------
-# RUN cd /app/SadTalker && \
-#     mkdir -p checkpoints && \
-#     echo "📦 Downloading SadTalker model checkpoints..." && \
-#     wget --progress=bar:force:noscroll --retry-connrefused --waitretry=1 --read-timeout=20 --timeout=15 -t 3 \
-#     https://github.com/OpenTalker/SadTalker/releases/download/v0.0.2-rc/checkpoints.zip -O checkpoints.zip && \
-#     echo "📦 Extracting checkpoints..." && \
-#     unzip -o checkpoints.zip && \
-#     rm checkpoints.zip && \
-#     echo "✅ Checkpoints downloaded successfully" && \
-#     ls -lh checkpoints/
+RUN cd /app/SadTalker && \
+    mkdir -p checkpoints && \
+    echo "📦 Downloading SadTalker model checkpoints..." && \
+    wget --progress=bar:force:noscroll --retry-connrefused --waitretry=1 --read-timeout=20 --timeout=15 -t 3 \
+    https://github.com/OpenTalker/SadTalker/releases/download/v0.0.2-rc/checkpoints.zip -O checkpoints.zip && \
+    echo "📦 Extracting checkpoints..." && \
+    unzip -o checkpoints.zip && \
+    rm checkpoints.zip && \
+    echo "✅ Checkpoints downloaded successfully" && \
+    ls -lh checkpoints/
 
 # Install SadTalker requirements with correct NumPy
 RUN cd /app/SadTalker && \
@@ -101,4 +101,4 @@ RUN cd /app/SadTalker && \
 EXPOSE 8501
 CMD ["streamlit", "run", "avatar_streamlit.py", "--server.port=8501", "--server.address=0.0.0.0"]
 
-# FORCE_REBUILD: 2025-01-04-v23
+# FORCE_REBUILD: 2025-01-04-v13

avatar_streamlit.py

@@ -47,26 +47,87 @@ def ensure_sadtalker_checkpoints():
         return True
     
     print("📦 Downloading SadTalker checkpoints (first run only)...")
+    print("⏳ This may take 2-3 minutes on first startup...")
+    
     try:
         import subprocess
         os.makedirs(checkpoint_dir, exist_ok=True)
         
-        # Download checkpoints
+        # Download checkpoints with better error handling
         zip_path = '/tmp/sadtalker_checkpoints.zip'
         download_url = 'https://github.com/OpenTalker/SadTalker/releases/download/v0.0.2-rc/checkpoints.zip'
         
-        subprocess.run(['wget', '-O', zip_path, download_url], check=True)
-        subprocess.run(['unzip', '-o', zip_path, '-d', '/app/SadTalker/'], check=True)
-        subprocess.run(['rm', zip_path], check=True)
+        print(f"🌐 Downloading from: {download_url}")
         
-        print("✅ Checkpoints downloaded successfully")
-        return True
+        # Try wget with verbose output
+        result = subprocess.run(
+            ['wget', '--timeout=60', '--tries=3', '-O', zip_path, download_url],
+            capture_output=True,
+            text=True
+        )
+        
+        if result.returncode != 0:
+            print(f"❌ wget failed: {result.stderr}")
+            # Try with curl as fallback
+            print("🔄 Trying with curl...")
+            result = subprocess.run(
+                ['curl', '-L', '-o', zip_path, download_url],
+                capture_output=True,
+                text=True,
+                timeout=120
+            )
+            if result.returncode != 0:
+                print(f"❌ curl failed: {result.stderr}")
+                return False
+        
+        # Check if file was downloaded
+        if not os.path.exists(zip_path):
+            print("❌ Download file not found")
+            return False
+        
+        file_size = os.path.getsize(zip_path)
+        print(f"✅ Downloaded {file_size / (1024*1024):.1f} MB")
+        
+        if file_size < 1024 * 1024:  # Less than 1MB is suspicious
+            print("❌ Downloaded file is too small, possibly corrupted")
+            return False
+        
+        # Extract checkpoints
+        print("📦 Extracting checkpoints...")
+        result = subprocess.run(
+            ['unzip', '-o', '-q', zip_path, '-d', '/app/SadTalker/'],
+            capture_output=True,
+            text=True
+        )
+        
+        if result.returncode != 0:
+            print(f"❌ Extraction failed: {result.stderr}")
+            return False
+        
+        # Clean up
+        os.remove(zip_path)
+        
+        # Verify extraction
+        if os.path.exists(checkpoint_file):
+            print("✅ Checkpoints downloaded and extracted successfully")
+            return True
+        else:
+            print(f"❌ Checkpoint file not found after extraction: {checkpoint_file}")
+            # List what was extracted
+            if os.path.exists(checkpoint_dir):
+                files = os.listdir(checkpoint_dir)
+                print(f"Files in checkpoints dir: {files}")
+            return False
+            
     except Exception as e:
+        import traceback
         print(f"❌ Failed to download checkpoints: {e}")
+        print(traceback.format_exc())
         return False
 
-# Download checkpoints on startup
-SADTALKER_AVAILABLE = ensure_sadtalker_checkpoints()
+# Download checkpoints on startup with progress indicator
+with st.spinner("🔄 Initializing SadTalker (first run may take 2-3 minutes)..."):
+    SADTALKER_AVAILABLE = ensure_sadtalker_checkpoints()
 
 # ========== Helpers ==========