Fix Dockerfile build errors: improve dependency installation order and error handling

Dockerfile

@@ -33,6 +33,9 @@ RUN apt-get update && apt-get install -y \
     g++ \
     cmake \
     pkg-config \
+    curl \
+    wget \
+    ca-certificates \
     && rm -rf /var/lib/apt/lists/*
 
 # Create user for Hugging Face Spaces (following HF Spaces best practices)
@@ -54,31 +57,38 @@ RUN pip install --no-cache-dir \
     python-dotenv==1.0.0 \
     dnspython==2.4.2
 
-# Install numpy first (required by other packages)
-RUN pip install --no-cache-dir numpy==1.24.3
-
-# Install OpenCV first (needed by insightface)
-RUN pip install --no-cache-dir opencv-python-headless
-
-# Install ONNX Runtime
-RUN pip install --no-cache-dir onnxruntime==1.17.3
+# Install numpy first (required by other packages) - use compatible version
+RUN pip install --no-cache-dir "numpy>=1.24.0,<1.27.0"
 
 # Install utility dependencies (install these before insightface as they might be dependencies)
 RUN pip install --no-cache-dir \
     psutil==5.9.5 \
     tqdm==4.65.0 \
     Pillow \
-    moviepy==1.0.3 \
     imageio-ffmpeg \
     huggingface_hub>=0.23.0
 
-# Install insightface last - try multiple approaches, allow pip to find compatible version
-RUN pip install --no-cache-dir --no-build-isolation insightface==0.7.3 2>&1 || \
-    pip install --no-cache-dir insightface>=0.7.0,<0.8.0 2>&1 || \
-    pip install --no-cache-dir insightface 2>&1 || \
-    pip install --no-cache-dir --no-build-isolation git+https://github.com/deepinsight/insightface.git@v0.7.3 2>&1 || \
-    pip install --no-cache-dir git+https://github.com/deepinsight/insightface.git 2>&1 || \
-    (echo "All insightface installation methods failed" && exit 1)
+# Install OpenCV first (needed by insightface)
+RUN pip install --no-cache-dir opencv-python-headless
+
+# Install ONNX Runtime - try CPU version first as it's more compatible
+RUN (pip install --no-cache-dir onnxruntime==1.17.3 || \
+     pip install --no-cache-dir "onnxruntime>=1.16.0,<1.18.0" || \
+     pip install --no-cache-dir onnxruntime-cpu==1.17.3 || \
+     pip install --no-cache-dir "onnxruntime-cpu>=1.16.0,<1.18.0") && \
+    python -c "import onnxruntime; print(f'ONNX Runtime version: {onnxruntime.__version__}')" || \
+    (echo "ERROR: Failed to install ONNX Runtime. Check build logs above." && exit 1)
+
+# Install moviepy after other dependencies
+RUN pip install --no-cache-dir moviepy==1.0.3
+
+# Install insightface last - try multiple approaches with better error handling
+RUN pip install --no-cache-dir --no-build-isolation insightface==0.7.3 || \
+    pip install --no-cache-dir "insightface>=0.7.0,<0.8.0" || \
+    pip install --no-cache-dir insightface || \
+    pip install --no-cache-dir --no-build-isolation git+https://github.com/deepinsight/insightface.git@v0.7.3 || \
+    pip install --no-cache-dir git+https://github.com/deepinsight/insightface.git || \
+    (echo "ERROR: All insightface installation methods failed. Check build logs above for details." && exit 1)
 
 # Copy only essential application files
 COPY --chown=user:user api_server.py /app/

api/COLLECTIONS_INDEX.md

@@ -0,0 +1,189 @@
+# Face Swap Video API - Collections Index
+
+**Base URL:** `https://logicgoinfotechspaces-face-swap-video.hf.space`
+
+---
+
+## 📊 **MongoDB Collections**
+
+### **1. source_images**
+**Collection Name:** `source_images`  
+**Database:** `face_swap_video`
+
+**Schema:**
+```json
+{
+  "_id": "ObjectId",
+  "filename": "string",
+  "file_path": "string",
+  "uploaded_at": "datetime",
+  "status": "string",
+  "content_type": "string",
+  "file_size": "number"
+}
+```
+
+**API Endpoints:**
+- **Upload:** `POST https://logicgoinfotechspaces-face-swap-video.hf.space/api/source-image`
+- **List All:** `GET https://logicgoinfotechspaces-face-swap-video.hf.space/api/source-images`
+- **Get One:** Use `_id` from list response
+
+---
+
+### **2. target_videos**
+**Collection Name:** `target_videos`  
+**Database:** `face_swap_video`
+
+**Schema:**
+```json
+{
+  "_id": "ObjectId",
+  "filename": "string",
+  "file_path": "string",
+  "uploaded_at": "datetime",
+  "status": "string",
+  "content_type": "string",
+  "file_size": "number"
+}
+```
+
+**API Endpoints:**
+- **Upload:** `POST https://logicgoinfotechspaces-face-swap-video.hf.space/api/target-video`
+- **List All:** `GET https://logicgoinfotechspaces-face-swap-video.hf.space/api/target-videos`
+- **Get One:** Use `_id` from list response
+
+---
+
+### **3. result_videos**
+**Collection Name:** `result_videos`  
+**Database:** `face_swap_video`
+
+**Schema:**
+```json
+{
+  "_id": "ObjectId",
+  "source_image_path": "string",
+  "target_video_path": "string",
+  "result_file_path": "string",
+  "created_at": "datetime",
+  "status": "string",
+  "job_id": "string",
+  "processing_time": "number (optional)"
+}
+```
+
+**API Endpoints:**
+- **Download:** `GET https://logicgoinfotechspaces-face-swap-video.hf.space/api/result-video/{result_video_id}`
+- **List All:** `GET https://logicgoinfotechspaces-face-swap-video.hf.space/api/result-videos`
+- **Get Download URL:** Use `result_video_url` from job status when `status = "completed"`
+
+---
+
+### **4. processing_jobs**
+**Collection Name:** `processing_jobs`  
+**Database:** `face_swap_video`
+
+**Schema:**
+```json
+{
+  "_id": "ObjectId",
+  "job_id": "string (UUID)",
+  "source_image_id": "string",
+  "target_video_id": "string",
+  "status": "string (queued|processing|completed|failed)",
+  "created_at": "datetime",
+  "progress": "number (0-100)",
+  "result_video_id": "string (optional)",
+  "result_video_url": "string (optional)",
+  "error": "string (optional)"
+}
+```
+
+**API Endpoints:**
+- **Create Job:** `POST https://logicgoinfotechspaces-face-swap-video.hf.space/api/face-swap`
+- **Get Status:** `GET https://logicgoinfotechspaces-face-swap-video.hf.space/api/job/{job_id}`
+
+---
+
+## 🔗 **Complete API Endpoints Index**
+
+### **Upload Endpoints**
+```
+POST https://logicgoinfotechspaces-face-swap-video.hf.space/api/source-image
+POST https://logicgoinfotechspaces-face-swap-video.hf.space/api/target-video
+```
+
+### **Processing Endpoints**
+```
+POST https://logicgoinfotechspaces-face-swap-video.hf.space/api/face-swap
+GET  https://logicgoinfotechspaces-face-swap-video.hf.space/api/job/{job_id}
+```
+
+### **Download Endpoints**
+```
+GET https://logicgoinfotechspaces-face-swap-video.hf.space/api/result-video/{result_video_id}
+```
+
+### **List Endpoints**
+```
+GET https://logicgoinfotechspaces-face-swap-video.hf.space/api/source-images
+GET https://logicgoinfotechspaces-face-swap-video.hf.space/api/target-videos
+GET https://logicgoinfotechspaces-face-swap-video.hf.space/api/result-videos
+```
+
+### **Utility Endpoints**
+```
+GET https://logicgoinfotechspaces-face-swap-video.hf.space/api/health
+GET https://logicgoinfotechspaces-face-swap-video.hf.space/docs
+GET https://logicgoinfotechspaces-face-swap-video.hf.space/
+```
+
+---
+
+## 📝 **MongoDB Connection String**
+
+```
+mongodb+srv://itishalogicgo_db_user:HR837xi0B9yh2vZK@cluster0.jeeytpz.mongodb.net/?retryWrites=true&w=majority&appName=Cluster0
+```
+
+**Database:** `face_swap_video`
+
+**Collections:**
+1. `source_images`
+2. `target_videos`
+3. `result_videos`
+4. `processing_jobs`
+
+---
+
+## 🔄 **Workflow Example**
+
+1. **Upload Source Image** → Get `source_image_id`
+2. **Upload Target Video** → Get `target_video_id`
+3. **Start Face Swap** → Get `job_id`
+4. **Poll Job Status** → Get `result_video_url` when `status = "completed"`
+5. **Download Result** → Use `result_video_url` directly
+
+---
+
+## 📋 **Quick Reference**
+
+| Collection | Insert Endpoint | List Endpoint | Get One |
+|------------|----------------|---------------|---------|
+| `source_images` | `POST /api/source-image` | `GET /api/source-images` | Use `_id` |
+| `target_videos` | `POST /api/target-video` | `GET /api/target-videos` | Use `_id` |
+| `result_videos` | Auto via job completion | `GET /api/result-videos` | `GET /api/result-video/{id}` |
+| `processing_jobs` | `POST /api/face-swap` | N/A | `GET /api/job/{job_id}` |
+
+---
+
+## 🌐 **MongoDB Atlas Dashboard**
+
+Access your collections at:
+```
+https://cloud.mongodb.com/
+```
+
+**Cluster:** `Cluster0`  
+**Database:** `face_swap_video`  
+**Collections:** 4 collections listed above

api/Face_Swap_Video_API.postman_collection.json

@@ -0,0 +1,332 @@
+{
+	"info": {
+		"_postman_id": "face-swap-video-api",
+		"name": "Face Swap Video API",
+		"description": "Complete API collection for Face Swap Video processing with MongoDB storage",
+		"schema": "https://schema.getpostman.com/json/collection/v2.1.0/collection.json"
+	},
+	"item": [
+		{
+			"name": "Source Image",
+			"item": [
+				{
+					"name": "Upload Source Image",
+					"request": {
+						"method": "POST",
+						"header": [
+							{
+								"key": "accept",
+								"value": "application/json"
+							}
+						],
+						"body": {
+							"mode": "formdata",
+							"formdata": [
+								{
+									"key": "file",
+									"type": "file",
+									"src": []
+								}
+							]
+						},
+						"url": {
+							"raw": "https://logicgoinfotechspaces-face-swap-video.hf.space/api/source-image",
+							"protocol": "https",
+							"host": [
+								"logicgoinfotechspaces-face-swap-video",
+								"hf",
+								"space"
+							],
+							"path": [
+								"api",
+								"source-image"
+							]
+						},
+						"description": "Upload and store source image in MongoDB"
+					},
+					"response": []
+				},
+				{
+					"name": "List All Source Images",
+					"request": {
+						"method": "GET",
+						"header": [],
+						"url": {
+							"raw": "https://logicgoinfotechspaces-face-swap-video.hf.space/api/source-images",
+							"protocol": "https",
+							"host": [
+								"logicgoinfotechspaces-face-swap-video",
+								"hf",
+								"space"
+							],
+							"path": [
+								"api",
+								"source-images"
+							]
+						},
+						"description": "List all uploaded source images"
+					},
+					"response": []
+				}
+			],
+			"description": "Source image management endpoints"
+		},
+		{
+			"name": "Target Video",
+			"item": [
+				{
+					"name": "Upload Target Video",
+					"request": {
+						"method": "POST",
+						"header": [
+							{
+								"key": "accept",
+								"value": "application/json"
+							}
+						],
+						"body": {
+							"mode": "formdata",
+							"formdata": [
+								{
+									"key": "file",
+									"type": "file",
+									"src": []
+								}
+							]
+						},
+						"url": {
+							"raw": "https://logicgoinfotechspaces-face-swap-video.hf.space/api/target-video",
+							"protocol": "https",
+							"host": [
+								"logicgoinfotechspaces-face-swap-video",
+								"hf",
+								"space"
+							],
+							"path": [
+								"api",
+								"target-video"
+							]
+						},
+						"description": "Upload and store target video in MongoDB"
+					},
+					"response": []
+				},
+				{
+					"name": "List All Target Videos",
+					"request": {
+						"method": "GET",
+						"header": [],
+						"url": {
+							"raw": "https://logicgoinfotechspaces-face-swap-video.hf.space/api/target-videos",
+							"protocol": "https",
+							"host": [
+								"logicgoinfotechspaces-face-swap-video",
+								"hf",
+								"space"
+							],
+							"path": [
+								"api",
+								"target-videos"
+							]
+						},
+						"description": "List all uploaded target videos"
+					},
+					"response": []
+				}
+			],
+			"description": "Target video management endpoints"
+		},
+		{
+			"name": "Face Swap Processing",
+			"item": [
+				{
+					"name": "Start Face Swap",
+					"request": {
+						"method": "POST",
+						"header": [
+							{
+								"key": "Content-Type",
+								"value": "application/json"
+							}
+						],
+						"body": {
+							"mode": "raw",
+							"raw": "{\n  \"source_image_id\": \"SOURCE_IMAGE_ID\",\n  \"target_video_id\": \"TARGET_VIDEO_ID\"\n}",
+							"options": {
+								"raw": {
+									"language": "json"
+								}
+							}
+						},
+						"url": {
+							"raw": "https://logicgoinfotechspaces-face-swap-video.hf.space/api/face-swap",
+							"protocol": "https",
+							"host": [
+								"logicgoinfotechspaces-face-swap-video",
+								"hf",
+								"space"
+							],
+							"path": [
+								"api",
+								"face-swap"
+							]
+						},
+						"description": "Start face swap processing. Returns job_id for status tracking."
+					},
+					"response": []
+				},
+				{
+					"name": "Get Job Status",
+					"request": {
+						"method": "GET",
+						"header": [],
+						"url": {
+							"raw": "https://logicgoinfotechspaces-face-swap-video.hf.space/api/job/{{job_id}}",
+							"protocol": "https",
+							"host": [
+								"logicgoinfotechspaces-face-swap-video",
+								"hf",
+								"space"
+							],
+							"path": [
+								"api",
+								"job",
+								"{{job_id}}"
+							]
+						},
+						"description": "Get job status. When status='completed', result_video_url will contain the download URL."
+					},
+					"response": []
+				}
+			],
+			"description": "Face swap processing endpoints"
+		},
+		{
+			"name": "Result Videos",
+			"item": [
+				{
+					"name": "Download Result Video",
+					"request": {
+						"method": "GET",
+						"header": [],
+						"url": {
+							"raw": "https://logicgoinfotechspaces-face-swap-video.hf.space/api/result-video/{{result_video_id}}",
+							"protocol": "https",
+							"host": [
+								"logicgoinfotechspaces-face-swap-video",
+								"hf",
+								"space"
+							],
+							"path": [
+								"api",
+								"result-video",
+								"{{result_video_id}}"
+							]
+						},
+						"description": "Download result video file. Use result_video_url from job status when completed."
+					},
+					"response": []
+				},
+				{
+					"name": "List All Result Videos",
+					"request": {
+						"method": "GET",
+						"header": [],
+						"url": {
+							"raw": "https://logicgoinfotechspaces-face-swap-video.hf.space/api/result-videos",
+							"protocol": "https",
+							"host": [
+								"logicgoinfotechspaces-face-swap-video",
+								"hf",
+								"space"
+							],
+							"path": [
+								"api",
+								"result-videos"
+							]
+						},
+						"description": "List all processed result videos"
+					},
+					"response": []
+				}
+			],
+			"description": "Result video management endpoints"
+		},
+		{
+			"name": "Utility",
+			"item": [
+				{
+					"name": "Health Check",
+					"request": {
+						"method": "GET",
+						"header": [],
+						"url": {
+							"raw": "https://logicgoinfotechspaces-face-swap-video.hf.space/api/health",
+							"protocol": "https",
+							"host": [
+								"logicgoinfotechspaces-face-swap-video",
+								"hf",
+								"space"
+							],
+							"path": [
+								"api",
+								"health"
+							]
+						},
+						"description": "Check API health status"
+					},
+					"response": []
+				},
+				{
+					"name": "API Root",
+					"request": {
+						"method": "GET",
+						"header": [],
+						"url": {
+							"raw": "https://logicgoinfotechspaces-face-swap-video.hf.space/",
+							"protocol": "https",
+							"host": [
+								"logicgoinfotechspaces-face-swap-video",
+								"hf",
+								"space"
+							],
+							"path": [
+								""
+							]
+						},
+						"description": "API root endpoint"
+					},
+					"response": []
+				}
+			],
+			"description": "Utility endpoints"
+		}
+	],
+	"variable": [
+		{
+			"key": "base_url",
+			"value": "https://logicgoinfotechspaces-face-swap-video.hf.space",
+			"type": "string"
+		},
+		{
+			"key": "source_image_id",
+			"value": "",
+			"type": "string"
+		},
+		{
+			"key": "target_video_id",
+			"value": "",
+			"type": "string"
+		},
+		{
+			"key": "job_id",
+			"value": "",
+			"type": "string"
+		},
+		{
+			"key": "result_video_id",
+			"value": "",
+			"type": "string"
+		}
+	]
+}

api/main.py

@@ -11,7 +11,6 @@ from bson import ObjectId
 import json
 import shutil
 from pathlib import Path
-from gridfs import GridFS
 from fastapi.responses import FileResponse, StreamingResponse, JSONResponse
 
 # Import face swap functionality
@@ -35,7 +34,6 @@ MONGODB_URL = os.getenv("MONGODB_URL", "mongodb+srv://itishalogicgo_db_user:HR83
 DATABASE_NAME = "face_swap_video"
 client = motor.motor_asyncio.AsyncIOMotorClient(MONGODB_URL)
 db = client[DATABASE_NAME]
-fs = GridFS(db)
 
 # Collections
 source_images_collection = db["source_images"]
@@ -362,40 +360,9 @@ async def list_result_videos():
 async def api_health():
     return {"status": "ok", "time": datetime.utcnow().isoformat()}
 
-# -------- Optional: GridFS variants to store binaries in MongoDB -------- #
-
-async def _gridfs_put(file: UploadFile, metadata: dict) -> str:
-    data = await file.read()
-    file_id = fs.put(data, filename=file.filename, content_type=file.content_type, metadata=metadata)
-    return str(file_id)
-
-@app.post("/api/source-image/gridfs")
-async def upload_source_image_gridfs(file: UploadFile = File(...)):
-    if not file.content_type.startswith('image/'):
-        raise HTTPException(status_code=400, detail="File must be an image")
-    file_id = await _gridfs_put(file, {"kind": "source_image", "uploaded_at": datetime.utcnow()})
-    doc = {"gridfs_id": file_id, "filename": file.filename, "uploaded_at": datetime.utcnow(), "status": "uploaded"}
-    result = await source_images_collection.insert_one(doc)
-    return {"id": str(result.inserted_id), "gridfs_id": file_id}
-
-@app.post("/api/target-video/gridfs")
-async def upload_target_video_gridfs(file: UploadFile = File(...)):
-    if not file.content_type.startswith('video/'):
-        raise HTTPException(status_code=400, detail="File must be a video")
-    file_id = await _gridfs_put(file, {"kind": "target_video", "uploaded_at": datetime.utcnow()})
-    doc = {"gridfs_id": file_id, "filename": file.filename, "uploaded_at": datetime.utcnow(), "status": "uploaded"}
-    result = await target_videos_collection.insert_one(doc)
-    return {"id": str(result.inserted_id), "gridfs_id": file_id}
-
-@app.get("/api/file/{file_id}")
-async def get_file_from_gridfs(file_id: str):
-    try:
-        oid = ObjectId(file_id)
-    except Exception:
-        raise HTTPException(status_code=400, detail="Invalid file id")
-    gf = fs.get(oid)
-    headers = {"Content-Disposition": f"inline; filename={gf.filename}"}
-    return StreamingResponse(gf, media_type=gf.content_type or "application/octet-stream", headers=headers)
+# -------- Optional: GridFS variants (disabled - using file system storage) -------- #
+# GridFS endpoints removed - using file system storage instead
+# If needed, can be re-enabled with motor's GridFSBucketAsync for async operations
 
 @app.get("/")
 async def root():

docker-compose.yml

@@ -0,0 +1,22 @@
+version: '3.8'
+
+services:
+  face-swap-api:
+    build: .
+    ports:
+      - "8000:8000"
+    environment:
+      - MONGODB_URL=${MONGODB_URL:-mongodb+srv://itishalogicgo_db_user:HR837xi0B9yh2vZK@cluster0.jeeytpz.mongodb.net/?retryWrites=true&w=majority&appName=Cluster0}
+      - BASE_URL=http://localhost:8000
+      - CUDA_VISIBLE_DEVICES=0
+    volumes:
+      - ./uploads:/app/uploads
+      - ./DeepFakeAI/.assets:/app/DeepFakeAI/.assets
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: 1
+              capabilities: [gpu]
+    restart: unless-stopped