Update app.py

app.py

@@ -1,19 +1,22 @@
-from fastapi import FastAPI, UploadFile, File, HTTPException
-from predict_helper import predict_image
-from PIL import Image
-
-app = FastAPI(title="Car Damage Detection API", version="1.0")
-
-@app.get("/")
-def read_root():
-    return {"status": "API is running"}
-
-@app.post("/predict")
-async def predict(image: UploadFile = File(...)):
-    try:
-        img = Image.open(image.file)
-    except Exception:
-        raise HTTPException(status_code=400, detail="Invalid image file")
-
-    prediction = predict_image(img)
-    return prediction
+from fastapi import FastAPI, UploadFile, File, HTTPException
+from PIL import Image
+from predict_helper import predict_damage
+
+app = FastAPI(
+    title="Car Damage Detection API",
+    version="1.0"
+)
+
+@app.get("/")
+def health():
+    return {"status": "API is running"}
+
+@app.post("/predict_damage")
+async def predict(image: UploadFile = File(...)):
+    try:
+        img = Image.open(image.file)
+    except Exception:
+        raise HTTPException(status_code=400, detail="Invalid image file")
+
+    result = predict_damage(img)
+    return result