github/privateGPT
1
0
Fork 1
mirror of https://github.com/imartinez/privateGPT.git synced 2025-06-23 06:00:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(ui): gradio bug fixes (#2021)

Browse Source 
* fix: when two user messages were sent

* fix: add source divider

* fix: add favicon

* fix: add zylon link

* refactor: update label
This commit is contained in:
Javier Martinez 2024-07-29 16:48:16 +02:00 committed by GitHub
parent 20bad17c98
commit d4375d078f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 32 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
private_gpt/ui/ui.py
View File

@ -1,6 +1,5 @@
"""This file should be imported if and only if you want to run the UI locally.""" """This file should be imported if and only if you want to run the UI locally."""
import base64
import itertools
import logging import logging
import time import time
from collections.abc import Iterable from collections.abc import Iterable
@ -31,7 +30,7 @@ AVATAR_BOT = THIS_DIRECTORY_RELATIVE / "avatar-bot.ico"
UI_TAB_TITLE = "My Private GPT" UI_TAB_TITLE = "My Private GPT"
SOURCES_SEPARATOR = "\n\n Sources: \n" SOURCES_SEPARATOR = "<hr>Sources: \n"
MODES = ["Query Files", "Search Files", "LLM Chat (no context from files)"] MODES = ["Query Files", "Search Files", "LLM Chat (no context from files)"]
@ -109,25 +108,25 @@ class PrivateGptUi:
+ f"{index}. {source.file} (page {source.page}) \n\n" + f"{index}. {source.file} (page {source.page}) \n\n"
) )
used_files.add(f"{source.file}-{source.page}") used_files.add(f"{source.file}-{source.page}")
sources_text += "<hr>\n\n"
full_response += sources_text full_response += sources_text
yield full_response yield full_response
def build_history() -> list[ChatMessage]: def build_history() -> list[ChatMessage]:
history_messages: list[ChatMessage] = list( history_messages: list[ChatMessage] = []
itertools.chain(
*[ for interaction in history:
[ history_messages.append(
ChatMessage(content=interaction[0], role=MessageRole.USER), ChatMessage(content=interaction[0], role=MessageRole.USER)
ChatMessage(
# Remove from history content the Sources information
content=interaction[1].split(SOURCES_SEPARATOR)[0],
role=MessageRole.ASSISTANT,
),
]
for interaction in history
]
) )
) if len(interaction) > 1 and interaction[1] is not None:
history_messages.append(
ChatMessage(
# Remove from history content the Sources information
content=interaction[1].split(SOURCES_SEPARATOR)[0],
role=MessageRole.ASSISTANT,
)
)
# max 20 messages to try to avoid context overflow # max 20 messages to try to avoid context overflow
return history_messages[:20] return history_messages[:20]
@ -314,7 +313,13 @@ class PrivateGptUi:
".contain { display: flex !important; flex-direction: column !important; }" ".contain { display: flex !important; flex-direction: column !important; }"
"#component-0, #component-3, #component-10, #component-8 { height: 100% !important; }" "#component-0, #component-3, #component-10, #component-8 { height: 100% !important; }"
"#chatbot { flex-grow: 1 !important; overflow: auto !important;}" "#chatbot { flex-grow: 1 !important; overflow: auto !important;}"
"#col { height: calc(100vh - 112px - 16px) !important; }", "#col { height: calc(100vh - 112px - 16px) !important; }"
"hr { margin-top: 1em; margin-bottom: 1em; border: 0; border-top: 1px solid #FFF; }"
".avatar-image { background-color: antiquewhite; border-radius: 2px; }"
".footer { text-align: center; margin-top: 20px; font-size: 14px; display: flex; align-items: center; justify-content: center; }"
".footer-zylon-link { display:flex; margin-left: 5px; text-decoration: auto; color: #fff; }"
".footer-zylon-link:hover { color: #C7BAFF; }"
".footer-zylon-ico { height: 20px; margin-left: 5px; background-color: antiquewhite; border-radius: 2px; }",
) as blocks: ) as blocks:
with gr.Row(): with gr.Row():
gr.HTML(f"<div class='logo'/><img src={logo_svg} alt=PrivateGPT></div") gr.HTML(f"<div class='logo'/><img src={logo_svg} alt=PrivateGPT></div")
@ -477,6 +482,14 @@ class PrivateGptUi:
), ),
additional_inputs=[mode, upload_button, system_prompt_input], additional_inputs=[mode, upload_button, system_prompt_input],
) )
with gr.Row():
avatar_byte = AVATAR_BOT.read_bytes()
f_base64 = f"data:image/png;base64,{base64.b64encode(avatar_byte).decode('utf-8')}"
gr.HTML(
f"<div class='footer'><a class='footer-zylon-link' href='https://zylon.ai/'>Maintained by Zylon <img class='footer-zylon-ico' src='{f_base64}' alt=Zylon></a></div>"
)
return blocks return blocks
def get_ui_blocks(self) -> gr.Blocks: def get_ui_blocks(self) -> gr.Blocks:
@ -488,7 +501,7 @@ class PrivateGptUi:
blocks = self.get_ui_blocks() blocks = self.get_ui_blocks()
blocks.queue() blocks.queue()
logger.info("Mounting the gradio UI, at path=%s", path) logger.info("Mounting the gradio UI, at path=%s", path)
gr.mount_gradio_app(app, blocks, path=path) gr.mount_gradio_app(app, blocks, path=path, favicon_path=AVATAR_BOT)
if __name__ == "__main__": if __name__ == "__main__":