From 7be3ea021d64a5243f3764db9ff750a658d373cc Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Wed, 21 Feb 2024 21:33:11 -0800
Subject: [PATCH] (fix) ui - use custom logo

---
 litellm/proxy/proxy_server.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py
index f7b708e1e1..8a36af0d14 100644
--- a/litellm/proxy/proxy_server.py
+++ b/litellm/proxy/proxy_server.py
@@ -5197,7 +5197,11 @@ def get_image():
     """Get logo to show on admin UI"""
     from fastapi.responses import FileResponse
 
-    logo_path = os.getenv("UI_LOGO_PATH", "./logo.jpg")
+    # get current_dir
+    current_dir = os.path.dirname(os.path.abspath(__file__))
+    default_logo = os.path.join(current_dir, "logo.jpg")
+
+    logo_path = os.getenv("UI_LOGO_PATH", default_logo)
     verbose_proxy_logger.debug(f"Reading logo from {logo_path}")
     return FileResponse(path=logo_path)