CaesarCloudSync
CaesarAI Deployed
9d3162f
from flask import Flask,request
from flask_cors import cross_origin
from flask_socketio import SocketIO,send,emit
from CaesarYolo import CaesarYolo
import numpy as np
import base64
caesaryolo = CaesarYolo()
app = Flask(__name__)
app.config['SECRET_KEY'] = 'secret!'
socketio = SocketIO(app)
@app.route("/",methods=["GET"])
@cross_origin()
def caesaraihome():
return "Welcome to CaesarAI's API's and CaesarAINL."
@app.route("/caesarobjectdetect",methods=["POST"])
def caesarobjectdetect():
frames = request.get_json()
#print(frames)
image = caesaryolo.caesar_object_detect(np.frombuffer(base64.b64decode(frames["frame"]),dtype="uint8").reshape(480,640,3))#base64.b64decode(frames["frame"]))
return {'frame': base64.b64encode(image).decode()}
@socketio.on('message')
def message(data):
print(data) # {'from': 'client'}
emit('response', {'from': 'server'})
@socketio.on('man')
def message(data):
print(data) # {'from': 'client'}
emit('response', {'from': 'server man'})
@socketio.on('caesarobjectdetect')
def caesarobjectdetect(image):
image = caesaryolo.caesar_object_detect(np.array(image["frame"]))
emit('caesarobjectresponse', {'frame': str(image)})
if __name__ == "__main__":
#port = int(os.environ.get('PORT', 5000)) # 80
app.run(debug=True,host="0.0.0.0",port=5000)
#socketio.run(app,debug=True,host="0.0.0.0",port=5000)