diff --git a/private_gpt/ui/avatar-bot.ico b/private_gpt/ui/avatar-bot.ico new file mode 100644 index 000000000..ab220adb2 Binary files /dev/null and b/private_gpt/ui/avatar-bot.ico differ diff --git a/private_gpt/ui/ui.py b/private_gpt/ui/ui.py index dea99f50e..eb5c11707 100644 --- a/private_gpt/ui/ui.py +++ b/private_gpt/ui/ui.py @@ -12,6 +12,7 @@ from llama_index.llms import ChatMessage, ChatResponse, MessageRole from pydantic import BaseModel +from private_gpt.constants import PROJECT_ROOT_PATH from private_gpt.di import global_injector from private_gpt.server.chat.chat_service import ChatService, CompletionGen from private_gpt.server.chunks.chunks_service import Chunk, ChunksService @@ -21,7 +22,12 @@ logger = logging.getLogger(__name__) +THIS_DIRECTORY_RELATIVE = Path(__file__).parent.relative_to(PROJECT_ROOT_PATH) +# Should be "private_gpt/ui/avatar-bot.ico" +AVATAR_BOT = THIS_DIRECTORY_RELATIVE / "avatar-bot.ico" + UI_TAB_TITLE = "My Private GPT" + SOURCES_SEPARATOR = "\n\n Sources: \n" @@ -211,9 +217,7 @@ def _build_ui_blocks(self) -> gr.Blocks: render=False, avatar_images=( None, - "https://lh3.googleusercontent.com/drive-viewer/AK7aPa" - "AicXck0k68nsscyfKrb18o9ak3BSaWM_Qzm338cKoQlw72Bp0UKN84" - "IFZjXjZApY01mtnUXDeL4qzwhkALoe_53AhwCg=s2560", + AVATAR_BOT, ), ), additional_inputs=[mode, upload_button],