Update handler.py

handler.py

@@ -155,37 +155,50 @@ class EndpointHandler:
         Returns:
             Tuple of (video data URI, metadata dictionary)
         """
-        # Process video with Varnish
-        result = await self.varnish(
-            input_data=frames,
-            input_fps=config.fps,
-            upscale_factor=config.upscale_factor if config.upscale_factor > 1 else None,
-            enable_interpolation=config.enable_interpolation,
-            output_fps=config.fps
-        )
-        
-        # Convert to data URI
-        video_uri = await result.write(
-            output_type="data-uri",
-            output_format="mp4",
-            output_codec="h264",
-            output_quality=23
-        )
-        
-        # Collect metadata
-        metadata = {
-            "width": result.metadata.width,
-            "height": result.metadata.height,
-            "num_frames": result.metadata.frame_count,
-            "fps": result.metadata.fps,
-            "duration": result.metadata.duration,
-            "num_inference_steps": config.num_inference_steps,
-            "seed": config.seed,
-            "upscale_factor": config.upscale_factor,
-            "interpolation_enabled": config.enable_interpolation
-        }
-        
-        return video_uri, metadata
+        try:
+            logger.info(f"Original frames shape: {frames.shape}")
+                    
+            # Remove batch dimension if present
+            if len(frames.shape) == 5:
+                frames = frames.squeeze(0)  # Remove batch dimension
+            
+            logger.info(f"Processed frames shape: {frames.shape}")
+    
+            # Process video with Varnish
+            result = await self.varnish(
+                input_data=frames,
+                input_fps=config.fps,
+                output_fps=config.fps,
+                upscale_factor=config.upscale_factor if config.upscale_factor > 1 else None,
+                enable_interpolation=config.enable_interpolation
+            )
+            
+            # Convert to data URI
+            video_uri = await result.write(
+                output_type="data-uri",
+                output_format="mp4",
+                output_codec="h264",
+                output_quality=23
+            )
+            
+            # Collect metadata
+            metadata = {
+                "width": result.metadata.width,
+                "height": result.metadata.height,
+                "num_frames": result.metadata.frame_count,
+                "fps": result.metadata.fps,
+                "duration": result.metadata.duration,
+                "num_inference_steps": config.num_inference_steps,
+                "seed": config.seed,
+                "upscale_factor": config.upscale_factor,
+                "interpolation_enabled": config.enable_interpolation
+            }
+            
+            return video_uri, metadata
+    
+        except Exception as e:
+            logger.error(f"Error in process_frames: {str(e)}")
+            raise RuntimeError(f"Failed to process frames: {str(e)}")
 
     def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
         """Process incoming requests for video generation