108 lines
4.6 KiB
Python
108 lines
4.6 KiB
Python
from flask import Flask, render_template, request, session, redirect, url_for, flash, jsonify
|
|
from flask_login import current_user
|
|
|
|
from flask_app.decorators import permission_required
|
|
from flask_app.models import Permission
|
|
from .predict_xray import detectxray
|
|
from .predict_ct import detectct
|
|
from .predict_sevxray import detectsev_xray
|
|
import os
|
|
from . import detect
|
|
@detect.route('/upload.html')
|
|
@permission_required(Permission.COMMENT)
|
|
def upload():
|
|
session['next'] = request.url # 将当前URL保存到session中
|
|
active_page = "index"
|
|
return render_template('detect/upload.html',active_page=active_page)
|
|
|
|
@detect.route('/upload_chest.html')
|
|
@permission_required(Permission.COMMENT)
|
|
def upload_chest():
|
|
session['next'] = request.url # 将当前URL保存到session中
|
|
return render_template('detect/upload_chest.html')
|
|
|
|
@detect.route('/upload_ct.html')
|
|
@permission_required(Permission.COMMENT)
|
|
def upload_ct():
|
|
session['next'] = request.url # 将当前URL保存到session中
|
|
return render_template('detect/upload_ct.html')
|
|
|
|
|
|
@detect.route('/uploaded_chest/<uid>', methods = ['POST', 'GET'])
|
|
def uploaded_chest(uid):
|
|
approot=detect.root_path
|
|
dir_path = os.path.dirname(approot) #flask_app
|
|
if request.method == 'POST':
|
|
# check if the post request has the file part
|
|
if 'file' not in request.files:
|
|
flash('No file part')
|
|
return redirect(request.url)
|
|
file = request.files['file']
|
|
if file.filename == '':
|
|
flash('No selected file')
|
|
return redirect(request.url)
|
|
if file:
|
|
temp_dir = os.path.join(dir_path, 'static/images/upload_img', str(uid))
|
|
if(not os.path.exists(temp_dir)):
|
|
os.makedirs(temp_dir)
|
|
file.save(os.path.join(temp_dir,'upload_chest.jpg')) #选择的图片保存到指定文件夹
|
|
return render_template('detect/predicting.html',
|
|
redirect_url=url_for('detect.result_chest'))
|
|
|
|
@detect.route('/result_chest')
|
|
@permission_required(Permission.COMMENT)
|
|
def result_chest():
|
|
session['next'] = request.url # 将当前URL保存到session中
|
|
approot = detect.root_path
|
|
dir_path = os.path.dirname(approot)
|
|
imagepath = os.path.join(dir_path, 'static/images/upload_img',str(current_user.id),'upload_chest.jpg')
|
|
pred_type, pred_normal, pred_pneu, pred_covid = detectxray(imagepath)
|
|
return render_template('detect/results_chest.html', pred_type=pred_type, pred_covid=pred_covid, pred_pneu=pred_pneu,
|
|
pred_normal=pred_normal)
|
|
|
|
@detect.route('/uploaded_ct/<uid>', methods = ['POST', 'GET'])
|
|
def uploaded_ct(uid):
|
|
approot=detect.root_path
|
|
dir_path = os.path.dirname(approot)
|
|
if request.method == 'POST':
|
|
# check if the post request has the file part
|
|
if 'file' not in request.files:
|
|
flash('No file part')
|
|
return redirect(request.url)
|
|
file = request.files['file']
|
|
if file.filename == '':
|
|
flash('No selected file')
|
|
return redirect(request.url)
|
|
if file:
|
|
temp_dir = os.path.join(dir_path, 'static/images/upload_img', str(uid))
|
|
if (not os.path.exists(temp_dir)):
|
|
os.makedirs(temp_dir)
|
|
file.save(os.path.join(temp_dir, 'upload_ct.jpg')) # 选择的图片保存到指定文件夹
|
|
return render_template('detect/predicting.html',
|
|
redirect_url=url_for('detect.result_ct'))
|
|
|
|
@detect.route('/result_ct')
|
|
@permission_required(Permission.COMMENT)
|
|
def result_ct():
|
|
session['next'] = request.url # 将当前URL保存到session中
|
|
approot = detect.root_path
|
|
dir_path = os.path.dirname(approot)
|
|
imagepath = os.path.join(dir_path, 'static/images/upload_img',str(current_user.id),'upload_ct.jpg')
|
|
pred_type,pred_normal,pred_pneu,pred_covid=detectct(imagepath)
|
|
return render_template('detect/results_ct.html',pred_type=pred_type,pred_covid=pred_covid,pred_pneu=pred_pneu,pred_normal=pred_normal)
|
|
|
|
@detect.route('/sev_xray')
|
|
def sev_xray():
|
|
return render_template('detect/predicting.html',
|
|
redirect_url=url_for('detect.result_sevxray'))
|
|
|
|
@detect.route('/result_sevxray')
|
|
@permission_required(Permission.COMMENT)
|
|
def result_sevxray():
|
|
session['next'] = request.url # 将当前URL保存到session中
|
|
approot = detect.root_path
|
|
dir_path = os.path.dirname(approot)
|
|
imagepath = os.path.join(dir_path, 'static/images/upload_img',str(current_user.id),'upload_chest.jpg')
|
|
pred_type,pred_mild,pred_sev=detectsev_xray(imagepath)
|
|
return render_template('detect/results_sevxray.html',pred_type=pred_type,pred_mild=pred_mild,pred_sev=pred_sev)
|