Fix connection reset: Improve video streaming, increase timeouts, and add better error handling

Dockerfile

@@ -27,6 +27,6 @@ EXPOSE 7860
 ENV FACE_SWAP_API_KEY=""
 ENV OMP_NUM_THREADS=1
 
-CMD ["python3", "-m", "uvicorn", "fastapi_app:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["python3", "-m", "uvicorn", "fastapi_app:app", "--host", "0.0.0.0", "--port", "7860", "--timeout-keep-alive", "1800", "--timeout-graceful-shutdown", "30"]
 
 

fastapi_app.py

@@ -92,9 +92,12 @@ def swap_video_file(
     cv2.imwrite(src_path, src_bgr)
     shutil.copy(video_path, dst_video_path)
     frame_paths = extract_frames(dst_video_path, frames_dir)
-    logger.info("Extracted %s frames to %s", len(frame_paths), frames_dir)
+    total_frames = len(frame_paths)
+    logger.info("Extracted %s frames to %s", total_frames, frames_dir)
 
     for idx, frame_path in enumerate(frame_paths):
+        if idx % 50 == 0 or idx == total_frames - 1:
+            logger.info("Processing frame %s/%s (%.1f%%)", idx + 1, total_frames, (idx + 1) / total_frames * 100)
         swapped_name = f"swapped_{idx:05d}.jpg"
         out_path = os.path.join(swapped_dir, swapped_name)
         try:
@@ -392,18 +395,51 @@ async def swap_video(
         video_path = tmp_video.name
 
     try:
+        logger.info("Starting video processing...")
         output_path, _ = swap_video_file(src_bgr, source_face_idx, video_path, destination_face_idx, add_audio, workdir)
+        logger.info("Video processing completed: %s", output_path)
+        
+        if not os.path.exists(output_path):
+            raise HTTPException(status_code=500, detail="Output video file was not created")
+        
+        file_size = os.path.getsize(output_path)
+        logger.info("Output video size: %s bytes", file_size)
+        
     except Exception as exc:
         logger.exception("Video swap failed: %s", exc)
         shutil.rmtree(workdir, ignore_errors=True)
-        raise HTTPException(status_code=500, detail="Video swap failed") from exc
+        raise HTTPException(status_code=500, detail=f"Video swap failed: {str(exc)}") from exc
 
     def cleanup() -> None:
         shutil.rmtree(workdir, ignore_errors=True)
         logger.info("Cleaned up workdir %s", workdir)
 
     background_tasks.add_task(cleanup)
-    return StreamingResponse(open(output_path, "rb"), media_type="video/mp4")
+    
+    def iterfile():
+        try:
+            with open(output_path, "rb") as f:
+                while True:
+                    chunk = f.read(8192)  # Read in 8KB chunks
+                    if not chunk:
+                        break
+                    yield chunk
+        except Exception as e:
+            logger.error("Error streaming video file: %s", e)
+            raise
+    
+    file_size = os.path.getsize(output_path)
+    logger.info("Streaming video file: %s bytes", file_size)
+    
+    return StreamingResponse(
+        iterfile(),
+        media_type="video/mp4",
+        headers={
+            "Content-Disposition": "attachment; filename=swapped_video.mp4",
+            "Content-Length": str(file_size),
+            "Accept-Ranges": "bytes"
+        }
+    )
 
 
 @app.post("/swap/video/all-faces")
@@ -484,7 +520,29 @@ async def swap_video_all_faces(
         logger.info("Cleaned up workdir %s", workdir)
 
     background_tasks.add_task(cleanup)
-    return StreamingResponse(open(output_path, "rb"), media_type="video/mp4")
+    
+    def iterfile():
+        try:
+            with open(output_path, "rb") as f:
+                while True:
+                    chunk = f.read(8192)
+                    if not chunk:
+                        break
+                    yield chunk
+        except Exception as e:
+            logger.error("Error streaming video file: %s", e)
+            raise
+    
+    file_size = os.path.getsize(output_path)
+    return StreamingResponse(
+        iterfile(),
+        media_type="video/mp4",
+        headers={
+            "Content-Disposition": "attachment; filename=swapped_video.mp4",
+            "Content-Length": str(file_size),
+            "Accept-Ranges": "bytes"
+        }
+    )
 
 
 @app.post("/swap/video/custom-mapping")
@@ -573,7 +631,29 @@ async def swap_video_custom_mapping(
         logger.info("Cleaned up workdir %s", workdir)
 
     background_tasks.add_task(cleanup)
-    return StreamingResponse(open(output_path, "rb"), media_type="video/mp4")
+    
+    def iterfile():
+        try:
+            with open(output_path, "rb") as f:
+                while True:
+                    chunk = f.read(8192)
+                    if not chunk:
+                        break
+                    yield chunk
+        except Exception as e:
+            logger.error("Error streaming video file: %s", e)
+            raise
+    
+    file_size = os.path.getsize(output_path)
+    return StreamingResponse(
+        iterfile(),
+        media_type="video/mp4",
+        headers={
+            "Content-Disposition": "attachment; filename=swapped_video.mp4",
+            "Content-Length": str(file_size),
+            "Accept-Ranges": "bytes"
+        }
+    )
 
 
 @app.post("/swap/video/single-src-multi-video")