Update app.py

app.py

@@ -1,15 +1,19 @@
 from fastapi import FastAPI, File, UploadFile
-from fastapi.responses import Response
+from fastapi.responses import Response, JSONResponse
 from insightface.app import FaceAnalysis
 from rembg import remove
 from PIL import Image
 import numpy as np
 import io
 
-
 app = FastAPI()
 
-# Load face detection model (SCRFD)
+# Simple homepage (fixes 404 on HF Spaces)
+@app.get("/")
+def home():
+    return {"message": "ID Photo API is running. Use POST /process"}
+
+# Load face detection model
 face_app = FaceAnalysis(name="buffalo_l", providers=["CPUExecutionProvider"])
 face_app.prepare(ctx_id=0, det_size=(640, 640))
 
@@ -26,9 +30,8 @@ async def process_img(file: UploadFile = File(...)):
 
     faces = face_app.get(np_img)
     if not faces:
-        return {"error": "No face detected"}
+        return JSONResponse({"error": "No face detected"}, status_code=400)
 
-    # Extract aligned face from insightface
     face = faces[0]
     aligned = Image.fromarray(face.normed)
 
@@ -39,10 +42,10 @@ async def process_img(file: UploadFile = File(...)):
     white = Image.new("RGB", no_bg.size, (255, 255, 255))
     white.paste(no_bg, mask=no_bg.split()[3])
 
-    # Resize to official 4x6 photo size
+    # Resize
     final_img = resize_to_4x6(white)
 
-    # Return JPEG
+    # Output
     buf = io.BytesIO()
     final_img.save(buf, format="JPEG")
     buf.seek(0)