注册后赠送12次基础访问

chatgpt-next-web/service.py

@@ -97,6 +97,9 @@ def register():
 
     # 清除验证码
     redis.delete(email)
+    
+    # 注册免费赠送12次
+    pick_up_free_chat_count(user_id, 12)
 
     return jsonify({
         'code': 0,
@@ -762,7 +765,7 @@ def initialize_user_free_data(user_id):
 
 
 # 每天领取免费次数
-def pick_up_free_chat_count(user_id):
+def pick_up_free_chat_count(user_id, basic_free_count = 5):
     free_data = get_user_free_data(user_id)
     if not free_data:
         free_data = initialize_user_free_data(user_id)
@@ -782,7 +785,6 @@ def pick_up_free_chat_count(user_id):
     if has_pick_up_free and last_gift_time.decode() == str(datetime.date.today()):
         return False
 
-    basic_free_count = 5
     # 更新用户聊天数据和领取状态
     user_free_name = f'user:{user_id}:free'
     redis.hset(user_free_name, b'basic_chat_count', basic_free_count)