SystemSh0cker/private-gpt
1
0
Fork 0
mirror of https://github.com/zylon-ai/private-gpt.git synced 2025-12-22 23:22:57 +01:00
Code Issues Projects Releases Packages Activity

change

Browse source
This commit is contained in:
Ben-A-Sabu 2024-05-28 12:05:36 +05:30
parent c464225fa1
commit f1f5e5ffa0
1 changed files with 15 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

24
private_gpt/ui/ui.py
View file

@ -300,6 +300,7 @@ class PrivateGptUi:
gr.components.Button(interactive=True),
gr.components.Textbox(self._selected_filename),
]
def _build_ui_blocks(self) -> gr.Blocks:
logger.debug("Creating the UI blocks")
with gr.Blocks(
@ -311,6 +312,12 @@ class PrivateGptUi:
with gr.Row(equal_height=False):
with gr.Column(scale=3):
HHeader = gr.components.Button(
"Doc Chat",
size="m",
)
mode = gr.Radio(
MODES,
label="Mode",
@ -444,18 +451,18 @@ class PrivateGptUi:
with gr.Column(scale=7, elem_id="col"):
# Determine the model label based on the value of PGPT_PROFILES
model_label = get_model_label()
if model_label is not None:
label_text = (
f"LLM: {settings().llm.mode} | Model: {model_label}"
)
else:
label_text = f"LLM: {settings().llm.mode}"
# model_label = get_model_label()
# if model_label is not None:
# label_text = (
# f"LLM: {settings().llm.mode} | Model: {model_label}"
# )
# else:
# label_text = f"LLM: {settings().llm.mode}"
_ = gr.ChatInterface(
self._chat,
chatbot=gr.Chatbot(
label=label_text,
# label=label_text,
show_copy_button=True,
elem_id="chatbot",
render=False,
@ -479,7 +486,6 @@ class PrivateGptUi:
logger.info("Mounting the gradio UI, at path=%s", path)
gr.mount_gradio_app(app, blocks, path=path)
if __name__ == "__main__":
ui = global_injector.get(PrivateGptUi)
_blocks = ui.get_ui_blocks()