Rubyonly/APP-Customer

1import streamlit as st2import pandas as pd3import classification  # 确保 classification.py 文件在同一目录下4 5# Load data6@st.cache7def load_data():8    activity_1 = pd.read_csv("dailyActivity_merged 3.12-4.11.csv")9    heartrate_1 = pd.read_csv("heartrate_seconds_merged 3.12-4.11.csv")10    weight_1 = pd.read_csv("weightLogInfo_merged 3.12-4.11.csv")11    12    activity_2 = pd.read_csv("dailyActivity_merged 4.12-5.12.csv")13    heartrate_2 = pd.read_csv("heartrate_seconds_merged 4.12-5.12.csv")14    weight_2 = pd.read_csv("weightLogInfo_merged 4.12-5.12.csv")15    16    return activity_1, heartrate_1, weight_1, activity_2, heartrate_2, weight_217 18activity_1, heartrate_1, weight_1, activity_2, heartrate_2, weight_2 = load_data()19# Process data20user_metrics_1 = classification.process_data(activity_1, heartrate_1, weight_1)21user_metrics_2 = classification.process_data(activity_2, heartrate_2, weight_2)22 23# Streamlit UI24st.title("User Risk Classification")25 26user_id = st.number_input("Enter User ID", min_value=1)27period = st.selectbox("Select Period", ["3.12-4.11", "4.12-5.12"])28 29if st.button("Classify User"):30    if period == "3.12-4.11":31        user_metrics = user_metrics_132    else:33        user_metrics = user_metrics_234    35    result = user_metrics[user_metrics['Id'] == user_id]36    if not result.empty:37        risk_category = result['Risk Category'].values[0]38        st.write(f"User {user_id} is classified as {risk_category} risk for period {period}.")39    else:40        st.write(f"User ID {user_id} not found in the selected period.")