import os
import shutil
from pathlib import Path
from flask import Flask, render_template, send_file
from update_predictions import Config as up_Config, main_task, get_business_info
app = Flask(__name__)
# 确保与update_predictions.py共享配置
Config = up_Config
# 确保模板目录存在
def initialize_html_template():
# 创建templates目录（如果不存在）
template_dir = Path(__file__).parent / "templates"
template_dir.mkdir(parents=True, exist_ok=True)
# 目标模板路径
dest_html = template_dir / "index.html"
# 如果模板不存在，创建基础模板
if not dest_html.exists():
base_html = """
清华大学Kronos股票K型大模型上证指数预测
清华大学Kronos股票K型大模型上证指数概率预测
最后更新时间：{{ update_time }}（UTC）
同 步 网 站：火狼工具站
24个交易日上涨概率：
{{ upside_prob }}%
波动率放大概率：
{{ vol_amp_prob }}%
"""
with open(dest_html, 'w', encoding='utf-8') as f:
f.write(base_html)
print(f"已创建基础模板文件：{dest_html}")
# 初始化模型（全局一次）
from update_predictions import load_local_model as load_model
predictor = load_model()
# 初始化HTML模板
initialize_html_template()
@app.route("/")
def index():
# 获取当前业务日
current_business_date, _ = get_business_info()
# 容错处理：检查配置键是否存在，不存在则执行主任务
if ("LAST_INFERENCED_BUSINESS_DATE" not in Config
or Config["LAST_INFERENCED_BUSINESS_DATE"] != current_business_date):
main_task(predictor)
# 从配置中获取需要显示的数据
update_time = Config.get("update_time", "未更新")
upside_prob = Config.get("upside_prob", "--")
vol_amp_prob = Config.get("vol_amp_prob", "--")
# 使用Jinja2模板渲染数据
return render_template("index.html",
update_time=update_time,
upside_prob=upside_prob,
vol_amp_prob=vol_amp_prob)
@app.route("/prediction_chart.png")
def get_chart():
# 提供图表访问
chart_path = Path(Config["CHART_PATH"])
if chart_path.exists():
return send_file(chart_path, mimetype='image/png')
else:
return "预测图表生成中，请稍后刷新", 202
if __name__ == "__main__":
app.run(host="0.0.0.0", port=7860)
