yangtb24 commited on
Commit
16b7f0b
1 Parent(s): db69584

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -10
app.py CHANGED
@@ -3,6 +3,7 @@ import time
3
  import logging
4
  import requests
5
  from apscheduler.schedulers.background import BackgroundScheduler
 
6
 
7
  # 配置日志记录
8
  logging.basicConfig(level=logging.INFO,
@@ -10,6 +11,8 @@ logging.basicConfig(level=logging.INFO,
10
 
11
  API_ENDPOINT = "https://api.siliconflow.cn/v1/usage/credit_summary"
12
 
 
 
13
  def get_credit_summary(api_key):
14
  """
15
  使用 API 密钥获取额度信息。
@@ -54,17 +57,22 @@ scheduler = BackgroundScheduler()
54
  # 添加定时任务,每小时执行一次 load_keys 函数
55
  scheduler.add_job(load_keys, 'interval', hours=1)
56
 
 
 
 
 
 
 
 
57
  if __name__ == '__main__':
58
  logging.info(f"===== Application Startup at {time.strftime('%Y-%m-%d %H:%M:%S')} =====")
59
- try:
60
- # 启动调度器
61
- scheduler.start()
62
 
63
- # 手动触发一次 load_keys 任务
64
- load_keys()
65
- logging.info("首次加载 keys 已手动触发执行")
 
 
 
66
 
67
- while True:
68
- time.sleep(120)
69
- except (KeyboardInterrupt, SystemExit):
70
- scheduler.shutdown()
 
3
  import logging
4
  import requests
5
  from apscheduler.schedulers.background import BackgroundScheduler
6
+ from flask import Flask
7
 
8
  # 配置日志记录
9
  logging.basicConfig(level=logging.INFO,
 
11
 
12
  API_ENDPOINT = "https://api.siliconflow.cn/v1/usage/credit_summary"
13
 
14
+ app = Flask(__name__)
15
+
16
  def get_credit_summary(api_key):
17
  """
18
  使用 API 密钥获取额度信息。
 
57
  # 添加定时任务,每小时执行一次 load_keys 函数
58
  scheduler.add_job(load_keys, 'interval', hours=1)
59
 
60
+ @app.route('/')
61
+ def index():
62
+ """
63
+ 处理根路由的访问请求。
64
+ """
65
+ return "<h1>Welcome to SiliconFlow</h1>"
66
+
67
  if __name__ == '__main__':
68
  logging.info(f"===== Application Startup at {time.strftime('%Y-%m-%d %H:%M:%S')} =====")
 
 
 
69
 
70
+ # 启动调度器
71
+ scheduler.start()
72
+
73
+ # 手动触发一次 load_keys 任务
74
+ load_keys()
75
+ logging.info("首次加载 keys 已手动触发执行")
76
 
77
+ # 启动 Flask 应用,监听所有 IP 的 5000 端口
78
+ app.run(debug=False, host='0.0.0.0', port=5000)