Update app.py

app.py

@@ -13,31 +13,18 @@ import httpx
 import spaces
 from spaces.zero.client import _get_token
 
-# Set environment variables
+# Set environment variables so Gradio builds correct URLs:
 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"] = "127.0.0.1"  # host only
+# IMPORTANT: include both the node port and a trailing slash!
+os.environ["GRADIO_NODE_SERVER_NAME"] = "127.0.0.1:7861/"
 os.environ["GRADIO_ROOT_PATH"] = "/"
 
 # Create FastAPI app
 app = FastAPI()
 
-# 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
-
-# Set up static files (if needed)
+# Configure and mount 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 +51,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, specifying the node port (7861)
 app = gr.mount_gradio_app(app, demo, path="/", ssr_mode=True, node_port=7861)
 
 if __name__ == "__main__":