Spaces:

stephenebert
/

Image_Tagger

Sleeping

App Files Files Community

stephenebert commited on Aug 13, 2025

Commit

7d79380

verified ·

1 Parent(s): ca79a30

Update app.py

Browse files

Files changed (1) hide show

app.py +61 -21

app.py CHANGED Viewed

@@ -1,51 +1,58 @@
 from __future__ import annotations
 from fastapi import FastAPI, File, HTTPException, Query, UploadFile
-from fastapi.responses import HTMLResponse, JSONResponse
 from pydantic import BaseModel
 from typing import List
 from pathlib import Path
 from PIL import Image
 import io
-from tagger import tag_pil_image  # returns (caption, tags)
-app = FastAPI(title="Image Tagger API", version="0.3.0")
 class TagOut(BaseModel):
     filename: str
     caption: str
     tags: List[str]
-@app.get("/", response_class=HTMLResponse)
-def home() -> str:
-    return """
-    <html><head><meta charset="utf-8"><title>Image Tagger API</title></head>
-    <body style="font-family:system-ui,-apple-system,Segoe UI,Roboto,Ubuntu,sans-serif;max-width:720px;margin:40px auto;padding:0 16px">
-      <h2>Image Tagger API</h2>
-      <p>Use <a href="/docs">/docs</a> for Swagger, or upload here:</p>
-      <form action="/upload" method="post" enctype="multipart/form-data" style="display:grid;gap:12px">
-        <input type="file" name="file" accept="image/png,image/jpeg,image/webp" required />
-        <label>top_k:
-          <input type="number" name="top_k" value="5" min="1" max="20">
-        </label>
-        <button type="submit" style="padding:.6rem 1rem;border-radius:10px;border:1px solid #ddd;background:#111;color:#fff">Upload</button>
-      </form>
-    </body></html>
-    """
 @app.get("/healthz")
 def healthz():
     return {"ok": True}
 @app.post("/upload", response_model=TagOut)
 async def upload(
     file: UploadFile = File(...),
     top_k: int = Query(5, ge=1, le=20, description="Max number of tags"),
 ):
     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")
-    # Load image
     try:
         data = await file.read()
         img = Image.open(io.BytesIO(data)).convert("RGB")
@@ -54,9 +61,42 @@ async def upload(
     stem = Path(file.filename).stem or "upload"
     try:
         caption, tags = tag_pil_image(img, stem, top_k=top_k)
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Tagging failed: {e}")
     return JSONResponse({"filename": file.filename, "caption": caption, "tags": tags})

 from __future__ import annotations
+# FastAPI REST API + a Gradio UI mounted at /ui
+# Endpoints:
+#   GET  /healthz
+#   POST /upload  (multipart image -> {filename, caption, tags})
+# UI:
+#   /ui  (upload image, choose top_k, see caption + tags)
 from fastapi import FastAPI, File, HTTPException, Query, UploadFile
+from fastapi.responses import JSONResponse, RedirectResponse
 from pydantic import BaseModel
 from typing import List
 from pathlib import Path
 from PIL import Image
 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.0")
+# ---------- Pydantic model for OpenAPI ----------
 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(
     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")
     stem = Path(file.filename).stem or "upload"
+    # Tag
     try:
         caption, tags = tag_pil_image(img, stem, top_k=top_k)
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Tagging failed: {e}")
     return JSONResponse({"filename": file.filename, "caption": caption, "tags": tags})
+# ---------- Gradio UI (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, 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 = gr.Tags(label="Tags")
+    btn.click(_ui_tag, inputs=[inp, k], outputs=[cap, tags])
+# Mount Gradio on the same FastAPI app
+app = gr.mount_gradio_app(app, gr_app, path="/ui")