Upload 4 files

Dockerfile

@@ -8,4 +8,6 @@ COPY ./app.py /code/app.py
 
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
+EXPOSE 7860
+
 CMD ["python", "app.py"]

README.md

@@ -6,5 +6,5 @@ colorTo: indigo
 sdk: docker
 pinned: false
 license: mit
-app_port: 8000
+app_port: 7860
 ---

app.py

@@ -1,6 +1,11 @@
 import requests
 import schedule
 import time
+import os
+
+import threading
+from fastapi import FastAPI
+app = FastAPI()
 
 def check_available_tokens():
     """
@@ -46,7 +51,7 @@ def read_and_prepare_tokens(file_path):
 
 def post_tokens():
     """
-    读取tokens并发送单次POST请求
+    检查, 读取tokens并发送单次POST请求
     """
     # 首先检查是否需要发送 tokens
     if not check_available_tokens():
@@ -75,17 +80,35 @@ def post_tokens():
     except Exception as e:
         print(f"请求发生错误: {e}")
 
+def run_schedule():
+    """
+    运行定时任务
+    """
+    while True:
+        schedule.run_pending()
+        time.sleep(1)
+
+@app.on_event("startup")
+async def startup_event():
+    """
+    应用启动时初始化定时任务
+    """
+    schedule.every(5).minutes.do(post_tokens)
+    # 在新线程中运行定时任务
+    threading.Thread(target=run_schedule, daemon=True).start()
+
+@app.get("/")
 def main():
     # 立即执行一次
-    post_tokens()
+    #post_tokens()
     
     # 每5分钟执行一次
-    schedule.every(5).minutes.do(post_tokens)
+    #schedule.every(5).minutes.do(post_tokens)
     
     # 持续运行
-    while True:
-        schedule.run_pending()
-        time.sleep(1)
+    #run_schedule()
 
 if __name__ == "__main__":
-    main()
+    #main()
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)

requirements.txt

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