Spaces:

fiewolf1000
/

Astock

Running

App Files Files Community

fiewolf1000 commited on Sep 2

Commit

4a918c0

verified ·

1 Parent(s): c2b2703

Update app.py

Browse files

Files changed (1) hide show

app.py +58 -71

app.py CHANGED Viewed

@@ -1,72 +1,60 @@
 import os
 import shutil
 from pathlib import Path
-from flask import Flask, render_template_string, send_file
-from update_predictions import Config, main_task, get_business_info
 app = Flask(__name__)
-# 统一配置（与update_predictions.py保持一致）
-Config = {
-    "REPO_PATH": Path(__file__).parent.resolve(),
-    "MODEL_PATH": os.path.join("/tmp", "Kronos_model"),
-    # 关键修改：HTML和图表都存放在/tmp目录（可写）
-    "HTML_PATH": os.path.join("/tmp", "index.html"),
-    "CHART_PATH": os.path.join("/tmp", "prediction_chart.png")
-}
-# 确保/tmp目录下有模板文件
 def initialize_html_template():
-    # 源模板路径（项目中的原始index.html）
-    src_html = Config["REPO_PATH"] / "templates" / "index.html"
-    # 目标路径（/tmp下的可写副本）
-    dest_html = Path(Config["HTML_PATH"])
-    # 如果目标不存在，从源模板复制
     if not dest_html.exists():
-        # 确保/tmp目录存在
-        dest_html.parent.mkdir(parents=True, exist_ok=True)
-        if src_html.exists():
-            shutil.copy2(src_html, dest_html)
-            print(f"已复制模板文件到可写目录：{dest_html}")
-        else:
-            # 如果源模板也不存在，创建基础HTML
-            base_html = """
-            <!DOCTYPE html>
-            <html>
-            <head>
-                <title>清华大学Kronos股票K型大模型上证指数预测</title>
-                <style>
-                    body { font-family: Arial, sans-serif; max-width: 1200px; margin: 0 auto; padding: 20px; }
-                    .metric { margin: 20px 0; padding: 10px; background: #f5f5f5; border-radius: 5px; }
-                    .metric-label { font-weight: bold; color: #333; }
-                    .metric-value { font-size: 1.5em; color: #2c3e50; }
-                    #chart-container { margin-top: 30px; }
-                    img { max-width: 100%; height: auto; }
-                </style>
-            </head>
-            <body>
-                <h1>清华大学Kronos股票K型大模型上证指数概率预测</h1>
-                <p>最后更新时间：<strong id="update-time">未更新</strong>（UTC）</p>
-                <p>同   步     网     站：<strong><a href="http://15115656.top" target="_blank">火狼工具站</a></strong></p>
-                <div class="metric">
-                    <p class="metric-label">24个交易日上涨概率：</p>
-                    <p class="metric-value" id="upside-prob">--%</p>
-                </div>
-                <div class="metric">
-                    <p class="metric-label">波动率放大概率：</p>
-                    <p class="metric-value" id="vol-amp-prob">--%</p>
-                </div>
-                <div id="chart-container">
-                    <img src="/prediction_chart.png" alt="上证指数预测图表">
-                </div>
-            </body>
-            </html>
-            """
-            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
@@ -76,27 +64,25 @@ predictor = load_model()
 initialize_html_template()
 @app.route("/")
 def index():
     # 获取当前业务日
     current_business_date, _ = get_business_info()
     # 容错处理：检查配置键是否存在，不存在则执行主任务
-    # 同时处理键存在但值为None的情况（首次运行）
     if ("LAST_INFERENCED_BUSINESS_DATE" not in Config
         or Config["LAST_INFERENCED_BUSINESS_DATE"] != current_business_date):
         main_task(predictor)
-    # 读取并返回 HTML
-    try:
-        with open(Config["HTML_PATH"], 'r', encoding='utf-8') as f:
-            html_content = f.read()
-        return render_template_string(html_content)
-    except FileNotFoundError:
-        # 处理HTML文件不存在的情况
-        return "预测页面生成中，请稍后刷新", 202
-    except Exception as e:
-        return f"页面加载错误：{str(e)}", 500
 @app.route("/prediction_chart.png")
@@ -110,3 +96,4 @@ def get_chart():
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)

 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 = """
+        <!DOCTYPE html>
+        <html>
+        <head>
+            <title>清华大学Kronos股票K型大模型上证指数预测</title>
+            <style>
+                body { font-family: Arial, sans-serif; max-width: 1200px; margin: 0 auto; padding: 20px; }
+                .metric { margin: 20px 0; padding: 10px; background: #f5f5f5; border-radius: 5px; }
+                .metric-label { font-weight: bold; color: #333; }
+                .metric-value { font-size: 1.5em; color: #2c3e50; }
+                #chart-container { margin-top: 30px; }
+                img { max-width: 100%; height: auto; }
+            </style>
+        </head>
+        <body>
+            <h1>清华大学Kronos股票K型大模型上证指数概率预测</h1>
+            <p>最后更新时间：<strong>{{ update_time }}</strong>（UTC）</p>
+            <p>同   步     网     站：<strong><a href="http://15115656.top" target="_blank">火狼工具站</a></strong></p>
+            <div class="metric">
+                <p class="metric-label">24个交易日上涨概率：</p>
+                <p class="metric-value">{{ upside_prob }}%</p>
+            </div>
+            <div class="metric">
+                <p class="metric-label">波动率放大概率：</p>
+                <p class="metric-value">{{ vol_amp_prob }}%</p>
+            </div>
+            <div id="chart-container">
+                <img src="/prediction_chart.png" alt="上证指数预测图表">
+            </div>
+        </body>
+        </html>
+        """
+        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
 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")
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)