Update app.py

app.py

@@ -9,19 +9,34 @@ 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
 
-# Set environment variables for SSR
+# Set environment variables
 os.environ["GRADIO_SSR_MODE"] = "True"
 os.environ["GRADIO_SERVER_PORT"] = "7860"
 os.environ["GRADIO_SERVER_NAME"] = "0.0.0.0"
-os.environ["GRADIO_NODE_SERVER_NAME"] = "0.0.0.0"
+os.environ["GRADIO_NODE_SERVER_NAME"] = "127.0.0.1"  # host only
+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)
+
+app.add_middleware(SSRPathRewriteMiddleware)
+
 # Create and configure static directory
 static_dir = Path("./static")
 static_dir.mkdir(parents=True, exist_ok=True)
@@ -47,24 +62,10 @@ with gr.Blocks() as demo:
     text_input = gr.Textbox(label="Enter some text")
     submit_btn = gr.Button("Process and Download")
     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)
 
-# Manual reverse proxy for SSR assets
-@app.get("/_app/{full_path:path}")
-async def proxy_to_node(full_path: str, request: Request):
-    node_server = "http://127.0.0.1:7861"
-    url = f"{node_server}/_app/{full_path}"
-    async with httpx.AsyncClient() as client:
-        node_response = await client.get(url)
-    return Response(
-        content=node_response.content,
-        status_code=node_response.status_code,
-        media_type=node_response.headers.get("content-type"),
-    )
-
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)