Update server.py

server.py

@@ -27,6 +27,7 @@ import timm
 import torch
 import uvicorn
 from fastapi import FastAPI, File, HTTPException, UploadFile
+from fastapi.responses import RedirectResponse
 from PIL import Image
 from pydantic import BaseModel, Field
 from pydantic_settings import BaseSettings
@@ -447,6 +448,8 @@ def create_app(settings: Settings) -> FastAPI:
         description="An API for tagging images using an ONNX model.",
         version="1.0.1",  # Incremented version
         lifespan=lifespan,
+         docs_url="/docs",
+        
     )
     app.state = AppState(settings)
     return app
@@ -467,7 +470,20 @@ def get_tagger(app: FastAPI) -> Tagger:
 def add_endpoints(app: FastAPI):
     tagger_dependency = lambda: get_tagger(app)
 
-    @app.post("/", response_model=BatchTaggingResponse, summary="Tag a batch of images")
+    # Root welcome/docs page
+    @app.get("/", include_in_schema=False)
+    async def root():
+        if app.docs_url:
+            return RedirectResponse(url=app.docs_url)
+        elif app.redoc_url:
+            return RedirectResponse(url=app.redoc_url)
+        return HTMLResponse(
+            content="<h1>Welcome to the Tagger API</h1><p>Use /batch to tag images.</p>",
+            status_code=200
+        )
+
+    # Tagging endpoint at /batch
+    @app.post("/batch", response_model=BatchTaggingResponse, summary="Tag a batch of images")
     async def tag_batch(
         tags_threshold: TaggerArgs = TaggerArgs(),
         file: UploadFile = File(
@@ -506,6 +522,7 @@ def add_endpoints(app: FastAPI):
             ),
         )
 
+    # Status endpoint
     @app.get("/status", response_model=StatusResponse, summary="Get server status")
     async def status():
         tagger = tagger_dependency()