made changes

Dockerfile

@@ -1,8 +1,7 @@
 # Optimized Dockerfile for Dance Movement Analyzer
-# Fixes MediaPipe model permission issues
+# Fixes MediaPipe models and FFmpeg H.264 encoder issues
 
-# FROM python:3.10-slim
-FROM python:3.10
+FROM python:3.10-slim
 
 # Set environment variables
 ENV PYTHONUNBUFFERED=1 \
@@ -11,31 +10,21 @@ ENV PYTHONUNBUFFERED=1 \
     PIP_DISABLE_PIP_VERSION_CHECK=1 \
     DEBIAN_FRONTEND=noninteractive
 
-# Install system dependencies
+# Install system dependencies including FFmpeg with H.264 support
 RUN apt-get update && apt-get install -y --no-install-recommends \
-    mesa-utils \
     libgl1 \
-    # libgl1-mesa-glx \
     libglib2.0-0 \
     libsm6 \
     libxext6 \
     libxrender-dev \
     libgomp1 \
     ffmpeg \
+    libavcodec-extra \
     curl \
-    tar \
-    xz-utils \
     && rm -rf /var/lib/apt/lists/*
 
-# Install static ffmpeg build (includes libx264)
-RUN echo "Installing static ffmpeg..." \
- && curl -L -o /tmp/ffmpeg.tar.xz "https://johnvansickle.com/ffmpeg/releases/ffmpeg-release-amd64-static.tar.xz" \
- && tar -xJf /tmp/ffmpeg.tar.xz -C /tmp \
- && cp /tmp/ffmpeg-*-static/ffmpeg /usr/local/bin/ffmpeg \
- && cp /tmp/ffmpeg-*-static/ffprobe /usr/local/bin/ffprobe \
- && chmod +x /usr/local/bin/ffmpeg /usr/local/bin/ffprobe \
- && rm -rf /tmp/ffmpeg* \
- && ffmpeg -version
+# Verify FFmpeg has H.264 encoder
+RUN ffmpeg -codecs | grep h264 || echo "Warning: H.264 encoder may not be available"
 
 # Create app directory
 WORKDIR /app

backend/app/config.py

@@ -78,8 +78,8 @@ class Config:
     BATCH_SIZE: int = int(os.getenv("BATCH_SIZE", 30))  # Frames per batch
     
     # Output settings
-    # OUTPUT_VIDEO_CODEC: str = os.getenv("OUTPUT_VIDEO_CODEC", "mp4v")
-    OUTPUT_VIDEO_CODEC: str = os.getenv("OUTPUT_VIDEO_CODEC", "avc1")
+    OUTPUT_VIDEO_CODEC: str = os.getenv("OUTPUT_VIDEO_CODEC", "mp4v")
+    # OUTPUT_VIDEO_CODEC: str = os.getenv("OUTPUT_VIDEO_CODEC", "avc1")
     OUTPUT_VIDEO_FPS: int = int(os.getenv("OUTPUT_VIDEO_FPS", 30))
     OUTPUT_VIDEO_QUALITY: int = int(os.getenv("OUTPUT_VIDEO_QUALITY", 90))
     

backend/app/video_processor.py

@@ -113,17 +113,58 @@ class VideoProcessor:
         cap = cv2.VideoCapture(str(video_path))
         
         # Setup video writer
-        # fourcc = cv2.VideoWriter_fourcc(*'mp4v')
-        fourcc = cv2.VideoWriter_fourcc(*Config.OUTPUT_VIDEO_CODEC)
-        out = cv2.VideoWriter(
-            str(output_path),
-            fourcc,
-            video_info['fps'],
-            (video_info['width'], video_info['height'])
-        )
-        
-        if not out.isOpened():
-            raise ValueError(f"Cannot create output video: {output_path}")
+        # # fourcc = cv2.VideoWriter_fourcc(*'mp4v')
+        # fourcc = cv2.VideoWriter_fourcc(*Config.OUTPUT_VIDEO_CODEC)
+        # out = cv2.VideoWriter(
+        #     str(output_path),
+        #     fourcc,
+        #     video_info['fps'],
+        #     (video_info['width'], video_info['height'])
+        # )
+        
+        # if not out.isOpened():
+        #     raise ValueError(f"Cannot create output video: {output_path}")
+
+        # Setup video writer with codec fallback
+        codecs_to_try = [
+            ('mp4v', 'MPEG-4'),          # Best compatibility
+            ('avc1', 'H.264'),           # High quality (if available)
+            ('XVID', 'Xvid'),            # Fallback option
+            ('MJPG', 'Motion JPEG')      # Last resort
+        ]
+
+        out = None
+        last_error = None
+
+        for codec_code, codec_name in codecs_to_try:
+            try:
+                logger.info(f"Trying codec: {codec_name} ({codec_code})")
+                fourcc = cv2.VideoWriter_fourcc(*codec_code)
+                out = cv2.VideoWriter(
+                    str(output_path),
+                    fourcc,
+                    video_info['fps'],
+                    (video_info['width'], video_info['height'])
+                )
+                
+                if out.isOpened():
+                    logger.info(f"✅ Successfully initialized VideoWriter with {codec_name} codec")
+                    break
+                else:
+                    logger.warning(f"Failed to open VideoWriter with {codec_name}")
+                    out.release()
+                    out = None
+            except Exception as e:
+                logger.warning(f"Error with {codec_name} codec: {e}")
+                last_error = e
+                if out:
+                    out.release()
+                out = None
+
+        if out is None or not out.isOpened():
+            error_msg = f"Cannot create output video with any codec. Last error: {last_error}"
+            logger.error(error_msg)
+            raise ValueError(error_msg)
         
         frame_number = 0
         processed_frames = 0