Update loader.py

loader.py

@@ -33,8 +33,12 @@ def _with_q(url: str, req: Request) -> str:
 def _looks_file(path: str) -> bool:
     return "." in path.split("/")[-1]
 
+def _is_html_path(path: str) -> bool:
+    # treat root/dirs and *.html as HTML routes
+    return (not _looks_file(path)) or path.lower().endswith(".html") or path == ""
+
 def _mime(path: str, default="application/octet-stream") -> str:
-    if path.endswith("/") or path == "":
+    if _is_html_path(path):
         return "text/html; charset=utf-8"
     m, _ = mimetypes.guess_type(path)
     return m or default
@@ -113,19 +117,22 @@ async def root(request: Request):
     except Exception as e:
         return PlainTextResponse(f"Error fetching root: {e}", status_code=500)
 
-    ctype = r.headers.get("content-type", "text/html; charset=utf-8")
-    if "text/html" in ctype:
-        return HTMLResponse(r.text, status_code=r.status_code, media_type=ctype)
-    return Response(content=r.content, status_code=r.status_code, media_type=ctype)
+    # Force HTML for the root page (even if upstream says text/plain)
+    return HTMLResponse(r.text, status_code=r.status_code, media_type="text/html; charset=utf-8")
 
 @app.get("/{path:path}")
 async def proxy(path: str, request: Request):
     try:
         async with httpx.AsyncClient(timeout=None, follow_redirects=True) as client:
             r, tried = await _fetch(client, path, request)
-        print(f"[proxy] path={path} status={r.status_code}")
+        print(f"[proxy] path={path} status={r.status_code}, upstream-ctype={r.headers.get('content-type')}")
     except Exception as e:
         return PlainTextResponse(f"Error fetching {path}: {e}", status_code=500)
 
+    if _is_html_path(path):
+        # Ensure HTML renders even if upstream mislabels as text/plain
+        return HTMLResponse(r.text, status_code=r.status_code, media_type="text/html; charset=utf-8")
+
+    # Non-HTML: keep upstream/guessed MIME and bytes
     ctype = r.headers.get("content-type") or _mime(path)
     return Response(content=r.content, media_type=ctype, status_code=r.status_code)