12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394 |
- # -*- coding:utf-8 -*-
- from flask import Flask, render_template, request, redirect, url_for, make_response, jsonify,Markup
- from werkzeug.utils import secure_filename
- from tools import get_text,get_image
- from datetime import timedelta
- import os
- from dev_image import run_cut
- import json
- ALLOWED_EXTENSIONS = set(['png', 'jpg', 'JPG', 'PNG', 'bmp'])
- def allowed_file(filename):
- return '.' in filename and filename.rsplit('.', 1)[1] in ALLOWED_EXTENSIONS
- app = Flask(__name__)
- # 设置静态文件缓存过期时间
- app.send_file_max_age_default = timedelta(seconds=1)
- # @app.route('/upload', methods=['POST', 'GET'])
- @app.route('/', methods=['POST', 'GET']) # 添加路由
- def upload():
- if request.method == 'POST':
- print(request.files)
- f = request.files['file']
- if not (f and allowed_file(f.filename)):
- return jsonify({"error": 1001, "msg": "请检查上传的图片类型,仅限于png、PNG、jpg、JPG、bmp"})
- # user_input = request.form.get("name")
- basepath = os.path.dirname(__file__) # 当前文件所在路径
- upload_path = os.path.join('./image_dir', secure_filename(f.filename)) # 注意:没有的文件夹一定要先创建,不然会提示没有该路径
- # upload_path = os.path.join(basepath, 'static/images','test.jpg') #注意:没有的文件夹一定要先创建,不然会提示没有该路径
- print(upload_path)
- f.save(upload_path)
- # 使用Opencv转换一下图片格式和名称
- # img = cv2.imread(upload_path)
- # cv2.imwrite(os.path.join(basepath, 'static/images', 'test.jpg'), img)
- # maple = requests.post('http://127.0.0.1:6666/static_pic',json={'path':upload_path}).text
- # print(maple)
- run_cut(upload_path,tabel=False)
- text = get_text()
- images = get_image()
- print(text)
- return render_template('show.html', w=500, contents=text,
- images=images
- )
- return render_template('upload.html')
- @app.route('/online', methods=['POST', 'GET']) # 添加路由
- def upload_online():
- if request.method == 'POST':
- f = request.files['file']
- if not (f and allowed_file(f.filename)):
- return jsonify({"error": 1001, "msg": "请检查上传的图片类型,仅限于png、PNG、jpg、JPG、bmp"})
- # user_input = request.form.get("name")
- basepath = os.path.dirname(__file__) # 当前文件所在路径
- upload_path = os.path.join('./image_dir', secure_filename(f.filename)) # 注意:没有的文件夹一定要先创建,不然会提示没有该路径
- # upload_path = os.path.join(basepath, 'static/images','test.jpg') #注意:没有的文件夹一定要先创建,不然会提示没有该路径
- print(upload_path)
- f.save(upload_path)
- # 使用Opencv转换一下图片格式和名称
- # img = cv2.imread(upload_path)
- # cv2.imwrite(os.path.join(basepath, 'static/images', 'test.jpg'), img)
- # maple = requests.post('http://127.0.0.1:6666/static_pic',json={'path':upload_path}).text
- # print(maple)
- run_cut(upload_path,tabel=False)
- text = get_text(online=False)
- images = get_image()
- return render_template('show.html', w=500, contents=text,
- images=images
- )
- return render_template('upload.html')
- if __name__ == '__main__':
- app.run('0.0.0.0',port=12535)
|