0

0

Flask 中异步训练模型后无法重定向到结果页面的解决方案

碧海醫心

碧海醫心

发布时间:2026-02-07 15:50:53

|

169人浏览过

|

来源于php中文网

原创

Flask 中异步训练模型后无法重定向到结果页面的解决方案

在 flask 中使用线程执行耗时模型训练时,`render_template()` 无法从后台线程直接返回响应;必须通过主线程完成渲染,推荐使用全局缓存 + 轮询或 celery 等任务队列实现真正的异步响应。

你当前的代码逻辑存在一个根本性误解:HTTP 响应只能由处理请求的主线程(即 inputs() 视图函数)生成并返回给客户端。train_models() 在新线程中运行,即使它调用了 render_template("visualize.html"),该返回值也仅是一个字符串对象,被丢弃在后台线程中,完全不会发送给浏览器——这正是 visualize.html 从未显示的原因。

✅ 正确做法:分离“触发”与“获取结果”两个阶段

你需要将流程拆解为:

  1. 前端提交 → 后端启动训练任务(立即返回 loading 页面)
  2. 前端轮询(或 WebSocket)检查任务状态
  3. 后端提供独立接口返回训练结果(HTML 或 JSON)

下面是一个轻量、无需额外服务(如 Redis/Celery)的可行方案:

闪电说
闪电说

AI语音输入法

下载

✅ 方案一:使用内存缓存(适用于单进程开发/测试)

from flask import Flask, render_template, request, jsonify
import threading
import time
import uuid
from collections import defaultdict

app = Flask(__name__)

# 全局任务存储:task_id → {"status": "running"/"done", "html": "...", "plots": {...}}
TASKS = {}

@app.route("/inputs", methods=["POST"])
def inputs():
    file = request.files.get("dataset")
    algo = request.form.getlist("algo")

    if not file:
        return render_template("input_form.html", error="Please upload a CSV file", algos=ALGO, selected_algo=algo)

    data = io.StringIO(file.stream.read().decode("UTF8"), newline=None)
    dataset = pd.read_csv(data)

    task_id = str(uuid.uuid4())
    TASKS[task_id] = {"status": "running", "html": None}

    # 启动后台训练(传入 task_id 用于写回结果)
    train_thread = threading.Thread(
        target=train_models,
        args=(app, algo, dataset, task_id)
    )
    train_thread.daemon = True
    train_thread.start()

    return render_template("loading.html", task_id=task_id)

更新 train_models 函数,不渲染模板,只生成内容并存入缓存

def train_models(app, algo, dataset, task_id):
    print("Starting training...")
    with app.app_context():
        plot_model = Gen_Plot()
        plots = {}
        selected_algo = {}

        for model in algo:
            # ... 训练 & 生成图表逻辑 ...
            pass

        # ✅ 关键:生成 HTML 字符串,存入全局 TASKS
        result_html = render_template("visualize.html", plots=plots, selected_algo=selected_algo)
        TASKS[task_id] = {
            "status": "done",
            "html": result_html
        }
        print(f"Task {task_id} completed and HTML cached.")

✅ 前端轮询接口(/task_status/

@app.route("/task_status/")
def task_status(task_id):
    task = TASKS.get(task_id)
    if not task:
        return jsonify({"status": "not_found"}), 404
    if task["status"] == "done":
        return jsonify({"status": "done", "html": task["html"]})
    return jsonify({"status": "running"})

✅ 前端 loading.html 添加轮询逻辑(JavaScript)


Training models... please wait.

⚠️ 注意事项与进阶建议

  • 线程安全:TASKS 是简单字典,在单进程下可用;若部署多 Worker(如 Gunicorn),需改用 Redis 或数据库共享状态。
  • 内存泄漏:定期清理 TASKS 中过期任务(例如添加时间戳 + 后台清理线程)。
  • 生产环境强烈推荐 Celery + Redis:支持任务重试、优先级、进度跟踪、错误日志等,参考官方教程
  • 不要在后台线程中调用 render_template 并期望它“跳转”或“响应” —— Flask 的请求上下文(request, session, g)在线程间不共享,且响应流已由主线程关闭。

通过以上改造,你既能保持用户界面友好(显示加载页),又能确保结果 HTML 正确送达浏览器,真正实现“后台训练 + 前端按需展示”的异步体验。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
Python Flask框架
Python Flask框架

本专题专注于 Python 轻量级 Web 框架 Flask 的学习与实战,内容涵盖路由与视图、模板渲染、表单处理、数据库集成、用户认证以及RESTful API 开发。通过博客系统、任务管理工具与微服务接口等项目实战,帮助学员掌握 Flask 在快速构建小型到中型 Web 应用中的核心技能。

96

2025.08.25

Python Flask Web框架与API开发
Python Flask Web框架与API开发

本专题系统介绍 Python Flask Web框架的基础与进阶应用,包括Flask路由、请求与响应、模板渲染、表单处理、安全性加固、数据库集成(SQLAlchemy)、以及使用Flask构建 RESTful API 服务。通过多个实战项目,帮助学习者掌握使用 Flask 开发高效、可扩展的 Web 应用与 API。

73

2025.12.15

json数据格式
json数据格式

JSON是一种轻量级的数据交换格式。本专题为大家带来json数据格式相关文章,帮助大家解决问题。

430

2023.08.07

json是什么
json是什么

JSON是一种轻量级的数据交换格式,具有简洁、易读、跨平台和语言的特点,JSON数据是通过键值对的方式进行组织,其中键是字符串,值可以是字符串、数值、布尔值、数组、对象或者null,在Web开发、数据交换和配置文件等方面得到广泛应用。本专题为大家提供json相关的文章、下载、课程内容,供大家免费下载体验。

541

2023.08.23

jquery怎么操作json
jquery怎么操作json

操作的方法有:1、“$.parseJSON(jsonString)”2、“$.getJSON(url, data, success)”;3、“$.each(obj, callback)”;4、“$.ajax()”。更多jquery怎么操作json的详细内容,可以访问本专题下面的文章。

313

2023.10.13

go语言处理json数据方法
go语言处理json数据方法

本专题整合了go语言中处理json数据方法,阅读专题下面的文章了解更多详细内容。

79

2025.09.10

session失效的原因
session失效的原因

session失效的原因有会话超时、会话数量限制、会话完整性检查、服务器重启、浏览器或设备问题等等。详细介绍:1、会话超时:服务器为Session设置了一个默认的超时时间,当用户在一段时间内没有与服务器交互时,Session将自动失效;2、会话数量限制:服务器为每个用户的Session数量设置了一个限制,当用户创建的Session数量超过这个限制时,最新的会覆盖最早的等等。

320

2023.10.17

session失效解决方法
session失效解决方法

session失效通常是由于 session 的生存时间过期或者服务器关闭导致的。其解决办法:1、延长session的生存时间;2、使用持久化存储;3、使用cookie;4、异步更新session;5、使用会话管理中间件。

760

2023.10.18

Golang处理数据库错误教程合集
Golang处理数据库错误教程合集

本专题整合了Golang数据库错误处理方法、技巧、管理策略相关内容,阅读专题下面的文章了解更多详细内容。

2

2026.02.06

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
React 教程
React 教程

共58课时 | 4.8万人学习

TypeScript 教程
TypeScript 教程

共19课时 | 2.8万人学习

Bootstrap 5教程
Bootstrap 5教程

共46课时 | 3.2万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号