oobabooga/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2026-04-05 06:35:15 +00:00
Code Issues Projects Releases Packages Wiki Activity

UI: Add a greeting when the chat history is empty

Browse source
This commit is contained in:
oobabooga 2025-04-24 20:33:40 -07:00
parent 8f2493cc60
commit a90df27ff5
2 changed files with 24 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

16
modules/html_generator.py
View file

@ -1,3 +1,4 @@
import datetime
import functools
import html
import os
@ -389,8 +390,21 @@ def generate_chat_html(history, name1, name2, reset_cache=False):
return output
def time_greeting():
current_hour = datetime.datetime.now().hour
if 5 <= current_hour < 12:
return "Good morning!"
elif 12 <= current_hour < 18:
return "Good afternoon!"
else:
return "Good evening!"
def chat_html_wrapper(history, name1, name2, mode, style, character, reset_cache=False):
if mode == 'instruct':
if len(history['visible']) == 0:
greeting = f"<div class=\"welcome-greeting\">{time_greeting()} How can I help you today?</div>"
result = f'<div class="chat" id="chat">{greeting}</div>'
elif mode == 'instruct':
result = generate_instruct_html(history)
elif style == 'wpp':
result = generate_chat_html(history, name1, name2)