Update app.py

app.py

@@ -0,0 +1,40 @@
+from fastapi import FastAPI, File, UploadFile
+from fastapi.staticfiles import StaticFiles
+from fastapi.responses import HTMLResponse
+import tensorflow as tf
+import numpy as np
+from PIL import Image
+import io
+
+app = FastAPI()
+
+# Load your model once at startup
+model = tf.keras.models.load_model("keras_model.h5")
+labels = ["Good Posture", "Bad Posture"]
+
+# Serve frontend files from /static
+app.mount("/static", StaticFiles(directory="static"), name="static")
+
+@app.get("/", response_class=HTMLResponse)
+async def root():
+    # Redirect to your static index.html
+    with open("static/index.html", "r") as f:
+        html = f.read()
+    return html
+
+@app.post("/predict")
+async def predict(file: UploadFile = File(...)):
+    # Read image bytes from upload
+    img_bytes = await file.read()
+    img = Image.open(io.BytesIO(img_bytes)).convert("RGB")
+    img = img.resize((224, 224))
+    img_array = np.array(img) / 255.0
+    img_array = np.expand_dims(img_array, axis=0)
+
+    # Predict
+    prediction = model.predict(img_array)[0]
+    top_index = np.argmax(prediction)
+    label = labels[top_index]
+    confidence = float(prediction[top_index])
+
+    return {"label": label, "confidence": confidence}