Update app.py

app.py

@@ -6,6 +6,7 @@ from typing import Dict, Any, Optional
 
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
+from fastapi.openapi.docs import get_swagger_ui_html, get_redoc_html
 from pydantic import BaseModel
 import gradio as gr
 
@@ -224,5 +225,27 @@ with gr.Blocks(title="ARF v4 Demo") as demo:
 # - The Gradio UI at the root `/`
 app = gr.mount_gradio_app(fastapi_app, demo, path="/api")
 
-# Note: No manual server start! The Dockerfile will run uvicorn on this `app` object.
-# Make sure your Dockerfile's CMD is: CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+# ========================= Add Documentation Routes =========================
+# The mounted fastapi_app's docs are not automatically exposed.
+# We manually add routes to serve Swagger UI and ReDoc at /api/docs and /api/redoc,
+# pointing to the OpenAPI schema from the mounted app.
+@app.get("/api/docs", include_in_schema=False)
+async def swagger_ui():
+    return get_swagger_ui_html(
+        openapi_url="/api/openapi.json",
+        title="ARF API Docs"
+    )
+
+@app.get("/api/redoc", include_in_schema=False)
+async def redoc_ui():
+    return get_redoc_html(
+        openapi_url="/api/openapi.json",
+        title="ARF API ReDoc"
+    )
+
+@app.get("/api/openapi.json", include_in_schema=False)
+async def openapi():
+    return fastapi_app.openapi()
+
+# Note: The Dockerfile should run uvicorn on this `app` object.
+# Ensure your Dockerfile's CMD is: CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]