import threading
|
import socket
|
import re
|
import time
|
import datetime
|
import os
|
import numpy as np
|
import pandas as pd
|
import faiss
|
import cv2
|
from flask import Flask
|
from flask import request
|
from flask import render_template
|
|
# live stream for channel 0, change to right url first
|
live_stream_0 = "http://192.168.77.60/flv?app=live&stream=math_review_0"
|
|
math_expr_checking_df = pd.DataFrame(columns=["time", "total", "num_of_yes", "num_of_no", "num_of_invalid", "all_expressions", "screenshot"])
|
math_expr_checking_df['time'] = pd.to_datetime(math_expr_checking_df['time'])
|
|
'''
|
receive math expression checking result from socket (sent from VideoPipe)
|
'''
|
def expr_checking_result_receiver():
|
udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
bind_ip = '192.168.77.68'
|
bind_port = 3333
|
udp_socket.bind((bind_ip, bind_port))
|
|
while True:
|
data, addr = udp_socket.recvfrom(1024 * 64)
|
s = data.decode()
|
print('############################################')
|
print(f'Received data: \n{s}')
|
print(f'From address: {addr}')
|
|
li = re.findall("\<\-\-.*?\-\-\>", s, re.DOTALL)
|
print(f"total {len(li)} records parsed! going to append to data frame...")
|
for l in li:
|
items = str(l).split('\n')
|
'''
|
items[0] <--
|
items[1] time
|
items[2] channel index, frame index
|
items[3] total num, num of yes, num of no, number of invalid
|
items[4] 1st expression string, 2nd expression string, 3rd expression string, ...
|
items[5] screenshot path
|
items[6] -->
|
'''
|
assert(len(items) == 7)
|
# append items to dataframe
|
math_expr_checking_df.loc[len(math_expr_checking_df)] = [items[1].split('.')[0], items[3].split(',')[0], items[3].split(',')[1], items[3].split(',')[2], items[3].split(',')[3], items[4], items[5]]
|
print(f"now total {len(math_expr_checking_df)} records in math_expr_checking_df!")
|
print('############################################')
|
|
'''
|
create flask app
|
'''
|
app = Flask(__name__)
|
|
# home page
|
@app.route("/", methods=["GET", "POST"])
|
def home():
|
'''
|
filter properties
|
'''
|
select_channel = "all"
|
if 'select_channel' in request.form:
|
select_channel = request.form['select_channel']
|
|
# 1 hour by default
|
now = time.time()
|
past_1_hour = now - 3600
|
select_start_time = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(past_1_hour))
|
select_end_time = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(now))
|
|
if 'select_start_time' in request.form and request.form['select_start_time'] != '':
|
select_start_time = datetime.datetime.strptime(request.form['select_start_time'], '%Y-%m-%dT%H:%M:%S')
|
|
if 'select_end_time' in request.form and request.form['select_end_time'] != '':
|
select_end_time = datetime.datetime.strptime(request.form['select_end_time'], '%Y-%m-%dT%H:%M:%S')
|
|
sql = "total != ''"
|
if select_channel != 'all':
|
sql += f" & channel == '{select_channel}'"
|
sql += f" & time >= '{select_start_time}'"
|
sql += f" & time < '{select_end_time}'"
|
|
print(sql)
|
filter_df = math_expr_checking_df.query(sql)
|
print(filter_df)
|
expr_checking_results = []
|
for index, row in filter_df.iterrows():
|
expr_checking_results.append({'uid':index, \
|
'time':row['time'], \
|
'total':row['total'], \
|
'num_of_yes':row['num_of_yes'], \
|
'num_of_no':row['num_of_no'], \
|
'num_of_invalid':row['num_of_invalid'], \
|
'all_expressions':row['all_expressions'], \
|
'screenshot': row['screenshot']})
|
return render_template("home.html", \
|
expr_checking_results=expr_checking_results, \
|
total_records=len(filter_df), \
|
select_start_time = select_start_time, \
|
select_end_time = select_end_time, \
|
live_stream_0=live_stream_0)
|
|
if __name__ == "__main__":
|
# start expr checking results receiving thread
|
receiver = threading.Thread(target = expr_checking_result_receiver)
|
receiver.start()
|
|
# run server
|
app.run(host="192.168.77.68", port=3636)
|
