Update app.py

app.py

@@ -13,7 +13,13 @@ import requests
 from pathlib import Path
 import uvicorn
 
-app = FastAPI()
+# Initialize FastAPI with explicit docs settings
+app = FastAPI(
+    title="Face Swap API",
+    description="API for swapping faces in images.",
+    docs_url="/docs",  # Explicitly set docs URL
+    redoc_url="/redoc",  # Explicitly set redoc URL
+)
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -33,6 +39,11 @@ app.add_middleware(
 async def root():
     return {"message": "Welcome to the Face Swap API! Use /swap-face/ to swap faces or /docs to test the API."}
 
+# Add a health check endpoint
+@app.get("/health")
+async def health_check():
+    return {"status": "healthy"}
+
 # Global flag to prevent multiple downloads
 MODEL_DOWNLOADED = False
 
@@ -64,7 +75,7 @@ def download_model():
         logger.info("Model already exists at: %s", model_path)
         MODEL_DOWNLOADED = True
 
-# Use lifespan event handler instead of on_event
+# Use lifespan event handler
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     # Startup code