brought back grad

main.py

@@ -4,10 +4,12 @@ from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 from contextlib import asynccontextmanager
 from huggingface_hub import hf_hub_download
+import gradio as gr
 from PIL import Image
 
 from api.v1 import router as v1_router
 from models.model_loader import load_skin_condition_model
+from utils.predictor import predict_skin_condition
 
 os.environ["TF_ENABLE_ONEDNN_OPTS"] = "0"
 logger = logging.getLogger("uvicorn.error")
@@ -58,10 +60,6 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# --- basic hello route at “/” ---
-@app.get("/", tags=["Root"])
-async def hello():
-    return {"message": "Hello from the Skin Analyzer API 👋"}
 
 @app.get("/healthz", tags=["Health"])
 async def health_check():
@@ -69,3 +67,35 @@ async def health_check():
 
 # include your versioned REST API
 app.include_router(v1_router)
+
+
+# === Gradio UI Setup ===
+def predict_skin_condition_grad(image: Image.Image):
+    if image is None:
+        return "No image provided"
+
+    model = app.state.model
+
+    # Preprocess image
+    img = image.resize((224, 224)).convert("RGB")
+    img_array = np.array(img)
+
+    # Predict
+    prediction = predict_skin_condition(img_array, model)
+
+    
+    confidence = prediction.get("confidence")
+    label = prediction.get("condition")
+
+    return f"{label} ({confidence:.2%} confidence)"
+
+gradio_interface = gr.Interface(
+    fn=predict_skin_condition_grad,
+    inputs=gr.Image(type="pil", label="Upload a skin image"),
+    outputs=gr.Text(label="Prediction"),
+    title="Skin Analyzer",
+    description="Upload a photo of skin to detect conditions like acne, eczema, dryness, etc."
+)
+
+# Mount Gradio on root
+app = gr.mount_gradio_app(app, gradio_interface, path="")

utils/predictor.py

@@ -11,5 +11,5 @@ def predict_skin_condition(img_array, model):
 
     return {
         "condition": CLASS_NAMES[top_index],
-        "confidence": float(pred_probs[top_index]) * 100
+        "confidence": float(pred_probs[top_index])
     }