为XMChat加入点赞/点踩的功能

ChuanhuChatbot.py

@@ -67,6 +67,11 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
                 retryBtn = gr.Button(i18n("🔄 重新生成"))
                 delFirstBtn = gr.Button(i18n("🗑️ 删除最旧对话"))
                 delLastBtn = gr.Button(i18n("🗑️ 删除最新对话"))
+                with gr.Row(visible=False) as like_dislike_area:
+                    with gr.Column(min_width=20, scale=1):
+                        likeBtn = gr.Button(i18n("👍"))
+                    with gr.Column(min_width=20, scale=1):
+                        dislikeBtn = gr.Button(i18n("👎"))
 
         with gr.Column():
             with gr.Column(min_width=50, scale=1):
@@ -269,7 +274,7 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
 
     gr.Markdown(CHUANHU_DESCRIPTION, elem_id="description")
     gr.HTML(FOOTER.format(versions=versions_html()), elem_id="footer")
-    demo.load(refresh_ui_elements_on_load, [current_model], [usageTxt], show_progress=False)
+    demo.load(refresh_ui_elements_on_load, [current_model, model_select_dropdown], [usageTxt, like_dislike_area], show_progress=False)
     chatgpt_predict_args = dict(
         fn=predict,
         inputs=[
@@ -361,6 +366,20 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
         show_progress=False
     )
 
+    likeBtn.click(
+        like,
+        [current_model],
+        [status_display],
+        show_progress=False
+    )
+
+    dislikeBtn.click(
+        dislike,
+        [current_model],
+        [status_display],
+        show_progress=False
+    )
+
     two_column.change(update_doc_config, [two_column], None)
 
     # LLM Models
@@ -368,6 +387,7 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
     keyTxt.submit(**get_usage_args)
     single_turn_checkbox.change(set_single_turn, [current_model, single_turn_checkbox], None)
     model_select_dropdown.change(get_model, [model_select_dropdown, lora_select_dropdown, user_api_key, temperature_slider, top_p_slider, systemPromptTxt], [current_model, status_display, lora_select_dropdown], show_progress=True)
+    model_select_dropdown.change(toggle_like_btn_visibility, [model_select_dropdown], [like_dislike_area], show_progress=False)
     lora_select_dropdown.change(get_model, [model_select_dropdown, lora_select_dropdown, user_api_key, temperature_slider, top_p_slider, systemPromptTxt], [current_model, status_display], show_progress=True)
 
     # Template

modules/base_model.py

@@ -549,3 +549,13 @@ class BaseLLMModel:
         except FileNotFoundError:
             logging.warning(f"{user_name} 没有找到对话历史文件，不执行任何操作")
             return filename, self.system_prompt, chatbot
+
+    def like(self):
+        """like the last response, implement if needed
+        """
+        return gr.update()
+
+    def dislike(self):
+        """dislike the last response, implement if needed
+        """
+        return gr.update()

modules/models.py

@@ -99,6 +99,8 @@ class OpenAIClient(BaseLLMModel):
             status_text = STANDARD_ERROR_MSG + READ_TIMEOUT_MSG + ERROR_RETRIEVE_MSG
             return status_text
         except Exception as e:
+            import traceback
+            traceback.print_exc()
             logging.error(i18n("获取API使用情况失败:") + str(e))
             return STANDARD_ERROR_MSG + ERROR_RETRIEVE_MSG
 
@@ -395,9 +397,11 @@ class XMChat(BaseLLMModel):
         self.image_path = None
         self.xm_history = []
         self.url = "https://xmbot.net/web"
+        self.last_conv_id = None
 
     def reset(self):
         self.session_id = str(uuid.uuid4())
+        self.last_conv_id = None
         return [], "已重置"
 
     def image_to_base64(self, image_path):
@@ -444,6 +448,26 @@ class XMChat(BaseLLMModel):
             self.image_bytes = None
             self.image_path = None
 
+    def like(self):
+        if self.last_conv_id is None:
+            return "点赞失败，你还没发送过消息"
+        data = {
+            "uuid": self.last_conv_id,
+            "appraise": "good"
+        }
+        response = requests.post(self.url, json=data)
+        return "👍点赞成功，，感谢反馈～"
+
+    def dislike(self):
+        if self.last_conv_id is None:
+            return "点踩失败，你还没发送过消息"
+        data = {
+            "uuid": self.last_conv_id,
+            "appraise": "bad"
+        }
+        response = requests.post(self.url, json=data)
+        return "👎点踩成功，感谢反馈～"
+
     def prepare_inputs(self, real_inputs, use_websearch, files, reply_language, chatbot):
         fake_inputs = real_inputs
         display_append = ""
@@ -479,6 +503,7 @@ class XMChat(BaseLLMModel):
     def get_answer_at_once(self):
         question = self.history[-1]["content"]
         conv_id = str(uuid.uuid4())
+        self.last_conv_id = conv_id
         data = {
             "user_id": self.api_key,
             "session_id": self.session_id,

modules/utils.py

@@ -113,6 +113,12 @@ def set_single_turn(current_model, *args):
 def handle_file_upload(current_model, *args):
     return current_model.handle_file_upload(*args)
 
+def like(current_model, *args):
+    return current_model.like(*args)
+
+def dislike(current_model, *args):
+    return current_model.dislike(*args)
+
 
 def count_token(message):
     encoding = tiktoken.get_encoding("cl100k_base")
@@ -532,5 +538,11 @@ def get_model_source(model_name, alternative_source):
     if model_name == "gpt2-medium":
         return "https://huggingface.co/gpt2-medium"
 
-def refresh_ui_elements_on_load(current_model):
-    return current_model.billing_info()
+def refresh_ui_elements_on_load(current_model, selected_model_name):
+    return current_model.billing_info(), toggle_like_btn_visibility(selected_model_name)
+
+def toggle_like_btn_visibility(selected_model_name):
+    if selected_model_name == "xmchat":
+        return gr.update(visible=True)
+    else:
+        return gr.update(visible=False)