2023-01-10 01:52:32 +08:00

72 lines
1.2 KiB
Python

import os
from flask import Blueprint, Flask, render_template, jsonify, send_file, request
image_path = "/tmp/photo.jpg"
state = "put"
trash_type = ""
api = Blueprint('api', __name__)
# 放下垃圾
@api.route("/putdown")
def putdown():
global state
state = "camera"
return "ok"
# 拍好照
@api.route("/pic")
def pic():
global state
state = "identify"
return "ok"
# 辨識好
@api.route("/result")
def result():
global state, trash_type
trash = request.args.get("type")
if not trash:
return "sad", 400
trash_type = trash
state = "identified"
return "ok"
# 準備好下一個
@api.route("/ready")
def ready():
global state
state = "put"
return "ok"
app = Flask(__name__)
app.register_blueprint(api, url_prefix="/api")
@app.route("/")
def index():
return send_file("index.html")
@app.route("/poll")
def poll():
data = {
"state": state
}
if state == "identified":
data.update({"type": trash_type})
return jsonify(data)
@app.route("/photo")
def photo():
return send_file(image_path)
if __name__ == "__main__":
app.run(host='0.0.0.0', debug=True)