From e6b43f19f6e14cdfe634d6ee387b14402909b472 Mon Sep 17 00:00:00 2001 From: LouieStark Date: Fri, 1 Nov 2024 17:10:56 +0800 Subject: [PATCH 1/2] update chatbot --- scepter/studio/chatbot/chatbot.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/scepter/studio/chatbot/chatbot.py b/scepter/studio/chatbot/chatbot.py index 57629d7..31d9051 100644 --- a/scepter/studio/chatbot/chatbot.py +++ b/scepter/studio/chatbot/chatbot.py @@ -852,7 +852,7 @@ class ChatBotUI(object): history = [(prompt, f'{pre_info} The generated image is:\n {img_str}')] return self.get_history(history), gr.update(value=''), gr.update( - visible=False) + visible=False), gr.update(value=-1) with self.eg: self.example_task = gr.Text(label='Task Name', @@ -878,7 +878,7 @@ class ChatBotUI(object): self.example_task, self.example_image, self.example_mask, self.example_ref_im1, self.text, self.seed ], - outputs=[self.chatbot, self.text, self.gallery], + outputs=[self.chatbot, self.text, self.gallery, self.seed], examples_per_page=4, run_on_click=True) From fe3e11b49ed73378b01a0122022c743668e093f2 Mon Sep 17 00:00:00 2001 From: LouieStark Date: Fri, 1 Nov 2024 21:13:21 +0800 Subject: [PATCH 2/2] update chatbot --- scepter/studio/chatbot/chatbot.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/scepter/studio/chatbot/chatbot.py b/scepter/studio/chatbot/chatbot.py index 31d9051..cd7e31b 100644 --- a/scepter/studio/chatbot/chatbot.py +++ b/scepter/studio/chatbot/chatbot.py @@ -74,7 +74,6 @@ class ChatBotUI(object): cfg_file=self.model_choices[self.model_name]) self.pipe = ACEInference() self.pipe.init_from_cfg(model_cfg) - self.retry_msg = '' self.max_msgs = 20 self.enable_i2v = cfg.get('ENABLE_I2V', False) @@ -176,6 +175,7 @@ class ChatBotUI(object): self.history = gr.State(value=[]) self.images = gr.State(value={}) self.history_result = gr.State(value={}) + self.retry_msg = gr.State(value='') with gr.Group(): with gr.Row(equal_height=True): with gr.Column(visible=True) as self.chat_page: @@ -595,7 +595,7 @@ class ChatBotUI(object): video_fps, video_seed, progress=gr.Progress(track_tqdm=True)): - self.retry_msg = message + retry_msg = message gen_id = get_md5(message)[:12] save_path = os.path.join(self.cache_dir, f'{gen_id}.png') @@ -768,7 +768,8 @@ class ChatBotUI(object): history.pop(0) return history, images, history_result, self.get_history( - history), gr.update(value=''), gr.update(visible=False) + history), gr.update(value=''), gr.update( + visible=False), retry_msg chat_inputs = [ self.extend_prompt, self.history, self.images, self.use_history, @@ -781,7 +782,7 @@ class ChatBotUI(object): chat_outputs = [ self.history, self.images, self.history_result, self.chatbot, - self.text, self.gallery + self.text, self.gallery, self.retry_msg ] self.chat_btn.click(run_chat, @@ -792,12 +793,11 @@ class ChatBotUI(object): inputs=[self.text] + chat_inputs, outputs=chat_outputs) - ######################################## - def retry_chat(*args): - return run_chat(self.retry_msg, *args) + def retry_fn(*args): + return run_chat(*args) - self.retry_btn.click(retry_chat, - inputs=chat_inputs, + self.retry_btn.click(retry_fn, + inputs=[self.retry_msg] + chat_inputs, outputs=chat_outputs) ########################################