2023-08-07 02:49:27 +02:00
import json
from functools import partial
from pathlib import Path
import gradio as gr
from PIL import Image
2023-11-17 03:16:42 +01:00
from modules import chat , shared , ui , utils
2023-08-07 02:49:27 +02:00
from modules . html_generator import chat_html_wrapper
from modules . text_generation import stop_everything_event
from modules . utils import gradio
2023-08-14 16:46:07 +02:00
inputs = ( ' Chat input ' , ' interface_state ' )
2023-12-04 02:45:50 +01:00
reload_arr = ( ' history ' , ' name1 ' , ' name2 ' , ' mode ' , ' chat_style ' , ' character_menu ' )
2023-09-21 22:19:32 +02:00
clear_arr = ( ' delete_chat-confirm ' , ' delete_chat ' , ' delete_chat-cancel ' )
2023-08-13 06:12:15 +02:00
2023-08-07 02:49:27 +02:00
def create_ui ( ) :
2023-09-26 14:44:04 +02:00
mu = shared . args . multi_user
2023-08-12 01:33:15 +02:00
shared . gradio [ ' Chat input ' ] = gr . State ( )
shared . gradio [ ' history ' ] = gr . State ( { ' internal ' : [ ] , ' visible ' : [ ] } )
2023-08-07 02:49:27 +02:00
2023-09-18 03:33:00 +02:00
with gr . Tab ( ' Chat ' , elem_id = ' chat-tab ' , elem_classes = ( " old-ui " if shared . args . chat_buttons else None ) ) :
2023-08-07 02:49:27 +02:00
with gr . Row ( ) :
2023-09-18 03:33:00 +02:00
with gr . Column ( elem_id = ' chat-col ' ) :
2023-12-04 02:45:50 +01:00
shared . gradio [ ' display ' ] = gr . HTML ( value = chat_html_wrapper ( { ' internal ' : [ ] , ' visible ' : [ ] } , ' ' , ' ' , ' chat ' , ' cai-chat ' , ' ' ) )
2023-09-13 07:36:12 +02:00
2023-09-18 03:33:00 +02:00
with gr . Row ( elem_id = " chat-input-row " ) :
with gr . Column ( scale = 1 , elem_id = ' gr-hover-container ' ) :
2023-09-14 12:06:42 +02:00
gr . HTML ( value = ' <div class= " hover-element " onclick= " void(0) " ><span style= " width: 100px; display: block " id= " hover-element-button " >☰</span><div class= " hover-menu " id= " hover-menu " ></div> ' , elem_id = ' gr-hover ' )
2023-09-18 03:33:00 +02:00
with gr . Column ( scale = 10 , elem_id = ' chat-input-container ' ) :
2023-09-14 12:06:42 +02:00
shared . gradio [ ' textbox ' ] = gr . Textbox ( label = ' ' , placeholder = ' Send a message ' , elem_id = ' chat-input ' , elem_classes = [ ' add_scrollbar ' ] )
shared . gradio [ ' show_controls ' ] = gr . Checkbox ( value = shared . settings [ ' show_controls ' ] , label = ' Show controls (Ctrl+S) ' , elem_id = ' show-controls ' )
shared . gradio [ ' typing-dots ' ] = gr . HTML ( value = ' <div class= " typing " ><span></span><span class= " dot1 " ></span><span class= " dot2 " ></span></div> ' , label = ' typing ' , elem_id = ' typing-container ' )
2023-09-18 03:33:00 +02:00
with gr . Column ( scale = 1 , elem_id = ' generate-stop-container ' ) :
2023-09-14 12:06:42 +02:00
with gr . Row ( ) :
2023-09-14 14:16:44 +02:00
shared . gradio [ ' Stop ' ] = gr . Button ( ' Stop ' , elem_id = ' stop ' , visible = False )
shared . gradio [ ' Generate ' ] = gr . Button ( ' Generate ' , elem_id = ' Generate ' , variant = ' primary ' )
2023-09-13 07:36:12 +02:00
# Hover menu buttons
2023-09-18 03:33:00 +02:00
with gr . Column ( elem_id = ' chat-buttons ' ) :
2023-09-16 05:39:37 +02:00
with gr . Row ( ) :
shared . gradio [ ' Regenerate ' ] = gr . Button ( ' Regenerate (Ctrl + Enter) ' , elem_id = ' Regenerate ' )
shared . gradio [ ' Continue ' ] = gr . Button ( ' Continue (Alt + Enter) ' , elem_id = ' Continue ' )
shared . gradio [ ' Remove last ' ] = gr . Button ( ' Remove last reply (Ctrl + Shift + Backspace) ' , elem_id = ' Remove-last ' )
2023-09-21 22:19:32 +02:00
2023-09-16 05:39:37 +02:00
with gr . Row ( ) :
shared . gradio [ ' Replace last reply ' ] = gr . Button ( ' Replace last reply (Ctrl + Shift + L) ' , elem_id = ' Replace-last ' )
shared . gradio [ ' Copy last reply ' ] = gr . Button ( ' Copy last reply (Ctrl + Shift + K) ' , elem_id = ' Copy-last ' )
shared . gradio [ ' Impersonate ' ] = gr . Button ( ' Impersonate (Ctrl + Shift + M) ' , elem_id = ' Impersonate ' )
2023-09-21 22:19:32 +02:00
2023-09-16 05:39:37 +02:00
with gr . Row ( ) :
shared . gradio [ ' Send dummy message ' ] = gr . Button ( ' Send dummy message ' )
shared . gradio [ ' Send dummy reply ' ] = gr . Button ( ' Send dummy reply ' )
2023-09-21 22:19:32 +02:00
2023-09-16 05:39:37 +02:00
with gr . Row ( ) :
2023-09-21 22:19:32 +02:00
shared . gradio [ ' Start new chat ' ] = gr . Button ( ' Start new chat ' )
2023-09-16 05:39:37 +02:00
with gr . Row ( ) :
shared . gradio [ ' send-chat-to-default ' ] = gr . Button ( ' Send to default ' )
shared . gradio [ ' send-chat-to-notebook ' ] = gr . Button ( ' Send to notebook ' )
2023-08-21 04:54:59 +02:00
2023-09-22 03:53:03 +02:00
with gr . Row ( elem_id = ' past-chats-row ' ) :
2023-09-26 14:44:04 +02:00
shared . gradio [ ' unique_id ' ] = gr . Dropdown ( label = ' Past chats ' , elem_classes = [ ' slim-dropdown ' ] , interactive = not mu )
shared . gradio [ ' rename_chat ' ] = gr . Button ( ' Rename ' , elem_classes = ' refresh-button ' , interactive = not mu )
shared . gradio [ ' delete_chat ' ] = gr . Button ( ' 🗑️ ' , elem_classes = ' refresh-button ' , interactive = not mu )
2023-09-21 22:19:32 +02:00
shared . gradio [ ' delete_chat-cancel ' ] = gr . Button ( ' Cancel ' , visible = False , elem_classes = ' refresh-button ' )
shared . gradio [ ' delete_chat-confirm ' ] = gr . Button ( ' Confirm ' , variant = ' stop ' , visible = False , elem_classes = ' refresh-button ' )
2023-09-22 03:53:03 +02:00
with gr . Row ( elem_id = ' rename-row ' ) :
shared . gradio [ ' rename_to ' ] = gr . Textbox ( label = ' Rename to: ' , placeholder = ' New name ' , visible = False , elem_classes = [ ' no-background ' ] )
shared . gradio [ ' rename_to-cancel ' ] = gr . Button ( ' Cancel ' , visible = False , elem_classes = ' refresh-button ' )
shared . gradio [ ' rename_to-confirm ' ] = gr . Button ( ' Confirm ' , visible = False , elem_classes = ' refresh-button ' )
2023-08-07 02:49:27 +02:00
with gr . Row ( ) :
shared . gradio [ ' start_with ' ] = gr . Textbox ( label = ' Start reply with ' , placeholder = ' Sure thing! ' , value = shared . settings [ ' start_with ' ] )
with gr . Row ( ) :
2023-09-21 22:19:32 +02:00
shared . gradio [ ' mode ' ] = gr . Radio ( choices = [ ' chat ' , ' chat-instruct ' , ' instruct ' ] , value = ' chat ' , label = ' Mode ' , info = ' Defines how the chat prompt is generated. In instruct and chat-instruct modes, the instruction template selected under Parameters > Instruction template must match the current model. ' , elem_id = ' chat-mode ' )
2023-08-07 02:49:27 +02:00
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 ' )
2023-08-13 06:12:15 +02:00
def create_chat_settings_ui ( ) :
2023-09-26 14:44:04 +02:00
mu = shared . args . multi_user
2023-08-13 06:12:15 +02:00
with gr . Tab ( ' Character ' ) :
with gr . Row ( ) :
with gr . Column ( scale = 8 ) :
2023-08-07 02:49:27 +02:00
with gr . Row ( ) :
2023-10-11 03:20:49 +02:00
shared . gradio [ ' character_menu ' ] = gr . Dropdown ( value = None , choices = utils . get_available_characters ( ) , label = ' Character ' , elem_id = ' character-menu ' , info = ' Used in chat and chat-instruct modes. ' , elem_classes = ' slim-dropdown ' )
2023-09-26 14:44:04 +02:00
ui . create_refresh_button ( shared . gradio [ ' character_menu ' ] , lambda : None , lambda : { ' choices ' : utils . get_available_characters ( ) } , ' refresh-button ' , interactive = not mu )
shared . gradio [ ' save_character ' ] = gr . Button ( ' 💾 ' , elem_classes = ' refresh-button ' , interactive = not mu )
shared . gradio [ ' delete_character ' ] = gr . Button ( ' 🗑️ ' , elem_classes = ' refresh-button ' , interactive = not mu )
2023-08-07 02:49:27 +02:00
2023-08-13 06:12:15 +02:00
shared . gradio [ ' name1 ' ] = gr . Textbox ( value = shared . settings [ ' name1 ' ] , lines = 1 , label = ' Your name ' )
2023-09-21 22:19:32 +02:00
shared . gradio [ ' name2 ' ] = gr . Textbox ( value = ' ' , lines = 1 , label = ' Character \' s name ' )
shared . gradio [ ' context ' ] = gr . Textbox ( value = ' ' , lines = 10 , label = ' Context ' , elem_classes = [ ' add_scrollbar ' ] )
shared . gradio [ ' greeting ' ] = gr . Textbox ( value = ' ' , lines = 5 , label = ' Greeting ' , elem_classes = [ ' add_scrollbar ' ] )
2023-08-13 06:12:15 +02:00
with gr . Column ( scale = 1 ) :
2023-09-26 14:44:04 +02:00
shared . gradio [ ' character_picture ' ] = gr . Image ( label = ' Character picture ' , type = ' pil ' , interactive = not mu )
shared . gradio [ ' your_picture ' ] = gr . Image ( label = ' Your picture ' , type = ' pil ' , value = Image . open ( Path ( ' cache/pfp_me.png ' ) ) if Path ( ' cache/pfp_me.png ' ) . exists ( ) else None , interactive = not mu )
2023-08-13 06:12:15 +02:00
with gr . Tab ( ' Instruction template ' ) :
with gr . Row ( ) :
2023-12-12 21:23:14 +01:00
with gr . Column ( ) :
with gr . Row ( ) :
shared . gradio [ ' instruction_template ' ] = gr . Dropdown ( choices = utils . get_available_instruction_templates ( ) , label = ' Instruction template ' , value = ' Custom ' , info = ' Change this according to the model/LoRA that you are using. Used in instruct and chat-instruct modes. ' , elem_classes = ' slim-dropdown ' )
ui . create_refresh_button ( shared . gradio [ ' instruction_template ' ] , lambda : None , lambda : { ' choices ' : utils . get_available_instruction_templates ( ) } , ' refresh-button ' , interactive = not mu )
shared . gradio [ ' load_template ' ] = gr . Button ( " Load " , elem_classes = ' refresh-button ' )
shared . gradio [ ' save_template ' ] = gr . Button ( ' 💾 ' , elem_classes = ' refresh-button ' , interactive = not mu )
shared . gradio [ ' delete_template ' ] = gr . Button ( ' 🗑️ ' , elem_classes = ' refresh-button ' , interactive = not mu )
with gr . Column ( ) :
pass
2023-08-14 03:14:09 +02:00
2023-08-13 06:12:15 +02:00
with gr . Row ( ) :
2023-12-12 21:23:14 +01:00
with gr . Column ( ) :
shared . gradio [ ' custom_system_message ' ] = gr . Textbox ( value = shared . settings [ ' custom_system_message ' ] , lines = 2 , label = ' Custom system message ' , info = ' If not empty, will be used instead of the default one. ' , elem_classes = [ ' add_scrollbar ' ] )
shared . gradio [ ' instruction_template_str ' ] = gr . Textbox ( value = ' ' , label = ' Instruction template ' , lines = 24 , elem_classes = [ ' add_scrollbar ' , ' monospace ' ] )
with gr . Row ( ) :
shared . gradio [ ' send_instruction_to_default ' ] = gr . Button ( ' Send to default ' , elem_classes = [ ' small-button ' ] )
shared . gradio [ ' send_instruction_to_notebook ' ] = gr . Button ( ' Send to notebook ' , elem_classes = [ ' small-button ' ] )
shared . gradio [ ' send_instruction_to_negative_prompt ' ] = gr . Button ( ' Send to negative prompt ' , elem_classes = [ ' small-button ' ] )
with gr . Column ( ) :
shared . gradio [ ' chat_template_str ' ] = gr . Textbox ( value = shared . settings [ ' chat_template_str ' ] , label = ' Chat template ' , lines = 22 , elem_classes = [ ' add_scrollbar ' , ' monospace ' ] )
shared . gradio [ ' chat-instruct_command ' ] = gr . Textbox ( value = shared . settings [ ' chat-instruct_command ' ] , lines = 4 , label = ' Command for chat-instruct mode ' , info = ' <|character|> gets replaced by the bot name, and <|prompt|> gets replaced by the regular chat prompt. ' , elem_classes = [ ' add_scrollbar ' ] )
2023-08-07 02:49:27 +02:00
2023-08-13 06:12:15 +02:00
with gr . Tab ( ' Chat history ' ) :
with gr . Row ( ) :
with gr . Column ( ) :
shared . gradio [ ' save_chat_history ' ] = gr . Button ( value = ' Save history ' )
2023-08-07 02:49:27 +02:00
2023-08-13 06:12:15 +02:00
with gr . Column ( ) :
shared . gradio [ ' load_chat_history ' ] = gr . File ( type = ' binary ' , file_types = [ ' .json ' , ' .txt ' ] , label = ' Upload History JSON ' )
2023-08-07 02:49:27 +02:00
2023-08-13 06:12:15 +02:00
with gr . Tab ( ' Upload character ' ) :
with gr . Tab ( ' YAML or JSON ' ) :
with gr . Row ( ) :
2023-09-26 14:44:04 +02:00
shared . gradio [ ' upload_json ' ] = gr . File ( type = ' binary ' , file_types = [ ' .json ' , ' .yaml ' ] , label = ' JSON or YAML File ' , interactive = not mu )
shared . gradio [ ' upload_img_bot ' ] = gr . Image ( type = ' pil ' , label = ' Profile Picture (optional) ' , interactive = not mu )
2023-08-07 02:49:27 +02:00
2023-08-13 06:12:15 +02:00
shared . gradio [ ' Submit character ' ] = gr . Button ( value = ' Submit ' , interactive = False )
2023-08-07 02:49:27 +02:00
2023-08-13 06:12:15 +02:00
with gr . Tab ( ' TavernAI PNG ' ) :
with gr . Row ( ) :
with gr . Column ( ) :
2023-09-26 14:44:04 +02:00
shared . gradio [ ' upload_img_tavern ' ] = gr . Image ( type = ' pil ' , label = ' TavernAI PNG File ' , elem_id = ' upload_img_tavern ' , interactive = not mu )
2023-08-13 06:12:15 +02:00
shared . gradio [ ' tavern_json ' ] = gr . State ( )
with gr . Column ( ) :
shared . gradio [ ' tavern_name ' ] = gr . Textbox ( value = ' ' , lines = 1 , label = ' Name ' , interactive = False )
shared . gradio [ ' tavern_desc ' ] = gr . Textbox ( value = ' ' , lines = 4 , max_lines = 4 , label = ' Description ' , interactive = False )
shared . gradio [ ' Submit tavern character ' ] = gr . Button ( value = ' Submit ' , interactive = False )
2023-08-07 02:49:27 +02:00
def create_event_handlers ( ) :
2023-08-14 18:39:06 +02:00
# Obsolete variables, kept for compatibility with old extensions
shared . input_params = gradio ( inputs )
shared . reload_inputs = gradio ( reload_arr )
2023-08-07 02:49:27 +02:00
2023-08-14 19:10:39 +02:00
shared . gradio [ ' Generate ' ] . click (
2023-08-07 02:49:27 +02:00
ui . gather_interface_values , gradio ( shared . input_elements ) , gradio ( ' interface_state ' ) ) . then (
lambda x : ( x , ' ' ) , gradio ( ' textbox ' ) , gradio ( ' Chat input ' , ' textbox ' ) , show_progress = False ) . then (
2023-08-13 06:12:15 +02:00
chat . generate_chat_reply_wrapper , gradio ( inputs ) , gradio ( ' display ' , ' history ' ) , show_progress = False ) . then (
2023-08-07 02:49:27 +02:00
ui . gather_interface_values , gradio ( shared . input_elements ) , gradio ( ' interface_state ' ) ) . then (
2023-09-21 22:19:32 +02:00
chat . save_history , gradio ( ' history ' , ' unique_id ' , ' character_menu ' , ' mode ' ) , None ) . then (
2023-08-12 01:33:15 +02:00
lambda : None , None , None , _js = f ' () => {{ { ui . audio_notification_js } }} ' )
2023-08-07 02:49:27 +02:00
2023-08-14 19:10:39 +02:00
shared . gradio [ ' textbox ' ] . submit (
2023-08-07 02:49:27 +02:00
ui . gather_interface_values , gradio ( shared . input_elements ) , gradio ( ' interface_state ' ) ) . then (
lambda x : ( x , ' ' ) , gradio ( ' textbox ' ) , gradio ( ' Chat input ' , ' textbox ' ) , show_progress = False ) . then (
2023-08-13 06:12:15 +02:00
chat . generate_chat_reply_wrapper , gradio ( inputs ) , gradio ( ' display ' , ' history ' ) , show_progress = False ) . then (
2023-08-07 02:49:27 +02:00
ui . gather_interface_values , gradio ( shared . input_elements ) , gradio ( ' interface_state ' ) ) . then (
2023-09-21 22:19:32 +02:00
chat . save_history , gradio ( ' history ' , ' unique_id ' , ' character_menu ' , ' mode ' ) , None ) . then (
2023-08-12 01:33:15 +02:00
lambda : None , None , None , _js = f ' () => {{ { ui . audio_notification_js } }} ' )
2023-08-07 02:49:27 +02:00
2023-08-14 19:10:39 +02:00
shared . gradio [ ' Regenerate ' ] . click (
2023-08-07 02:49:27 +02:00
ui . gather_interface_values , gradio ( shared . input_elements ) , gradio ( ' interface_state ' ) ) . then (
2023-08-13 06:12:15 +02:00
partial ( chat . generate_chat_reply_wrapper , regenerate = True ) , gradio ( inputs ) , gradio ( ' display ' , ' history ' ) , show_progress = False ) . then (
2023-08-07 02:49:27 +02:00
ui . gather_interface_values , gradio ( shared . input_elements ) , gradio ( ' interface_state ' ) ) . then (
2023-09-21 22:19:32 +02:00
chat . save_history , gradio ( ' history ' , ' unique_id ' , ' character_menu ' , ' mode ' ) , None ) . then (
2023-08-12 01:33:15 +02:00
lambda : None , None , None , _js = f ' () => {{ { ui . audio_notification_js } }} ' )
2023-08-07 02:49:27 +02:00
2023-08-14 19:10:39 +02:00
shared . gradio [ ' Continue ' ] . click (
2023-08-07 02:49:27 +02:00
ui . gather_interface_values , gradio ( shared . input_elements ) , gradio ( ' interface_state ' ) ) . then (
2023-08-13 06:12:15 +02:00
partial ( chat . generate_chat_reply_wrapper , _continue = True ) , gradio ( inputs ) , gradio ( ' display ' , ' history ' ) , show_progress = False ) . then (
2023-08-07 02:49:27 +02:00
ui . gather_interface_values , gradio ( shared . input_elements ) , gradio ( ' interface_state ' ) ) . then (
2023-09-21 22:19:32 +02:00
chat . save_history , gradio ( ' history ' , ' unique_id ' , ' character_menu ' , ' mode ' ) , None ) . then (
2023-08-12 01:33:15 +02:00
lambda : None , None , None , _js = f ' () => {{ { ui . audio_notification_js } }} ' )
2023-08-07 02:49:27 +02:00
2023-08-14 19:10:39 +02:00
shared . gradio [ ' Impersonate ' ] . click (
2023-08-07 02:49:27 +02:00
ui . gather_interface_values , gradio ( shared . input_elements ) , gradio ( ' interface_state ' ) ) . then (
lambda x : x , gradio ( ' textbox ' ) , gradio ( ' Chat input ' ) , show_progress = False ) . then (
2023-09-13 20:22:53 +02:00
chat . impersonate_wrapper , gradio ( inputs ) , gradio ( ' textbox ' , ' display ' ) , show_progress = False ) . then (
2023-08-07 02:49:27 +02:00
ui . gather_interface_values , gradio ( shared . input_elements ) , gradio ( ' interface_state ' ) ) . then (
2023-08-12 01:33:15 +02:00
lambda : None , None , None , _js = f ' () => {{ { ui . audio_notification_js } }} ' )
2023-08-07 02:49:27 +02:00
shared . gradio [ ' Replace last reply ' ] . click (
ui . gather_interface_values , gradio ( shared . input_elements ) , gradio ( ' interface_state ' ) ) . then (
chat . replace_last_reply , gradio ( ' textbox ' , ' interface_state ' ) , gradio ( ' history ' ) ) . then (
lambda : ' ' , None , gradio ( ' textbox ' ) , show_progress = False ) . then (
2023-08-13 06:12:15 +02:00
chat . redraw_html , gradio ( reload_arr ) , gradio ( ' display ' ) ) . then (
2023-09-21 22:19:32 +02:00
chat . save_history , gradio ( ' history ' , ' unique_id ' , ' character_menu ' , ' mode ' ) , None )
2023-08-07 02:49:27 +02:00
shared . gradio [ ' Send dummy message ' ] . click (
ui . gather_interface_values , gradio ( shared . input_elements ) , gradio ( ' interface_state ' ) ) . then (
chat . send_dummy_message , gradio ( ' textbox ' , ' interface_state ' ) , gradio ( ' history ' ) ) . then (
lambda : ' ' , None , gradio ( ' textbox ' ) , show_progress = False ) . then (
2023-08-13 06:12:15 +02:00
chat . redraw_html , gradio ( reload_arr ) , gradio ( ' display ' ) ) . then (
2023-09-21 22:19:32 +02:00
chat . save_history , gradio ( ' history ' , ' unique_id ' , ' character_menu ' , ' mode ' ) , None )
2023-08-07 02:49:27 +02:00
shared . gradio [ ' Send dummy reply ' ] . click (
ui . gather_interface_values , gradio ( shared . input_elements ) , gradio ( ' interface_state ' ) ) . then (
chat . send_dummy_reply , gradio ( ' textbox ' , ' interface_state ' ) , gradio ( ' history ' ) ) . then (
lambda : ' ' , None , gradio ( ' textbox ' ) , show_progress = False ) . then (
2023-08-13 06:12:15 +02:00
chat . redraw_html , gradio ( reload_arr ) , gradio ( ' display ' ) ) . then (
2023-09-21 22:19:32 +02:00
chat . save_history , gradio ( ' history ' , ' unique_id ' , ' character_menu ' , ' mode ' ) , None )
2023-08-07 02:49:27 +02:00
shared . gradio [ ' Remove last ' ] . click (
ui . gather_interface_values , gradio ( shared . input_elements ) , gradio ( ' interface_state ' ) ) . then (
chat . remove_last_message , gradio ( ' history ' ) , gradio ( ' textbox ' , ' history ' ) , show_progress = False ) . then (
2023-08-13 06:12:15 +02:00
chat . redraw_html , gradio ( reload_arr ) , gradio ( ' display ' ) ) . then (
2023-09-21 22:19:32 +02:00
chat . save_history , gradio ( ' history ' , ' unique_id ' , ' character_menu ' , ' mode ' ) , None )
2023-08-07 02:49:27 +02:00
shared . gradio [ ' Stop ' ] . click (
2023-08-14 19:10:39 +02:00
stop_everything_event , None , None , queue = False ) . then (
2023-08-13 06:12:15 +02:00
chat . redraw_html , gradio ( reload_arr ) , gradio ( ' display ' ) )
2023-08-07 02:49:27 +02:00
2023-09-21 22:19:32 +02:00
if not shared . args . multi_user :
shared . gradio [ ' unique_id ' ] . select (
chat . load_history , gradio ( ' unique_id ' , ' character_menu ' , ' mode ' ) , gradio ( ' history ' ) ) . then (
chat . redraw_html , gradio ( reload_arr ) , gradio ( ' display ' ) )
shared . gradio [ ' Start new chat ' ] . click (
ui . gather_interface_values , gradio ( shared . input_elements ) , gradio ( ' interface_state ' ) ) . then (
chat . start_new_chat , gradio ( ' interface_state ' ) , gradio ( ' history ' ) ) . then (
chat . redraw_html , gradio ( reload_arr ) , gradio ( ' display ' ) ) . then (
lambda x : gr . update ( choices = ( histories := chat . find_all_histories ( x ) ) , value = histories [ 0 ] ) , gradio ( ' interface_state ' ) , gradio ( ' unique_id ' ) )
shared . gradio [ ' delete_chat ' ] . click ( lambda : [ gr . update ( visible = True ) , gr . update ( visible = False ) , gr . update ( visible = True ) ] , None , gradio ( clear_arr ) )
shared . gradio [ ' delete_chat-cancel ' ] . click ( lambda : [ gr . update ( visible = False ) , gr . update ( visible = True ) , gr . update ( visible = False ) ] , None , gradio ( clear_arr ) )
shared . gradio [ ' delete_chat-confirm ' ] . click (
ui . gather_interface_values , gradio ( shared . input_elements ) , gradio ( ' interface_state ' ) ) . then (
chat . delete_history , gradio ( ' unique_id ' , ' character_menu ' , ' mode ' ) , None ) . then (
chat . load_latest_history , gradio ( ' interface_state ' ) , gradio ( ' history ' ) ) . then (
chat . redraw_html , gradio ( reload_arr ) , gradio ( ' display ' ) ) . then (
lambda x : gr . update ( choices = ( histories := chat . find_all_histories ( x ) ) , value = histories [ 0 ] ) , gradio ( ' interface_state ' ) , gradio ( ' unique_id ' ) ) . then (
lambda : [ gr . update ( visible = False ) , gr . update ( visible = True ) , gr . update ( visible = False ) ] , None , gradio ( clear_arr ) )
2023-09-22 03:53:03 +02:00
shared . gradio [ ' rename_chat ' ] . click (
lambda x : x , gradio ( ' unique_id ' ) , gradio ( ' rename_to ' ) ) . then (
lambda : [ gr . update ( visible = True ) ] * 3 , None , gradio ( ' rename_to ' , ' rename_to-confirm ' , ' rename_to-cancel ' ) , show_progress = False )
shared . gradio [ ' rename_to-cancel ' ] . click (
lambda : [ gr . update ( visible = False ) ] * 3 , None , gradio ( ' rename_to ' , ' rename_to-confirm ' , ' rename_to-cancel ' ) , show_progress = False )
shared . gradio [ ' rename_to-confirm ' ] . click (
chat . rename_history , gradio ( ' unique_id ' , ' rename_to ' , ' character_menu ' , ' mode ' ) , None ) . then (
lambda : [ gr . update ( visible = False ) ] * 3 , None , gradio ( ' rename_to ' , ' rename_to-confirm ' , ' rename_to-cancel ' ) , show_progress = False ) . then (
lambda x , y : gr . update ( choices = chat . find_all_histories ( x ) , value = y ) , gradio ( ' interface_state ' , ' rename_to ' ) , gradio ( ' unique_id ' ) )
2023-09-21 22:19:32 +02:00
shared . gradio [ ' load_chat_history ' ] . upload (
ui . gather_interface_values , gradio ( shared . input_elements ) , gradio ( ' interface_state ' ) ) . then (
chat . start_new_chat , gradio ( ' interface_state ' ) , gradio ( ' history ' ) ) . then (
chat . load_history_json , gradio ( ' load_chat_history ' , ' history ' ) , gradio ( ' history ' ) ) . then (
chat . redraw_html , gradio ( reload_arr ) , gradio ( ' display ' ) ) . then (
lambda x : gr . update ( choices = ( histories := chat . find_all_histories ( x ) ) , value = histories [ 0 ] ) , gradio ( ' interface_state ' ) , gradio ( ' unique_id ' ) ) . then (
chat . save_history , gradio ( ' history ' , ' unique_id ' , ' character_menu ' , ' mode ' ) , None ) . then (
lambda : None , None , None , _js = f ' () => {{ { ui . switch_tabs_js } ; switch_to_chat() }} ' )
shared . gradio [ ' character_menu ' ] . change (
2023-12-12 21:23:14 +01:00
chat . load_character , gradio ( ' character_menu ' , ' name1 ' , ' name2 ' ) , gradio ( ' name1 ' , ' name2 ' , ' character_picture ' , ' greeting ' , ' context ' ) ) . success (
2023-09-21 22:19:32 +02:00
ui . gather_interface_values , gradio ( shared . input_elements ) , gradio ( ' interface_state ' ) ) . then (
chat . load_latest_history , gradio ( ' interface_state ' ) , gradio ( ' history ' ) ) . then (
chat . redraw_html , gradio ( reload_arr ) , gradio ( ' display ' ) ) . then (
2023-11-19 06:05:17 +01:00
lambda x : gr . update ( choices = ( histories := chat . find_all_histories ( x ) ) , value = histories [ 0 ] ) , gradio ( ' interface_state ' ) , gradio ( ' unique_id ' ) ) . then (
lambda : None , None , None , _js = f ' () => {{ { ui . update_big_picture_js } ; updateBigPicture() }} ' )
2023-09-21 22:19:32 +02:00
2023-08-07 02:49:27 +02:00
shared . gradio [ ' mode ' ] . change (
lambda x : gr . update ( visible = x != ' instruct ' ) , gradio ( ' mode ' ) , gradio ( ' chat_style ' ) , show_progress = False ) . then (
2023-09-21 22:19:32 +02:00
ui . gather_interface_values , gradio ( shared . input_elements ) , gradio ( ' interface_state ' ) ) . then (
partial ( chat . character_is_loaded , raise_exception = True ) , gradio ( ' interface_state ' ) , None ) . success (
chat . load_latest_history , gradio ( ' interface_state ' ) , gradio ( ' history ' ) ) . then (
chat . redraw_html , gradio ( reload_arr ) , gradio ( ' display ' ) ) . then (
lambda x : gr . update ( choices = ( histories := chat . find_all_histories ( x ) ) , value = histories [ 0 ] ) , gradio ( ' interface_state ' ) , gradio ( ' unique_id ' ) )
2023-08-07 02:49:27 +02:00
2023-08-13 06:12:15 +02:00
shared . gradio [ ' chat_style ' ] . change ( chat . redraw_html , gradio ( reload_arr ) , gradio ( ' display ' ) )
2023-08-07 02:49:27 +02:00
shared . gradio [ ' Copy last reply ' ] . click ( chat . send_last_reply_to_input , gradio ( ' history ' ) , gradio ( ' textbox ' ) , show_progress = False )
# Save/delete a character
shared . gradio [ ' save_character ' ] . click (
lambda x : x , gradio ( ' name2 ' ) , gradio ( ' save_character_filename ' ) ) . then (
lambda : gr . update ( visible = True ) , None , gradio ( ' character_saver ' ) )
shared . gradio [ ' delete_character ' ] . click ( lambda : gr . update ( visible = True ) , None , gradio ( ' character_deleter ' ) )
2023-12-12 21:23:14 +01:00
shared . gradio [ ' load_template ' ] . click ( chat . load_instruction_template , gradio ( ' instruction_template ' ) , gradio ( ' instruction_template_str ' ) )
2023-08-07 02:49:27 +02:00
shared . gradio [ ' save_template ' ] . click (
lambda : ' My Template.yaml ' , None , gradio ( ' save_filename ' ) ) . then (
2023-08-07 02:50:07 +02:00
lambda : ' instruction-templates/ ' , None , gradio ( ' save_root ' ) ) . then (
2023-12-12 21:23:14 +01:00
chat . generate_instruction_template_yaml , gradio ( ' instruction_template_str ' ) , gradio ( ' save_contents ' ) ) . then (
2023-08-07 02:49:27 +02:00
lambda : gr . update ( visible = True ) , None , gradio ( ' file_saver ' ) )
shared . gradio [ ' delete_template ' ] . click (
lambda x : f ' { x } .yaml ' , gradio ( ' instruction_template ' ) , gradio ( ' delete_filename ' ) ) . then (
2023-08-07 02:50:07 +02:00
lambda : ' instruction-templates/ ' , None , gradio ( ' delete_root ' ) ) . then (
2023-08-07 02:49:27 +02:00
lambda : gr . update ( visible = True ) , None , gradio ( ' file_deleter ' ) )
shared . gradio [ ' save_chat_history ' ] . click (
lambda x : json . dumps ( x , indent = 4 ) , gradio ( ' history ' ) , gradio ( ' temporary_text ' ) ) . then (
2023-08-12 01:33:15 +02:00
None , gradio ( ' temporary_text ' , ' character_menu ' , ' mode ' ) , None , _js = f ' (hist, char, mode) => {{ { ui . save_files_js } ; saveHistory(hist, char, mode) }} ' )
2023-08-07 02:49:27 +02:00
shared . gradio [ ' Submit character ' ] . click (
chat . upload_character , gradio ( ' upload_json ' , ' upload_img_bot ' ) , gradio ( ' character_menu ' ) ) . then (
2023-08-14 03:14:09 +02:00
lambda : None , None , None , _js = f ' () => {{ { ui . switch_tabs_js } ; switch_to_character() }} ' )
2023-08-07 02:49:27 +02:00
shared . gradio [ ' Submit tavern character ' ] . click (
chat . upload_tavern_character , gradio ( ' upload_img_tavern ' , ' tavern_json ' ) , gradio ( ' character_menu ' ) ) . then (
2023-08-14 03:14:09 +02:00
lambda : None , None , None , _js = f ' () => {{ { ui . switch_tabs_js } ; switch_to_character() }} ' )
2023-08-07 02:49:27 +02:00
shared . gradio [ ' upload_json ' ] . upload ( lambda : gr . update ( interactive = True ) , None , gradio ( ' Submit character ' ) )
shared . gradio [ ' upload_json ' ] . clear ( lambda : gr . update ( interactive = False ) , None , gradio ( ' Submit character ' ) )
shared . gradio [ ' upload_img_tavern ' ] . upload ( chat . check_tavern_character , gradio ( ' upload_img_tavern ' ) , gradio ( ' tavern_name ' , ' tavern_desc ' , ' tavern_json ' , ' Submit tavern character ' ) , show_progress = False )
shared . gradio [ ' upload_img_tavern ' ] . clear ( lambda : ( None , None , None , gr . update ( interactive = False ) ) , None , gradio ( ' tavern_name ' , ' tavern_desc ' , ' tavern_json ' , ' Submit tavern character ' ) , show_progress = False )
shared . gradio [ ' your_picture ' ] . change (
chat . upload_your_profile_picture , gradio ( ' your_picture ' ) , None ) . then (
2023-08-13 06:12:15 +02:00
partial ( chat . redraw_html , reset_cache = True ) , gradio ( reload_arr ) , gradio ( ' display ' ) )
2023-08-14 03:14:09 +02:00
shared . gradio [ ' send_instruction_to_default ' ] . click (
2023-11-17 03:16:42 +01:00
ui . gather_interface_values , gradio ( shared . input_elements ) , gradio ( ' interface_state ' ) ) . then (
lambda x : x . update ( { ' mode ' : ' instruct ' , ' history ' : { ' internal ' : [ ] , ' visible ' : [ ] } } ) , gradio ( ' interface_state ' ) , None ) . then (
partial ( chat . generate_chat_prompt , ' Input ' ) , gradio ( ' interface_state ' ) , gradio ( ' textbox-default ' ) ) . then (
2023-08-14 03:14:09 +02:00
lambda : None , None , None , _js = f ' () => {{ { ui . switch_tabs_js } ; switch_to_default() }} ' )
shared . gradio [ ' send_instruction_to_notebook ' ] . click (
2023-11-17 03:16:42 +01:00
ui . gather_interface_values , gradio ( shared . input_elements ) , gradio ( ' interface_state ' ) ) . then (
lambda x : x . update ( { ' mode ' : ' instruct ' , ' history ' : { ' internal ' : [ ] , ' visible ' : [ ] } } ) , gradio ( ' interface_state ' ) , None ) . then (
partial ( chat . generate_chat_prompt , ' Input ' ) , gradio ( ' interface_state ' ) , gradio ( ' textbox-notebook ' ) ) . then (
2023-08-14 03:14:09 +02:00
lambda : None , None , None , _js = f ' () => {{ { ui . switch_tabs_js } ; switch_to_notebook() }} ' )
shared . gradio [ ' send_instruction_to_negative_prompt ' ] . click (
2023-11-17 03:16:42 +01:00
ui . gather_interface_values , gradio ( shared . input_elements ) , gradio ( ' interface_state ' ) ) . then (
lambda x : x . update ( { ' mode ' : ' instruct ' , ' history ' : { ' internal ' : [ ] , ' visible ' : [ ] } } ) , gradio ( ' interface_state ' ) , None ) . then (
partial ( chat . generate_chat_prompt , ' Input ' ) , gradio ( ' interface_state ' ) , gradio ( ' negative_prompt ' ) ) . then (
2023-08-14 03:14:09 +02:00
lambda : None , None , None , _js = f ' () => {{ { ui . switch_tabs_js } ; switch_to_generation_parameters() }} ' )
2023-08-16 16:03:53 +02:00
2023-08-21 04:54:59 +02:00
shared . gradio [ ' send-chat-to-default ' ] . click (
ui . gather_interface_values , gradio ( shared . input_elements ) , gradio ( ' interface_state ' ) ) . then (
partial ( chat . generate_chat_prompt , ' ' , _continue = True ) , gradio ( ' interface_state ' ) , gradio ( ' textbox-default ' ) ) . then (
lambda : None , None , None , _js = f ' () => {{ { ui . switch_tabs_js } ; switch_to_default() }} ' )
shared . gradio [ ' send-chat-to-notebook ' ] . click (
ui . gather_interface_values , gradio ( shared . input_elements ) , gradio ( ' interface_state ' ) ) . then (
partial ( chat . generate_chat_prompt , ' ' , _continue = True ) , gradio ( ' interface_state ' ) , gradio ( ' textbox-notebook ' ) ) . then (
lambda : None , None , None , _js = f ' () => {{ { ui . switch_tabs_js } ; switch_to_notebook() }} ' )
2023-08-16 16:03:53 +02:00
shared . gradio [ ' show_controls ' ] . change ( None , gradio ( ' show_controls ' ) , None , _js = f ' (x) => {{ { ui . show_controls_js } ; toggle_controls(x) }} ' )
