rabak

Dockerfile

@@ -65,48 +65,11 @@ RUN pip install --no-cache-dir --force-reinstall "numpy>=1.23.5,<1.24"
 COPY avatar_streamlit.py /app/
 
 # ---------- Fix NumPy Compatibility in SadTalker ----------
-RUN python3 << 'EOF'
-import os
-import re
-
-sadtalker_path = '/app/SadTalker'
-
-# Fix np.float
-for root, dirs, files in os.walk(sadtalker_path):
-    for file in files:
-        if file.endswith('.py'):
-            filepath = os.path.join(root, file)
-            try:
-                with open(filepath, 'r') as f:
-                    content = f.read()
-                new_content = re.sub(r'\bnp\.float\b(?!\d)', 'np.float64', content)
-                if new_content != content:
-                    with open(filepath, 'w') as f:
-                        f.write(new_content)
-            except:
-                pass
-
-# Fix preprocess.py
-preprocess_file = os.path.join(sadtalker_path, 'src/face3d/util/preprocess.py')
-try:
-    with open(preprocess_file, 'r') as f:
-        content = f.read()
-
-    content = content.replace(
-        'warnings.filterwarnings("ignore", category=np.VisibleDeprecationWarning)',
-        '# warnings.filterwarnings("ignore", category=np.VisibleDeprecationWarning)'
-    )
-    content = content.replace(
-        "trans_params = np.array([w0, h0, s, t[0], t[1]])",
-        "trans_params = np.array([w0, h0, s, float(t[0]), float(t[1])])"
-    )
-
-    with open(preprocess_file, 'w') as f:
-        f.write(content)
-    print("✅ SadTalker fixes applied!")
-except Exception as e:
-    print(f"⚠️ Could not fix preprocess.py: {e}")
-EOF
+RUN cd /app/SadTalker && \
+    find . -name "*.py" -type f -exec sed -i 's/\bnp\.float\b/np.float64/g' {} + && \
+    sed -i 's/warnings.filterwarnings("ignore", category=np.VisibleDeprecationWarning)/# warnings.filterwarnings("ignore", category=np.VisibleDeprecationWarning)/' src/face3d/util/preprocess.py && \
+    sed -i 's/trans_params = np.array(\[w0, h0, s, t\[0\], t\[1\]\])/trans_params = np.array([w0, h0, s, float(t[0]), float(t[1])])/' src/face3d/util/preprocess.py && \
+    echo "✅ SadTalker fixes applied!"
 
 # ---------- Preload Coqui Model (Skip on build to save memory) ----------
 # Model will download on first run instead
@@ -116,4 +79,4 @@ EOF
 EXPOSE 8501
 CMD ["streamlit", "run", "avatar_streamlit.py", "--server.port=8501", "--server.address=0.0.0.0"]
 
-# FORCE_REBUILD: 2025-01-04-v2
+# FORCE_REBUILD: 2025-01-04-v4