Upload 6 files

Dockerfile

@@ -0,0 +1,14 @@
+FROM python:3.9-slim
+
+WORKDIR /code
+
+COPY ./requirements.txt /code/requirements.txt
+COPY ./access_token.txt /code/access_token.txt
+COPY ./app.py /code/app.py
+COPY ./at.py /code/at.py
+
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+EXPOSE 7860
+
+CMD ["python", "app.py"]

README.md

@@ -1,10 +1,10 @@
 ---
-title: At
-emoji: 🔥
-colorFrom: pink
+title: at
+emoji: 🌍
+colorFrom: yellow
 colorTo: indigo
 sdk: docker
 pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+license: mit
+app_port: 7860
+---

app.py

@@ -0,0 +1,90 @@
+import requests
+import schedule
+import time
+import os
+
+import threading
+from fastapi import FastAPI
+app = FastAPI()
+
+# 从环境变量获取基础 URL
+BASE_URL = os.getenv('BASE_URL', "")
+
+def check_available_tokens():
+    """
+    检查可用的 Tokens 数量
+    """
+    try:
+        response = requests.get(f"{BASE_URL}")
+        if response.status_code == 200:
+            content = response.text
+            if '当前可用 Tokens 数量：<span class="text-blue-600">0' in content:
+                return True
+        return False
+    except Exception as e:
+        print(f"检查 Tokens 时发生错误: {e}")
+        return False
+
+def read_and_prepare_tokens(file_path):
+    """
+    从文件读取tokens并使用\r\n连接
+    """
+    try:
+        # 适配 Hugging Face Spaces 的文件路径
+        if os.path.exists(file_path):
+            with open(file_path, 'r') as file:
+                # 去除每行的空白字符，过滤掉空行
+                tokens = [line.strip() for line in file if line.strip()]
+        else:
+            # 如果文件不存在，尝试在 /home/user 目录下查找
+            alternative_path = os.path.join('/home/user', file_path)
+            with open(alternative_path, 'r') as file:
+                tokens = [line.strip() for line in file if line.strip()]
+        
+        # 使用\r\n连接tokens
+        combined_token = '\r\n'.join(tokens)
+        return combined_token
+    
+    except FileNotFoundError:
+        print(f"错误：文件 {file_path} 未找到")
+        return None
+    except Exception as e:
+        print(f"读取文件发生错误: {e}")
+        return None
+
+def post_tokens():
+    """
+    读取tokens并发送单次POST请求
+    """
+    url = f"{BASE_URL}/upload"
+    
+    # 读取并准备tokens
+    combined_token = read_and_prepare_tokens('access_token.txt')
+    
+    if not combined_token:
+        print("没有可用的tokens")
+        # 触发一次at.py环境变量读取生成access_token.txt
+        os.system('python at.py')
+        return
+    
+    try:
+        payload = {"text": combined_token}
+        response = requests.post(url, data=payload)
+        
+        #print(f"响应状态码: {response.status_code}")
+        print(f"响应内容: {response.text}")
+        
+    except Exception as e:
+        print(f"请求发生错误: {e}")
+
+@app.get("/")
+async def main():
+    # 首先检查是否需要发送 tokens
+    if not check_available_tokens():
+        print("当前仍有可用 Tokens，无需发送")
+        return
+    return {"message": post_tokens()}
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)

at.py

@@ -0,0 +1,70 @@
+import requests
+import json
+
+# 定义常量
+TOKEN_URL_NEO = 'https://token.oaifree.com/api/auth/refresh'
+TOKEN_URL_OAI = 'https://auth0.openai.com/oauth/token'
+
+def get_access_token(refresh_token, mode):
+    """
+    获取 access token 的函数
+    
+    :param refresh_token: refresh token
+    :param mode: 'neo' 或 'oai'
+    :return: access token 或错误信息
+    """
+    url = TOKEN_URL_OAI if mode == 'oai' else TOKEN_URL_NEO
+    
+    if mode == 'neo':
+        headers = {'Content-Type': 'application/x-www-form-urlencoded;charset=UTF-8'}
+        data = f'refresh_token={refresh_token}'
+    else:
+        headers = {'Content-Type': 'application/json'}
+        data = json.dumps({
+            'redirect_uri': "com.openai.chat://auth0.openai.com/ios/com.openai.chat/callback",
+            'grant_type': "refresh_token",
+            'client_id': "pdlLIX2Y72MIl2rhLhTE9VV9bN905kBh",
+            'refresh_token': refresh_token
+        })
+    
+    try:
+        response = requests.post(url, headers=headers, data=data)
+        result = response.json()
+        
+        if 'access_token' in result:
+            return result['access_token']
+        else:
+            return f"获取失败：{result.get('detail', '未知错误')}"
+    
+    except requests.RequestException as e:
+        return f"请求错误：{str(e)}"
+
+def batch_get_access_tokens(input_file='refresh_token.txt', output_file='access_token.txt'):
+    """
+    批量获取 access tokens
+    
+    :param input_file: 包含 refresh tokens 的输入文件
+    :param output_file: 写入 access tokens 的输出文件
+    """
+    # 从环境变量读取 refresh tokens
+    refresh_tokens = os.getenv('REFRESH_TOKENS', '')
+    if refresh_tokens:
+        refresh_tokens = refresh_tokens.split(',')
+    else:
+        # 如果没有环境变量，使用默认文件读取
+        with open(input_file, 'r') as f_in:
+            refresh_tokens = [line.strip() for line in f_in]
+
+    with open(output_file, 'w') as f_out:
+        for refresh_token in refresh_tokens:
+            # 默认使用 'neo' 模式，可以根据需要修改
+            access_token = get_access_token(refresh_token, mode='oai')
+            
+            # 写入 access token
+            f_out.write(f"{access_token}\n")
+    
+    print(f"Access tokens 已写入 {output_file}")
+
+# 主程序入口
+if __name__ == "__main__":
+    batch_get_access_tokens()

requirements.txt

@@ -0,0 +1,4 @@
+requests
+schedule
+fastapi
+uvicorn