Fix API routes: Use add_api_route to properly register FastAPI endpoints in Gradio

app.py

@@ -511,15 +511,13 @@ with gr.Blocks(
         outputs=[prompt_input, output],
     )
     
-    # Add FastAPI routes directly to Gradio's underlying FastAPI app
-    # This must be done inside the Blocks context
+    # Add FastAPI routes directly to Gradio's router
+    # This ensures they're registered and accessible alongside Gradio routes
     try:
         from fastapi.responses import JSONResponse
         
-        @gradio_app.app.post("/v1/generate")
-        async def fastapi_generate_endpoint(request):
-            """FastAPI endpoint mounted in Gradio app."""
-            from fastapi import Request
+        async def generate_handler(request):
+            """Handle POST /v1/generate requests."""
             try:
                 data = await request.json()
                 payload = GeneratePayload(**data)
@@ -534,22 +532,30 @@ with gr.Blocks(
                 from fastapi import HTTPException
                 raise HTTPException(status_code=500, detail=str(exc))
         
-        @gradio_app.app.get("/gradio")
-        async def fastapi_gradio_ui():
-            """FastAPI /gradio endpoint."""
-            return HTMLResponse(interactive_ui())
-        
-        @gradio_app.app.get("/")
-        async def fastapi_healthcheck():
-            """FastAPI healthcheck endpoint."""
-            return {
+        async def healthcheck_handler(request):
+            """Handle GET /api/health requests."""
+            return JSONResponse(content={
                 "status": "ok",
                 "model": MODEL_ID,
                 "strategy": ACTIVE_STRATEGY or "pending",
-            }
-        print("FastAPI routes added successfully")
+            })
+        
+        async def gradio_ui_handler(request):
+            """Handle GET /api/gradio requests."""
+            return HTMLResponse(interactive_ui())
+        
+        # Add routes to Gradio's router
+        # Use /api prefix to avoid conflicts with Gradio's routes
+        gradio_app.app.add_api_route("/v1/generate", generate_handler, methods=["POST"])
+        gradio_app.app.add_api_route("/api/health", healthcheck_handler, methods=["GET"])
+        gradio_app.app.add_api_route("/api/gradio", gradio_ui_handler, methods=["GET"])
+        # Also add /gradio for backward compatibility
+        gradio_app.app.add_api_route("/gradio", gradio_ui_handler, methods=["GET"])
+        print("FastAPI routes added successfully to Gradio router")
     except Exception as e:
         print(f"Warning: Could not add FastAPI routes: {e}")
+        import traceback
+        traceback.print_exc()
 
 # Set app to Gradio Blocks for Spaces - ZeroGPU requires Gradio SDK
 app = gradio_app