fix

app/layout.tsx

@@ -2,6 +2,8 @@
 import type { Metadata, Viewport } from "next";
 import { Inter, PT_Sans } from "next/font/google";
 import Script from "next/script";
+import { headers } from "next/headers";
+import { redirect } from "next/navigation";
 
 import "@/assets/globals.css";
 import { Toaster } from "@/components/ui/sonner";
@@ -74,6 +76,24 @@ export default async function RootLayout({
 }: Readonly<{
   children: React.ReactNode;
 }>) {
+  // Domain redirect check
+  const headersList = await headers();
+  const forwardedHost = headersList.get("x-forwarded-host");
+  const host = headersList.get("host");
+  const hostname = (forwardedHost || host || "").split(":")[0];
+
+  const isLocalDev =
+    hostname === "localhost" ||
+    hostname === "127.0.0.1" ||
+    hostname.startsWith("192.168.");
+  const isHuggingFace =
+    hostname === "huggingface.co" || hostname.endsWith(".huggingface.co");
+
+  if (!isHuggingFace && !isLocalDev) {
+    const pathname = headersList.get("x-invoke-path") || "/deepsite";
+    redirect(`https://huggingface.co${pathname}`);
+  }
+
   // const data = await getMe();
 
   // Generate structured data

middleware.ts

@@ -1,29 +1,10 @@
 import { NextResponse } from "next/server";
 import type { NextRequest } from "next/server";
 
-export function middleware(request: NextRequest) {
-  // Get the actual hostname from headers (important for proxied environments like HF Spaces)
-  const forwardedHost = request.headers.get("x-forwarded-host");
-  const host = request.headers.get("host");
-  const hostname = forwardedHost || host || request.nextUrl.hostname;
-  
-  console.log("[Middleware] x-forwarded-host:", forwardedHost, "host:", host, "hostname:", hostname);
-  
-  const isLocalDev = hostname === "localhost" || hostname === "127.0.0.1" || hostname.startsWith("localhost:") || hostname.startsWith("127.0.0.1:");
-  const isHuggingFace = hostname === "huggingface.co" || hostname.endsWith(".huggingface.co");
-  
-  console.log("[Middleware] isHuggingFace:", isHuggingFace, "isLocalDev:", isLocalDev);
-  
-  if (!isHuggingFace && !isLocalDev) {
-    console.log("[Middleware] Redirecting to huggingface.co");
-    const canonicalUrl = new URL("https://huggingface.co/deepsite");
-    canonicalUrl.pathname = request.nextUrl.pathname;
-    canonicalUrl.search = request.nextUrl.search;
-    return NextResponse.redirect(canonicalUrl, 301);
-  }
-  
+export function middleware(request: NextRequest) {  
   const headers = new Headers(request.headers);
   headers.set("x-current-host", request.nextUrl.host);
+  headers.set("x-invoke-path", request.nextUrl.pathname + request.nextUrl.search);
   
   const response = NextResponse.next({ headers });