修复从UI中粘贴key时，无法显示OpenAI余额的问题

ChuanhuChatbot.py

@@ -274,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, model_select_dropdown], [usageTxt, like_dislike_area], show_progress=False)
+    demo.load(refresh_ui_elements_on_load, [current_model, model_select_dropdown], [like_dislike_area], show_progress=False)
     chatgpt_predict_args = dict(
         fn=predict,
         inputs=[

modules/models.py

@@ -209,6 +209,11 @@ class OpenAIClient(BaseLLMModel):
         if error_msg:
             raise Exception(error_msg)
 
+    def set_key(self, new_access_key):
+        ret = super().set_key(new_access_key)
+        self._refresh_header()
+        return ret
+
 
 class ChatGLM_Client(BaseLLMModel):
     def __init__(self, model_name) -> None:

modules/utils.py

@@ -539,7 +539,7 @@ def get_model_source(model_name, alternative_source):
         return "https://huggingface.co/gpt2-medium"
 
 def refresh_ui_elements_on_load(current_model, selected_model_name):
-    return current_model.billing_info(), toggle_like_btn_visibility(selected_model_name)
+    return toggle_like_btn_visibility(selected_model_name)
 
 def toggle_like_btn_visibility(selected_model_name):
     if selected_model_name == "xmchat":