Update app.py

app.py

@@ -2,11 +2,12 @@
 import logging
 import uuid
 from datetime import datetime, timezone
-from typing import Dict, Any, Optional
+from typing import Dict, Optional
 
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.openapi.docs import get_swagger_ui_html, get_redoc_html
+from fastapi.responses import RedirectResponse
 from pydantic import BaseModel
 import gradio as gr
 
@@ -137,10 +138,7 @@ async def record_outcome(decision_id: str, success: bool):
             return {"status": "ok", "decision_id": decision_id, "outcome": dec["outcome"]}
     return {"error": "decision not found"}
 
-# ========================= GRADIO UI (Main App) =========================
-# Gradio interface will be served at the root path ("/").
-# The FastAPI app is mounted under the "/api" subpath.
-
+# ========================= GRADIO UI =========================
 def get_risk_snapshot():
     try:
         intent = _DemoIntent()
@@ -218,34 +216,30 @@ with gr.Blocks(title="ARF v4 Demo") as demo:
     memory_btn.click(fn=get_memory_snapshot, outputs=memory_output)
     history_btn.click(fn=lambda: decision_history[-10:], outputs=decision_output)
 
-# ========================= Mount FastAPI under /api =========================
-# This mounts the Gradio interface onto the FastAPI app.
-# The resulting `app` is a FastAPI application that serves both:
-# - The API at `/api` (and its subpaths)
-# - The Gradio UI at the root `/`
+# ========================= Mount Gradio and Add Documentation Routes =========================
+# Mount Gradio at "/api" – this means Gradio will handle all requests starting with "/api".
 app = gr.mount_gradio_app(fastapi_app, demo, path="/api")
 
-# ========================= 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)
+# Add documentation routes at "/docs" (outside the Gradio mount path) to avoid conflict.
+@app.get("/docs", include_in_schema=False)
 async def swagger_ui():
     return get_swagger_ui_html(
-        openapi_url="/api/openapi.json",
+        openapi_url="/openapi.json",
         title="ARF API Docs"
     )
 
-@app.get("/api/redoc", include_in_schema=False)
+@app.get("/redoc", include_in_schema=False)
 async def redoc_ui():
     return get_redoc_html(
-        openapi_url="/api/openapi.json",
+        openapi_url="/openapi.json",
         title="ARF API ReDoc"
     )
 
-@app.get("/api/openapi.json", include_in_schema=False)
+@app.get("/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"]
+# Optional redirect from /api/docs to /docs for backward compatibility.
+@app.get("/api/docs", include_in_schema=False)
+async def redirect_docs():
+    return RedirectResponse(url="/docs")