simcop2387/text-generation-webui-mirror
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2025-06-07 14:17:09 -04:00
Code Issues Projects Releases Packages Wiki Activity

Clean up

Browse source
This commit is contained in:
Th-Underscore 2025-05-24 14:22:42 -04:00
parent b4e008c27d
commit 9d1c1a8892
No known key found for this signature in database
GPG key ID: 8D0551EF8593B2F0
4 changed files with 4 additions and 24 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
js/main.js
View file

@ -796,7 +796,6 @@ createMobileTopBar();
//------------------------------------------------
// --- Message Versioning Variables ---
// let versioningSelectedMessageElement = null; // Deprecated due to persistent selection state
let selectedMessageHistoryIndex = null;
let selectedMessageType = null;
@ -856,7 +855,6 @@ function versioningSelectMessage(element, historyIndex, messageType) {
versioningDeselectMessages();
if (element) {
// versioningSelectedMessageElement = element;
selectedMessageHistoryIndex = historyIndex;
selectedMessageType = messageType;
element.classList.add('selected-message');
@ -865,12 +863,9 @@ function versioningSelectMessage(element, historyIndex, messageType) {
function versioningDeselectMessages() {
const selectedMessageElement = gradioApp().querySelector('#chat .selected-message');
// if (versioningSelectedMessageElement) {
// versioningSelectedMessageElement.classList.remove('selected-message');
if (selectedMessageElement) {
selectedMessageElement.classList.remove('selected-message');
}
// versioningSelectedMessageElement = null;
selectedMessageHistoryIndex = null;
selectedMessageType = null;
}

18
modules/chat.py
View file

@ -667,7 +667,6 @@ def generate_chat_reply_wrapper(text, state, regenerate=False, _continue=False):
send_dummy_reply(state['start_with'], state)
history = state['history']
last_save_time = time.monotonic()
save_interval = 8
for i, history in enumerate(generate_chat_reply(text, state, regenerate, _continue, loading_message=True, for_ui=True)):
@ -711,9 +710,8 @@ def send_last_reply_to_input(history):
return ''
def replace_last_reply(textbox, state):
def replace_last_reply(text, state):
history = state['history']
text = textbox['text']
# Initialize metadata if not present
if 'metadata' not in history:
@ -1321,7 +1319,6 @@ def my_yaml_output(data):
def handle_replace_last_reply_click(text, state):
last_reply = state['history']['internal'][-1][1] if len(state['history']['internal']) > 0 else None
history = replace_last_reply(text, state)
save_history(history, state['unique_id'], state['character_menu'], state['mode'])
html = redraw_html(history, state['name1'], state['name2'], state['mode'], state['chat_style'], state['character_menu'])
@ -1378,17 +1375,8 @@ def handle_start_new_chat_click(state):
def handle_delete_chat_confirm_click(state):
unique_id_to_delete = state['unique_id']
character_to_delete = state['character_menu']
mode_to_delete = state['mode']
all_histories = find_all_histories(state)
index = '0'
if unique_id_to_delete in all_histories:
index = str(all_histories.index(unique_id_to_delete))
delete_history(unique_id_to_delete, character_to_delete, mode_to_delete)
# Load the next appropriate history
index = str(find_all_histories(state).index(state['unique_id']))
delete_history(state['unique_id'], state['character_menu'], state['mode'])
history, unique_id = load_history_after_deletion(state, index)
html = redraw_html(history, state['name1'], state['name2'], state['mode'], state['chat_style'], state['character_menu'])

3
modules/html_generator.py
View file

@ -425,7 +425,6 @@ def generate_instruct_html(history):
selected_class = " selected-message" if message_versioning.is_message_selected(i, 0) else ""
output += (
f'<div class="user-message{selected_class}" '
f'data-history-index="{i}" '
f'data-raw="{html.escape(row_internal[0], quote=True)}">'
f'<div class="text">'
f'<div class="message-body">{converted_visible[0]}</div>'
@ -483,7 +482,6 @@ def generate_cai_chat_html(history, name1, name2, style, character, reset_cache=
selected_class = " selected-message" if message_versioning.is_message_selected(i, 0) else ""
output += (
f'<div class="message{selected_class}" '
f'data-history-index="{i}" '
f'data-raw="{html.escape(row_internal[0], quote=True)}">'
f'<div class="circle-you">{img_me}</div>'
f'<div class="text">'
@ -547,7 +545,6 @@ def generate_chat_html(history, name1, name2, reset_cache=False):
selected_class = " selected-message" if message_versioning.is_message_selected(i, 0) else ""
output += (
f'<div class="message{selected_class}" '
f'data-history-index="{i}"'
f'data-raw="{html.escape(row_internal[0], quote=True)}">'
f'<div class="text-you">'
f'<div class="message-body">{converted_visible[0]}</div>'

2
modules/ui_chat.py
View file

@ -94,7 +94,7 @@ def create_ui():
with gr.Row():
shared.gradio['chat_style'] = gr.Dropdown(choices=utils.get_available_chat_styles(), label='Chat style', value=shared.settings['chat_style'], visible=shared.settings['mode'] != 'instruct')
with gr.Row(visible=True): # TODO: Make visible based on mode
with gr.Row(visible=True):
shared.gradio['message_versioning_display_mode'] = gr.Radio(choices=['html', 'off'], value='html', label="Message Versioning Display", info="Controls how message version navigation is displayed.", elem_id="message-versioning-display-mode", elem_classes=['slim-dropdown'])
with gr.Row():