Upload 6 files

README.md

@@ -10,3 +10,5 @@ app_port: 3000
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+
+[email protected]|password123

app.py

@@ -6,6 +6,7 @@ import time
 import uuid
 import re
 import socket
+import subprocess  # 新增 subprocess 模块用于执行外部脚本
 from concurrent.futures import ThreadPoolExecutor
 from functools import lru_cache, wraps
 from typing import Dict, Any, Callable, List, Tuple
@@ -650,7 +651,39 @@ def make_request(payload, auth_manager, model_id):
         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}")
+            
+            # 尝试自动注册新账号
+            try:
+                register_bot_path = os.path.join(os.path.dirname(__file__), 'register_bot.py')
+                
+                # 执行 register_bot.py 脚本
+                result = subprocess.run(
+                    [sys.executable, register_bot_path],  # 使用当前 Python 解释器执行脚本
+                    capture_output=True,  # 捕获标准输出和标准错误
+                    text=True,  # 返回文本输出而非字节
+                    timeout=600  # 设置超时时间为10分钟
+                )
+                
+                # 记录注册脚本的输出
+                logger.info(f"Register bot stdout: {result.stdout}")
+                logger.error(f"Register bot stderr: {result.stderr}")
+                
+                # 重新初始化 multi_auth_manager
+                credentials = get_auth_credentials()
+                if credentials:
+                    multi_auth_manager = MultiAuthManager(credentials)
+                
+                # 重新尝试获取账号
+                auth_manager = multi_auth_manager.get_next_auth_manager(model_id)
+                if not auth_manager:
+                    raise Exception("Failed to register new accounts")
+                
+            except subprocess.CalledProcessError as e:
+                logger.error(f"Error executing register_bot.py: {e}")
+                raise Exception(f"No available accounts for model {model_id} and failed to register new accounts")
+            except Exception as e:
+                logger.error(f"Unexpected error during account registration: {e}")
+                raise Exception(f"No available accounts for model {model_id} and failed to register new accounts")
 
         for attempt in range(max_retries):
             try:

gitattributes

@@ -0,0 +1,35 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

output.txt

@@ -0,0 +1 @@
+[email protected]|zhoudan9.28;[email protected]|zhoudan9.28;[email protected]|zhoudan9.28;[email protected]|zhoudan9.28;[email protected]|zhoudan9.28;[email protected]|zhoudan9.28;[email protected]|zhoudan9.28;[email protected]|zhoudan9.28;[email protected]|zhoudan9.28;[email protected]|zhoudan9.28;[email protected]|zhoudan9.28;[email protected]|zhoudan9.28;[email protected]|password123;[email protected]|password123;[email protected]|password123;[email protected]|password123;[email protected]|password123;[email protected]|password123;[email protected]|password123;[email protected]|password123;[email protected]|password123;[email protected]|password123;[email protected]|password123;[email protected]|password123;[email protected]|password123;[email protected]|password123;[email protected]|password123;[email protected]|password123;[email protected]|password123;[email protected]|password123;[email protected]|password123;[email protected]|password123;[email protected]|password123;[email protected]|password123;[email protected]|password123;[email protected]|password123

register_bot.py

