| import streamlit as st | |
| from chatbot import predict_multi, generate_response | |
| st.set_page_config(page_title="๊ฐ์ ๊ณต๊ฐ ์ฑ๋ด") | |
| st.title("T / F ๊ฐ์ ๊ณต๊ฐ ์ฑ๋ด ๐ฌ") | |
| text = st.text_area("๋ง์์ ์ด์ผ๊ธฐ๋ฅผ ๋ค๋ ค์ฃผ์ธ์") | |
| mode = st.radio("์ฑ๋ด ์๋ต ์คํ์ผ", ["T - ์ด์ฑ์ ์กฐ์ธ", "F - ๊ฐ์ฑ์ ์๋ก"]) | |
| if st.button("์ฑ๋ด์๊ฒ ๋ง ๊ฑธ๊ธฐ"): | |
| flag = 'T' if mode.startswith('T') else 'F' | |
| emotions = predict_multi(text) | |
| st.write("### ์์ธก๋ ๊ฐ์ :", ", ".join(emotions)) | |
| response = generate_response(emotions, flag) | |
| st.success(response) | |