y / app.py
Tushdm's picture
s
2969792
Raw
History Blame Contribute Delete
1.09 kB
import wget
import os
import socket
hostname=socket.gethostname()
IPAddr=socket.gethostbyname(hostname)
if os.path.isfile("yolov7.pt") == False:
wget.download("https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7.pt")
import eventlet
import socketio
import base64
from blur import blur_check
from detect import analyze
sio = socketio.Server(cors_allowed_origins='*')
app = socketio.WSGIApp(sio)
MOCK_AI = os.getenv("MOCK_AI") == "true"
@sio.event
def connect(sid, environ):
print('connect ', sid)
@sio.event
def webcam(sid, data):
if MOCK_AI:
return {
"blurry": False,
"detected": [{
"type": "person",
"coords": [0,0,0,0],
"conf": 1
}]
}
detect = analyze(str(data))
output = {
"blurry": blur_check(data),
"detected": analyze(str(data))
}
print(output)
return output
@sio.event
def disconnect(sid):
print('disconnect ', sid)
if __name__ == '__main__':
eventlet.wsgi.server(eventlet.listen((IPAddr, 5000)), app)