@@ -0,0 +1,281 @@
+import requests
+from bs4 import BeautifulSoup
+from requests.adapters import HTTPAdapter
+from requests.packages.urllib3.util.retry import Retry
+import uuid
+import json
+import time
+import re
+import hashlib
+import base64
+import random
+import string
+import os
+import sys
+from urllib.parse import quote, urlparse, parse_qs
+import concurrent.futures
+import logging
+
+# 获取当前脚本所在目录
+SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
+
+# 配置日志
+try:
+    LOG_FILE = os.path.join(SCRIPT_DIR, 'registration_log.txt')
+    logging.basicConfig(level=logging.INFO, 
+                        format='%(asctime)s - %(levelname)s - %(message)s', 
+                        filename=LOG_FILE, 
+                        filemode='a')
+except PermissionError:
+    # 如果无法写入文件，使用控制台日志
+    logging.basicConfig(level=logging.INFO, 
+                        format='%(asctime)s - %(levelname)s - %(message)s', 
+                        stream=sys.stdout)
+    logging.warning("Could not write to log file. Logging to console instead.")
+
+# 获取环境变量
+PASTE_API_URL = os.getenv('PASTE_API_URL')
+PASTE_API_PASSWORD = os.getenv('PASTE_API_PASSWORD')
+
+# 构建上传 URL
+UPLOAD_URL = PASTE_API_URL.replace('/api/paste/', '/api/admin/paste/') + '/content'
+
+# 新的临时邮箱API
+TEMP_EMAIL_API = "https://www.1secmail.com/api/v1/?action=genRandomMailbox"
+
+# 注册 API 地址
+REGISTER_API_URL = "https://spuckhogycrxcbomznwo.supabase.co/auth/v1/signup?redirect_to=https%3A%2F%2Fchat.notdiamond.ai"
+
+# Supabase API Key
+SUPABASE_API_KEY = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJzdXBhYmFzZSIsInJlZiI6InNwdWNraG9neWNyeGNib216bndvIiwicm9sZSI6ImFub24iLCJpYXQiOjE3MDcyNDYwMzksImV4cCI6MjAyMjgyMjAzOX0.tvlGT7NZY8bijMjNIu1WhAtPnSKuDeYhtveo4DRt6xg"
+
+# 请求限制和重试相关配置
+MAX_RETRIES = 3
+RETRY_DELAY = 10  # 秒
+RATE_LIMIT_DELAY = 60  # 秒
+
+def generate_strong_password():
+    """固定密码为password123"""
+    return "password123"
+
+def generate_code_verifier():
+    return base64.urlsafe_b64encode(uuid.uuid4().bytes).decode('utf-8').rstrip('=')
+
+def generate_code_challenge(code_verifier):
+    code_challenge = hashlib.sha256(code_verifier.encode('utf-8')).digest()
+    return base64.urlsafe_b64encode(code_challenge).decode('utf-8').rstrip('=')
+
+def get_temp_email():
+    session = requests.Session()
+    retries = Retry(total=5, backoff_factor=1, status_forcelist=[500, 502, 503, 504])
+    session.mount('https://', HTTPAdapter(max_retries=retries))
+    
+    try:
+        response = session.get(TEMP_EMAIL_API)
+        if response.status_code == 200:
+            return response.json()[0]
+    except requests.exceptions.SSLError as e:
+        logging.error(f"SSL Error: {e}")
+    return None
+
+def register(email, password):
+    for attempt in range(MAX_RETRIES):
+        session = requests.Session()
+        headers = {
+            'apikey': SUPABASE_API_KEY,
+            'authorization': f'Bearer {SUPABASE_API_KEY}',
+            'content-type': 'application/json'
+        }
+        
+        code_verifier = generate_code_verifier()
+        code_challenge = generate_code_challenge(code_verifier)
+        
+        payload = {
+            'email': email,
+            'password': password,
+            'code_challenge': code_challenge,
+            'code_challenge_method': 's256',
+            'data': {},
+            'gotrue_meta_security': {}
+        }
+        
+        try:
+            response = session.post(REGISTER_API_URL, headers=headers, json=payload)
+            
+            if response.status_code == 200:
+                logging.info(f"Registered with email: {email}")
+                return True
+            elif response.status_code == 429:
+                logging.warning(f"Rate limit reached. Waiting {RATE_LIMIT_DELAY} seconds.")
+                time.sleep(RATE_LIMIT_DELAY)
+            else:
+                logging.error(f"Failed to register with email: {email}. Response: {response.text}")
+                return False
+        
+        except requests.exceptions.RequestException as e:
+            logging.error(f"Request error: {e}")
+            if attempt < MAX_RETRIES - 1:
+                logging.info(f"Retrying in {RETRY_DELAY} seconds...")
+                time.sleep(RETRY_DELAY)
+            else:
+                logging.error("Max retries reached.")
+                return False
+    
+    return False
+
+def save_account(email, password):
+    # 上传到指定 URL
+    try:
+        # 先获取现有内容
+        session = requests.Session()
+        
+        # 首先进行认证
+        auth_url = PASTE_API_URL
+        auth_headers = {
+            'accept': '*/*',
+            'accept-language': 'zh-CN,zh;q=0.9',
+            'user-agent': 'Mozilla/5.0',
+            'x-password': PASTE_API_PASSWORD
+        }
+        
+        # 获取现��内容
+        response = session.get(auth_url, headers=auth_headers)
+        
+        if response.status_code != 200:
+            logging.error(f"认证失败，状态码：{response.status_code}")
+            return
+        
+        # 从响应获取现有内容
+        existing_data = response.json()
+        existing_content = existing_data.get('content', '')
+        
+        # 构建新的内容
+        if existing_content:
+            new_content = f"{existing_content};{email}|{password}"
+        else:
+            new_content = f"{email}|{password}"
+        
+        # 上传新内容
+        upload_headers = {
+            'Authorization': 'Basic emhvdWRhbjp6aG91ZGFu',
+            'Content-Type': 'application/json'
+        }
+        
+        upload_payload = {
+            'content': new_content
+        }
+        
+        upload_response = session.put(UPLOAD_URL, headers=upload_headers, json=upload_payload)
+        
+        if upload_response.status_code == 200:
+            logging.info(f"Successfully uploaded account {email}")
+        else:
+            logging.error(f"Failed to upload account {email}. Status code: {upload_response.status_code}")
+    
+    except Exception as e:
+        logging.error(f"Error uploading account: {e}")
+
+def check_email(email):
+    domain = email.split('@')[1]
+    login = email.split('@')[0]
+    
+    # 获取消息ID
+    mailbox_url = f"https://www.1secmail.com/api/v1/?action=getMessages&login={login}&domain={domain}"
+    
+    session = requests.Session()
+    retries = Retry(total=5, backoff_factor=1, status_forcelist=[500, 502, 503, 504])
+    session.mount('https://', HTTPAdapter(max_retries=retries))
+    
+    time.sleep(10)
+    
+    for _ in range(20):
+        response = session.get(mailbox_url)
+        if response.status_code == 200 and response.json():
+            for message in response.json():
+                if "Confirm Your Signup" in message.get('subject', ''):
+                    # 获取完整邮件内容
+                    message_url = f"https://www.1secmail.com/api/v1/?action=readMessage&login={login}&domain={domain}&id={message['id']}"
+                    message_response = session.get(message_url)
+                    
+                    if message_response.status_code == 200:
+                        full_message = message_response.json()
+                        logging.info(f"Full message content: {full_message}")
+                        
+                        # 提取验证链接
+                        match = re.search(r'https?://chat\.notdiamond\.ai/auth/confirm\?[^\s]+', full_message.get('body', ''))
+                        if match:
+                            verification_link = match.group(0)
+                            logging.info(f"Verification link found: {verification_link}")
+                            
+                            # 解析 URL 并获取参数
+                            parsed_url = urlparse(verification_link)
+                            query_params = parse_qs(parsed_url.query)
+                            
+                            # 使用 Supabase API 进行验证
+                            verify_url = "https://spuckhogycrxcbomznwo.supabase.co/auth/v1/verify"
+                            headers = {
+                                'apikey': SUPABASE_API_KEY,
+                                'authorization': f'Bearer {SUPABASE_API_KEY}',
+                                'content-type': 'application/json'
+                            }
+                            
+                            payload = {
+                                'token_hash': query_params['token_hash'][0],
+                                'type': 'signup'  # 硬编码为 'signup'
+                            }
+                            
+                            verify_response = session.post(verify_url, headers=headers, json=payload)
+                            
+                            if verify_response.status_code == 200:
+                                logging.info(f"Email verified for {email}")
+                                return True
+                            else:
+                                logging.error(f"Failed to verify email for {email}: {verify_response.text}")
+                                return False
+                        else:
+                            logging.error(f"Failed to extract verification link from email for {email}")
+                            return False
+        time.sleep(5)
+    logging.warning(f"No verification email found for {email}")
+    return False
+
+def register_and_verify(num_accounts=1):
+    successful_accounts = []
+    with concurrent.futures.ThreadPoolExecutor(max_workers=2) as executor:  # 降低并发数
+        futures = []
+        for _ in range(num_accounts):
+            email = get_temp_email()
+            if email:
+                futures.append(executor.submit(process_account, email))
+                time.sleep(RETRY_DELAY)  # 添加延迟
+        
+        for future in concurrent.futures.as_completed(futures):
+            result = future.result()
+            if result:
+                successful_accounts.append(result)
+    
+    return successful_accounts
+
+def process_account(email):
+    try:
+        # 使用固定密码
+        password = generate_strong_password()
+        
+        if register(email, password):
+            save_account(email, password)
+            if check_email(email):
+                return {'email': email, 'password': password}
+    except Exception as e:
+        logging.error(f"Error processing account {email}: {e}")
+    return None
+
+def main():
+    num_accounts = 5  # 可以根据需要修改注册数量
+    successful_accounts = register_and_verify(num_accounts)
+    logging.info(f"Successfully registered {len(successful_accounts)} accounts:")
+    for account in successful_accounts:
+        logging.info(f"Email: {account['email']}, Password: {account['password']}")
+    print(f"Successfully registered {len(successful_accounts)} accounts. Check registration_log.txt for details.")
+
+if __name__ == "__main__":
+    main()