kraken/backend/pages/3_test.py

"""IMU放置于人体模型后的motion capture分析，包括离线、在线。读入或在线接入数据进行波形绘制、数据分析及分析结果的人体模型渲染和结果图表"""
import streamlit as st
import os
from datetime import datetime
from db.models import test, subject
from components.remove_style import hide_footer

import page_utils


def _create_test(conn, subjects):
    with st.form("test_form"):
        st.write("创建训练")
        position = st.selectbox("训练部位", ['左手', '右手'])
        finger_model_names = st.multiselect("气动手手势", list(page_utils.fingermodel_trans.keys()))
        owner_name = st.selectbox("用户", subjects.name.to_list())
        model_path = page_utils.file_selector(os.path.join(f'./model/{owner_name}'))
        submitted = st.form_submit_button("开始训练")
        if submitted:
            start_time = datetime.strptime(datetime.now().strftime("%Y-%m-%d %H:%M:%S"), "%Y-%m-%d %H:%M:%S")
            test_new = {"position": position, 
                         "finger_model": ','.join([page_utils.fingermodel_trans[f] for f in finger_model_names]), 
                         "start_time": start_time, 
                         "owner_name": owner_name,
                         "model_path": model_path.name if model_path is not None else None}
            test.create_test(conn, test_new)
            return owner_name


def render():
    st.set_page_config(
        page_title="自由手势训练", page_icon=":chart_with_upwards_trend:"
    )
    hide_footer()

    st.markdown("# 手势训练")
    st.sidebar.success("训练")
    conn = st.connection("sql_app", type="sql")
    test.create_table(conn)
    subjects = subject.get_subjects(conn)
    sub_name = _create_test(conn, subjects)
    if sub_name:
        tests = test.get_tests(conn, sub_name)
        st.write("# 训练列表")
        st.dataframe(tests)

render()