Update app.py

app.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-# FastAPI REST API + a Gradio UI mounted at /ui
+# FastAPI REST API + Gradio Interface mounted at /ui
 # Endpoints:
 #   GET  /healthz
 #   POST /upload   -> {filename, caption, tags}
@@ -18,28 +18,24 @@ 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.1")
+app = FastAPI(title="Image Tagger API", version="0.4.2")
 
-
-# ---------- Pydantic model for OpenAPI ----------
+# ---------- Pydantic model ----------
 class TagOut(BaseModel):
     filename: str
     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)
 
-
 @app.get("/healthz")
 def healthz():
     return {"ok": True}
 
-
 # ---------- REST endpoint ----------
 @app.post("/upload", response_model=TagOut)
 async def upload(
@@ -69,31 +65,28 @@ async def upload(
 
     return JSONResponse({"filename": file.filename, "caption": caption, "tags": tags})
 
-
-# ---------- Gradio UI (mounted at /ui) ----------
+# ---------- Gradio Interface (mounted at /ui) ----------
 def _ui_tag(image: Image.Image, top_k: int):
     if image is None:
         return "", ""
     caption, tags = tag_pil_image(image.convert("RGB"), "upload", top_k=top_k)
     return caption, ", ".join(tags)
 
-
-with gr.Blocks(title="Image Tagger", analytics_enabled=False) as gr_app:
-    gr.Markdown(
-        "## 🏷️ Image Tagger\n"
-        "Upload an image to get a caption and top-k tags. "
-        "Programmatic API is available at **/docs**."
-    )
-    with gr.Row():
-        with gr.Column(scale=1):
-            inp = gr.Image(type="pil", label="Upload image", sources=["upload"], height=360)
-            k = gr.Slider(1, 20, value=5, step=1, label="Top-k tags")
-            btn = gr.Button("Tag it")
-        with gr.Column(scale=1):
-            cap = gr.Textbox(label="Caption", lines=2)
-            tags_box = gr.Textbox(label="Tags (comma-separated)", lines=2)
-
-    btn.click(_ui_tag, inputs=[inp, k], outputs=[cap, tags_box])
+demo = gr.Interface(
+    fn=_ui_tag,
+    inputs=[
+        gr.Image(type="pil", label="Upload image"),
+        gr.Slider(1, 20, value=5, step=1, label="Top-k tags"),
+    ],
+    outputs=[
+        gr.Textbox(label="Caption", lines=2),
+        gr.Textbox(label="Tags (comma-separated)", lines=2),
+    ],
+    allow_flagging="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, gr_app, path="/ui")
+app = gr.mount_gradio_app(app, demo, path="/ui")
+