Update app.py

app.py

@@ -9,7 +9,6 @@ from fastapi import FastAPI, Request, Response
 from fastapi.staticfiles import StaticFiles
 import uvicorn
 import httpx
-from fastapi.middleware.base import BaseHTTPMiddleware
 
 import spaces
 from spaces.zero.client import _get_token
@@ -24,20 +23,21 @@ os.environ["GRADIO_ROOT_PATH"] = "/"
 # Create FastAPI app
 app = FastAPI()
 
-# Optional middleware to fix malformed SSR asset paths.
-class SSRPathRewriteMiddleware(BaseHTTPMiddleware):
-    async def dispatch(self, request: Request, call_next):
-        path = request.url.path
-        # If the path starts with the node port number attached (e.g., "/7861")
-        if path.startswith("/7861"):
-            # Replace "/7861" with "/_app" (or adjust to the correct base path)
-            fixed = path.replace("/7861", "/_app", 1)
-            request.scope["path"] = fixed
-        return await call_next(request)
+# Add middleware using the decorator
+@app.middleware("http")
+async def ssr_path_rewrite(request: Request, call_next):
+    path = request.scope.get("path", "")
+    if path.startswith("/7861"):
+        remainder = path[len("/7861"):]
+        if not remainder.startswith("/"):
+            remainder = "/" + remainder
+        new_path = "/_app" + remainder
+        request.scope["path"] = new_path
+        print(f"Rewrote path from {path} to {new_path}")
+    response = await call_next(request)
+    return response
 
-app.add_middleware(SSRPathRewriteMiddleware)
-
-# Create and configure static directory
+# Set up static files (if needed)
 static_dir = Path("./static")
 static_dir.mkdir(parents=True, exist_ok=True)
 app.mount("/static", StaticFiles(directory="static"), name="static")
@@ -64,7 +64,7 @@ with gr.Blocks() as demo:
     output = gr.Textbox(label="Output")
     submit_btn.click(fn=process_and_save, inputs=[text_input], outputs=output)
 
-# Mount the Gradio app with SSR mode, using node_port 7861.
+# Mount the Gradio app with SSR mode, using node_port 7861
 app = gr.mount_gradio_app(app, demo, path="/", ssr_mode=True, node_port=7861)
 
 if __name__ == "__main__":