404 Handling

app.py

@@ -10,6 +10,7 @@ from PIL import Image, UnidentifiedImageError
 from io import BytesIO
 import base64
 import traceback
+from starlette.exceptions import HTTPException as StarletteHTTPException
 
 # ===============================
 # SIMPLE DPT MODEL (DEPTH ESTIMATION)
@@ -141,9 +142,6 @@ def _process_saree_core(base_image: Image.Image, pattern_image: Image.Image):
 # WRAPPER: ACCEPT BYTES OR BASE64
 # ===============================
 def process_saree(data):
-    """
-    Accepts [base_blob, pattern_blob] as bytes OR base64 strings
-    """
     if not isinstance(data, (list, tuple)) or len(data) != 2:
         raise HTTPException(status_code=422, detail="Expected an array with two elements: [base_blob, pattern_blob]")
 
@@ -161,7 +159,6 @@ def process_saree(data):
 
     except (base64.binascii.Error, UnidentifiedImageError) as e:
         raise HTTPException(status_code=422, detail=f"Invalid image data: {str(e)}")
-
     except Exception as e:
         raise HTTPException(status_code=400, detail=f"Error reading input images: {str(e)}")
 
@@ -211,6 +208,48 @@ async def predict_saree(request: Request):
             content={"error": "Processing Error", "details": str(e), "trace": tb}
         )
 
+# Alias for backward compatibility
+@app.post("/api/predict/")
+async def alias_predict(request: Request):
+    return await predict_saree(request)
+
+# ===============================
+# GLOBAL ERROR HANDLERS
+# ===============================
+@app.exception_handler(StarletteHTTPException)
+async def http_exception_handler(request: Request, exc: StarletteHTTPException):
+    if exc.status_code == 404:
+        return JSONResponse(
+            status_code=404,
+            content={
+                "error": "Endpoint Not Found",
+                "details": f"The requested URL {request.url.path} does not exist. "
+                           "Valid endpoints: /predict-saree or /api/predict/."
+            }
+        )
+    elif exc.status_code == 405:
+        return JSONResponse(
+            status_code=405,
+            content={
+                "error": "Method Not Allowed",
+                "details": f"Method {request.method} not allowed on {request.url.path}"
+            }
+        )
+    return JSONResponse(
+        status_code=exc.status_code,
+        content={"error": exc.detail or "HTTP Error"}
+    )
+
+@app.exception_handler(Exception)
+async def unhandled_exception_handler(request: Request, exc: Exception):
+    return JSONResponse(
+        status_code=500,
+        content={
+            "error": "Internal Server Error",
+            "details": str(exc)
+        }
+    )
+
 # Run (Hugging Face will call uvicorn automatically)
 if __name__ == "__main__":
     import uvicorn