Update api.py

api.py

@@ -1,41 +1,47 @@
-# api.py
-import os, tempfile
-from fastapi import FastAPI, UploadFile, File
-from fastapi.responses import JSONResponse
-from inference import predict
-
-app = FastAPI()
-
-@app.get("/health")
-def health():
-    return {"ok": True}
-
-@app.post("/predict")
-async def predict_endpoint(file: UploadFile = File(...)):
-    tmp_path = None
-    try:
-        # Lưu tạm file ảnh
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".jpg") as tmp:
-            tmp.write(await file.read())
-            tmp_path = tmp.name
-
-        label_en, conf = predict(tmp_path)
-
-        # Mapping theo UI hiện tại của bạn
-        map_vi  = {"FRESH": "Tươi", "HALF-FRESH": "Vừa", "SPOILED": "Hỏng"}
-        map_pct = {"FRESH": 100,     "HALF-FRESH": 50,     "SPOILED": 0}
-
-        result = {
-            "meat_type": "pork",  # sửa nếu bạn phân loại nhiều loại thịt
-            "freshness_percent": map_pct.get(label_en, 50),
-            "label_vi": map_vi.get(label_en, label_en),
-            "label_en": label_en,
-            "confidence": conf
-        }
-        return JSONResponse(result)
-    except Exception as e:
-        return JSONResponse({"error": str(e)}, status_code=500)
-    finally:
-        if tmp_path:
-            try: os.remove(tmp_path)
-            except Exception: pass
+# api.py
+import os, tempfile
+from fastapi import FastAPI, UploadFile, File
+from fastapi.responses import JSONResponse
+from inference import predict
+from fastapi.responses import JSONResponse, PlainTextResponse
+
+
+app = FastAPI()
+
+@app.get("/", response_class=PlainTextResponse)
+def root():
+    return "Meat Freshness API is running. Try /health or POST /predict"
+    
+@app.get("/health")
+def health():
+    return {"ok": True}
+
+@app.post("/predict")
+async def predict_endpoint(file: UploadFile = File(...)):
+    tmp_path = None
+    try:
+        # Lưu tạm file ảnh
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".jpg") as tmp:
+            tmp.write(await file.read())
+            tmp_path = tmp.name
+
+        label_en, conf = predict(tmp_path)
+
+        # Mapping theo UI hiện tại của bạn
+        map_vi  = {"FRESH": "Tươi", "HALF-FRESH": "Vừa", "SPOILED": "Hỏng"}
+        map_pct = {"FRESH": 100,     "HALF-FRESH": 50,     "SPOILED": 0}
+
+        result = {
+            "meat_type": "pork",  # sửa nếu bạn phân loại nhiều loại thịt
+            "freshness_percent": map_pct.get(label_en, 50),
+            "label_vi": map_vi.get(label_en, label_en),
+            "label_en": label_en,
+            "confidence": conf
+        }
+        return JSONResponse(result)
+    except Exception as e:
+        return JSONResponse({"error": str(e)}, status_code=500)
+    finally:
+        if tmp_path:
+            try: os.remove(tmp_path)
+            except Exception: pass