Update app.py

app.py

@@ -668,12 +668,14 @@ def make_request(payload, auth_manager, model_id):
     global multi_auth_manager
     max_retries = 3
     retry_delay = 1
+    accounts_exhausted = False
     
     for _ in range(len(multi_auth_manager.auth_managers)):  # 尝试所有可用的账号
         auth_manager = multi_auth_manager.get_next_auth_manager(model_id)
         if not auth_manager:
             logger.error(f"No available accounts for model {model_id}")
-            raise Exception(f"No available accounts for model {model_id}")
+            accounts_exhausted = True
+            break
 
         for attempt in range(max_retries):
             try:
@@ -713,6 +715,26 @@ def make_request(payload, auth_manager, model_id):
         # If we've exhausted all retries for this account, continue to the next account
         continue
 
+    # 如果所有账号都被耗尽，尝试自动注册
+    if accounts_exhausted:
+        logger.info("尝试自动注册新账号")
+        try:
+            successful_accounts = register_bot.register_and_verify(5)  # 注册5个账号
+            
+            if successful_accounts:
+                # 重新初始化 multi_auth_manager
+                credentials = [(account['email'], account['password']) for account in successful_accounts]
+                multi_auth_manager = MultiAuthManager(credentials)
+                
+                # 重新尝试请求
+                return make_request(payload, None, model_id)
+            else:
+                logger.error("无法自动注册新账号")
+                raise Exception("Failed to register new accounts")
+        except Exception as e:
+            logger.error(f"自动注册过程发生错误: {e}")
+            raise Exception("Failed to make request after trying all accounts")
+
     raise Exception("Failed to make request after trying all accounts")
 
 def health_check():
@@ -745,3 +767,4 @@ if __name__ == "__main__":
     
     port = int(os.environ.get("PORT", 3000))
     app.run(debug=False, host='0.0.0.0', port=port, threaded=True)
+