Update app.py

app.py

@@ -1,14 +1,16 @@
 from __future__ import annotations
 
-# FastAPI REST API + Gradio Interface mounted at /ui
+# FastAPI REST API + Gradio UI at /
 # Endpoints:
 #   GET  /healthz
 #   POST /upload   -> {filename, caption, tags}
 # UI:
-#   /ui  (upload image, choose top_k, see caption + tags)
+#   /  (upload image, choose top_k, see caption + tags)
+# Docs:
+#   /docs
 
 from fastapi import FastAPI, File, HTTPException, Query, UploadFile
-from fastapi.responses import JSONResponse, RedirectResponse
+from fastapi.responses import JSONResponse
 from pydantic import BaseModel
 from typing import List
 from pathlib import Path
@@ -18,7 +20,7 @@ import io
 import gradio as gr
 from tagger import tag_pil_image  # returns (caption: str, tags: List[str])
 
-app = FastAPI(title="Image Tagger API", version="0.4.2")
+app = FastAPI(title="Image Tagger API", version="0.4.3")
 
 # ---------- Pydantic model ----------
 class TagOut(BaseModel):
@@ -26,12 +28,7 @@ class TagOut(BaseModel):
     caption: str
     tags: List[str]
 
-# ---------- Basic routes ----------
-@app.get("/", include_in_schema=False)
-def root_redirect():
-    # Send users to the nicer UI
-    return RedirectResponse(url="/ui", status_code=302)
-
+# ---------- Health ----------
 @app.get("/healthz")
 def healthz():
     return {"ok": True}
@@ -42,13 +39,11 @@ async def upload(
     file: UploadFile = File(...),
     top_k: int = Query(5, ge=1, le=20, description="Max number of tags"),
 ):
-    # MIME guard
     if file.content_type not in {"image/png", "image/jpeg", "image/webp"}:
         raise HTTPException(
             status_code=415, detail="Only PNG, JPEG, or WebP images are supported"
         )
 
-    # Decode image
     try:
         data = await file.read()
         img = Image.open(io.BytesIO(data)).convert("RGB")
@@ -57,7 +52,6 @@ async def upload(
 
     stem = Path(file.filename).stem or "upload"
 
-    # Tag
     try:
         caption, tags = tag_pil_image(img, stem, top_k=top_k)
     except Exception as e:
@@ -65,7 +59,7 @@ async def upload(
 
     return JSONResponse({"filename": file.filename, "caption": caption, "tags": tags})
 
-# ---------- Gradio Interface (mounted at /ui) ----------
+# ---------- Gradio UI at root ----------
 def _ui_tag(image: Image.Image, top_k: int):
     if image is None:
         return "", ""
@@ -82,11 +76,11 @@ demo = gr.Interface(
         gr.Textbox(label="Caption", lines=2),
         gr.Textbox(label="Tags (comma-separated)", lines=2),
     ],
-    allow_flagging="never",
+    flagging_mode="never",
     title="Image Tagger",
     description="Upload an image to get a caption and top-k tags. Programmatic API at /docs.",
 )
 
-# Mount Gradio on the same FastAPI app
-app = gr.mount_gradio_app(app, demo, path="/ui")
+# Mount Gradio on the same FastAPI app at root (/) to avoid redirects
+app = gr.mount_gradio_app(app, demo, path="/")