add deps for backend dockerfile

README.md

@@ -46,7 +46,7 @@ A simple GUI app to synchronize recorded audio with video lip movements using th
 ## Docker
 - In the root project directory, run
 ```bash
-docker-compose build
+docker-compose up
 ```
 
 

backend/docker/Dockerfile.backend

@@ -5,9 +5,10 @@ FROM pytorch/pytorch:1.8.1-cuda11.1-cudnn8-runtime
 WORKDIR /backend
 
 # Install system dependencies
-RUN apt-get update && apt-get install -y git wget && \
+RUN apt-get update && apt-get install -y git wget libgl1-mesa-glx && \
     rm -rf /var/lib/apt/lists/*
 
+
 # Install backend dependencies
 RUN pip install poetry
 

backend/docker/download_models.py

@@ -1,5 +1,8 @@
 import gdown
 
+base_directory = "/backend/wav2lip-hq/"
+checkpoints_directory = base_directory + "checkpoints/"
+
 urls = {
     "wav2lip_gan.pth": "10Iu05Modfti3pDbxCFPnofmfVlbkvrCm",
     "face_segmentation.pth": "154JgKpzCPW82qINcVieuPH3fZ2e0P812",
@@ -8,6 +11,6 @@
 
 for name, id in urls.items():
     url = f"https://drive.google.com/uc?id={id}"
-    output = f"checkpoints/{name}"
+    output = checkpoints_directory + name
     gdown.download(url, output, quiet=False)
     print(f"Loaded {name}")

backend/services/wav2lip_inference.py

@@ -12,12 +12,14 @@ def run_command(cmd, workdir=None):
         cmd, text=True, cwd=workdir, stdout=subprocess.PIPE, stderr=subprocess.PIPE
     )
     if result.returncode != 0:
-        raise Exception(
-            f"Command {cmd[0]} failed with return code {result.returncode}."
-        )
+        error_details = f"Command '{' '.join(cmd)}' failed with return code {result.returncode}. "
+        error_details += f"STDOUT: {result.stdout} "
+        error_details += f"STDERR: {result.stderr}"
+        raise Exception(error_details)
     return result
 
 
+
 def convert_webm_to_wav(input_path, output_path):
     cmd = ["ffmpeg", "-y", "-i", input_path, output_path]
     run_command(cmd)
@@ -71,6 +73,6 @@ def run_wav2lip_inference(face_path, audio_path, outfile_path):
 
     if not os.path.exists(outfile_path):
         error_msg = result.stderr if result.stderr else result.stdout
-        raise Exception(f"Output video not found after processing. Error: {error_msg}")
+        raise Exception(f"Error: {error_msg}")
 
     return outfile_path

backend/tests/services/test_wav2lip_inference.py

@@ -88,10 +88,11 @@ def test_wav2lip_inference_processing_error(self, mock_run, mock_exists):
 
         with self.assertRaises(Exception) as context:
             run_wav2lip_inference(self.face, self.audio, self.outfile)
+
+        self.assertTrue(str(context.exception).startswith(f"Command 'ffmpeg -y -i test_audio.webm test_audio.wav' failed with return code 1."))
+        self.assertTrue("Error in Wav2Lip" in str(context.exception))
+
 
-        self.assertTrue(
-            "Command ffmpeg failed with return code 1." in str(context.exception)
-        )
 
     @patch("os.path.exists", return_value=True)
     @patch(

frontend/.env.example

@@ -0,0 +1 @@
+BACKEND_URL=http://localhost:5000