Update app.py

app.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 # FastAPI REST API + a Gradio UI mounted at /ui
 # Endpoints:
 #   GET  /healthz
-#   POST /upload  (multipart image -> {filename, caption, tags})
+#   POST /upload   -> {filename, caption, tags}
 # UI:
 #   /ui  (upload image, choose top_k, see caption + tags)
 
@@ -18,7 +18,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.0")
+app = FastAPI(title="Image Tagger API", version="0.4.1")
 
 
 # ---------- Pydantic model for OpenAPI ----------
@@ -73,9 +73,9 @@ async def upload(
 # ---------- Gradio UI (mounted at /ui) ----------
 def _ui_tag(image: Image.Image, top_k: int):
     if image is None:
-        return "", []
+        return "", ""
     caption, tags = tag_pil_image(image.convert("RGB"), "upload", top_k=top_k)
-    return caption, tags
+    return caption, ", ".join(tags)
 
 
 with gr.Blocks(title="Image Tagger", analytics_enabled=False) as gr_app:
@@ -86,17 +86,14 @@ with gr.Blocks(title="Image Tagger", analytics_enabled=False) as gr_app:
     )
     with gr.Row():
         with gr.Column(scale=1):
-            inp = gr.Image(
-                type="pil", label="Upload image", sources=["upload"], height=360
-            )
+            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")
+            tags_box = gr.Textbox(label="Tags (comma-separated)", lines=2)
 
-    btn.click(_ui_tag, inputs=[inp, k], outputs=[cap, tags])
+    btn.click(_ui_tag, inputs=[inp, k], outputs=[cap, tags_box])
 
 # Mount Gradio on the same FastAPI app
 app = gr.mount_gradio_app(app, gr_app, path="/ui")
-