실습 1 — 예측 화면 만들기

from pathlib import Path

Path("out_app_predict.py").write_text('''"""배송 지연 예측 화면. 실행: streamlit run out_app_predict.py"""
import joblib
import pandas as pd
import streamlit as st


@st.cache_resource                       # 모델은 한 번만 불러 모든 사용자가 함께 쓴다
def load_bundle():
    return joblib.load("out_delay_model.joblib")


bundle = load_bundle()
choices = bundle["범주값"]
st.title("배송 지연 예측")
st.caption(f"모델 판 {bundle['판']} · 시험 AUC {bundle['시험성능']['AUC']}")

with st.form("order"):                   # 폼: 칸을 다 채우고 제출을 눌러야 한 번에 다시 실행된다
    left, right = st.columns(2)
    with left:
        지역 = st.selectbox("지역", choices["지역"], key="지역")
        상품분류 = st.selectbox("상품분류", choices["상품분류"], key="상품분류")
        택배사 = st.selectbox("택배사", choices["택배사"], key="택배사")
        날씨 = st.selectbox("날씨", choices["날씨"], key="날씨")
    with right:
        주문시각 = st.slider("주문시각", 0, 23, 14, key="주문시각")
        거리km = st.number_input("거리km", 0.5, 1000.0, 20.0, key="거리km")
        무게kg = st.number_input("무게kg", 0.0, 100.0, 2.0, key="무게kg")
    sent = st.form_submit_button("예측하기")

if sent:
    row = pd.DataFrame([{"주문시각": 주문시각, "거리km": 거리km, "무게kg": 무게kg, "지역": 지역,
                         "상품분류": 상품분류, "날씨": 날씨, "택배사": 택배사}])
    p = float(bundle["파이프라인"].predict_proba(row)[0, 1])
    st.metric("지연 확률", f"{p:.1%}")
    if p >= bundle["기준값"]:
        st.warning("지연될 수 있습니다. 고객에게 미리 알려 주세요.")
    else:
        st.success("제때 도착할 것으로 보입니다.")
''', encoding="utf-8")
print("화면 파일:", Path("out_app_predict.py").stat().st_size, "바이트")

from streamlit.testing.v1 import AppTest

# AppTest: 화면 스크립트를 이 프로그램 안에서 실행하고, 그려진 요소를 꺼내 볼 수 있게 한다
at = AppTest.from_file(str(Path("out_app_predict.py").resolve()), default_timeout=60).run()
print("제목:", at.title[0].value, "·", at.caption[0].value)
print("선택 상자:", [(s.label, s.value, len(s.options)) for s in at.selectbox])
print("단추:", [b.label for b in at.button], "· 지연 확률 칸:", len(at.metric))     # 제출 전이라 결과가 없다

# 밤 10시, 지방, 눈, 도서 주문(13차시와 같은 주문)
at.selectbox(key="지역").set_value("지방")
at.selectbox(key="상품분류").set_value("도서")
at.selectbox(key="택배사").set_value("빠른별택배")
at.selectbox(key="날씨").set_value("눈")
at.slider(key="주문시각").set_value(22)
at.number_input(key="거리km").set_value(197.6)
at.number_input(key="무게kg").set_value(4.6)
at.button[0].click().run()                      # 제출 단추를 누르면 스크립트가 다시 실행된다
print(at.metric[0].label, at.metric[0].value, "·", [w.value for w in at.warning], [s.value for s in at.success])

at.selectbox(key="지역").set_value("서울")
at.selectbox(key="날씨").set_value("맑음")
at.selectbox(key="택배사").set_value("한결택배")
at.slider(key="주문시각").set_value(9)
at.number_input(key="거리km").set_value(7.3)
at.button[0].click().run()
print(at.metric[0].label, at.metric[0].value, "·", [w.value for w in at.warning], [s.value for s in at.success])
print("오류:", [e.value for e in at.exception])