Upload app.py

app.py

@@ -670,6 +670,21 @@ def make_request(payload, auth_manager, model_id):
     retry_delay = 1
     accounts_exhausted = False
     
+    logger.info(f"尝试发送请求，模型：{model_id}")
+    
+    # 首先检查 multi_auth_manager 是否存在
+    if not multi_auth_manager:
+        logger.error("MultiAuthManager 不存在，尝试重新初始化")
+        credentials = get_auth_credentials()
+        if not credentials:
+            logger.error("无法获取凭据，尝试注册新账号")
+            successful_accounts = register_bot.register_and_verify(5)
+            if successful_accounts:
+                credentials = [(account['email'], account['password']) for account in successful_accounts]
+                multi_auth_manager = MultiAuthManager(credentials)
+            else:
+                raise Exception("无法注册新账号")
+    
     for _ in range(len(multi_auth_manager.auth_managers)):  # 尝试所有可用的账号
         auth_manager = multi_auth_manager.get_next_auth_manager(model_id)
         if not auth_manager:
@@ -715,13 +730,15 @@ 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("尝试自动注册新账号")
+        logger.info("所有账号已耗尽，强制尝试注册新账号")
         try:
+            # 直接调用注册方法
             successful_accounts = register_bot.register_and_verify(5)  # 注册5个账号
             
             if successful_accounts:
+                logger.info(f"成功注册 {len(successful_accounts)} 个新账号")
                 # 重新初始化 multi_auth_manager
                 credentials = [(account['email'], account['password']) for account in successful_accounts]
                 multi_auth_manager = MultiAuthManager(credentials)
@@ -733,7 +750,7 @@ def make_request(payload, auth_manager, model_id):
                 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 and registration")
 
     raise Exception("Failed to make request after trying all accounts")