Fix: Add routes after Blocks context when router is ready

app.py

@@ -548,12 +548,15 @@ with gr.Blocks(
             html_content = interactive_ui()
             return HTMLResponse(content=html_content)
         
-        # Insert routes at the beginning of Gradio's router to ensure they're processed first
-        # This prevents Gradio's catch-all routes from intercepting our API endpoints
-        gradio_app.app.router.routes.insert(0, Route("/health", health_handler, methods=["GET"]))
-        gradio_app.app.router.routes.insert(0, Route("/v1/generate", generate_handler, methods=["POST"]))
-        gradio_app.app.router.routes.insert(0, Route("/console", console_handler, methods=["GET"]))
-        print(f"FastAPI routes mounted successfully on Gradio app. Total routes: {len(gradio_app.app.router.routes)}")
+        # Store handlers to add routes after Blocks context (router is ready then)
+        # Store in a list that will be accessible after the Blocks context
+        if not hasattr(gradio_app, '_api_handlers'):
+            gradio_app._api_handlers = []
+        gradio_app._api_handlers.extend([
+            ("/health", health_handler, ["GET"]),
+            ("/v1/generate", generate_handler, ["POST"]),
+            ("/console", console_handler, ["GET"]),
+        ])
     except Exception as e:
         print(f"Warning: Could not mount FastAPI routes: {e}")
         import traceback
@@ -562,6 +565,20 @@ with gr.Blocks(
 # Enable queued execution so ZeroGPU can schedule GPU work reliably
 gradio_app.queue(max_size=8)
 
+# Add routes after Blocks context when router is definitely ready
+try:
+    if hasattr(gradio_app, '_api_handlers'):
+        from starlette.routing import Route
+        for path, handler, methods in gradio_app._api_handlers:
+            gradio_app.app.router.routes.insert(0, Route(path, handler, methods=methods))
+        print(f"FastAPI routes added after Blocks context. Total routes: {len(gradio_app.app.router.routes)}")
+        # Clean up
+        del gradio_app._api_handlers
+except Exception as e:
+    print(f"Warning: Could not add routes after Blocks context: {e}")
+    import traceback
+    traceback.print_exc()
+
 # Set app to Gradio for Spaces compatibility (sdk: gradio requires Gradio app)
 # Spaces will handle running the server automatically
 app = gradio_app