From e035efeec7bfd534fbf02f2b42b41f8b3eb9c28f Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Sat, 16 Aug 2025 14:30:26 -0700
Subject: [PATCH] fix(public_model_hub.tsx): fix ui

---
 .../src/components/public_model_hub.tsx       | 29 ++++++++++---------
 1 file changed, 16 insertions(+), 13 deletions(-)

diff --git a/ui/litellm-dashboard/src/components/public_model_hub.tsx b/ui/litellm-dashboard/src/components/public_model_hub.tsx
index 9145fec4b5..4c15517376 100644
--- a/ui/litellm-dashboard/src/components/public_model_hub.tsx
+++ b/ui/litellm-dashboard/src/components/public_model_hub.tsx
@@ -18,6 +18,7 @@ import { EndpointType, getEndpointType } from "./chat_ui/mode_endpoint_mapping";
 import { MessageType } from "./chat_ui/types";
 import { getProviderLogoAndName } from "./provider_info_helpers";
 import Navbar from "./navbar";
+import { ThemeProvider } from "@/contexts/ThemeContext";
 // Simple approach without react-markdown dependency
 
 interface ModelGroupInfo {
@@ -476,18 +477,19 @@ const PublicModelHub: React.FC<PublicModelHubProps> = ({ accessToken }) => {
   ];
 
   return (
-    <div className="min-h-screen bg-white">
-      {/* Navigation */}
-      <Navbar 
-        userID={null}
-        userEmail={null}
-        userRole={null}
-        premiumUser={false}
-        setProxySettings={setProxySettings}
-        proxySettings={proxySettings}
-        accessToken={accessToken || null}
-        isPublicPage={true}
-      />
+    <ThemeProvider accessToken={accessToken}>
+      <div className="min-h-screen bg-white">
+        {/* Navigation */}
+        <Navbar 
+          userID={null}
+          userEmail={null}
+          userRole={null}
+          premiumUser={false}
+          setProxySettings={setProxySettings}
+          proxySettings={proxySettings}
+          accessToken={accessToken || null}
+          isPublicPage={true}
+        />
 
       <div className="w-full px-8 py-12">
         {/* About Section */}
@@ -856,7 +858,8 @@ const PublicModelHub: React.FC<PublicModelHubProps> = ({ accessToken }) => {
           </div>
         )}
       </Modal>
-    </div>
+      </div>
+    </ThemeProvider>
   );
 };