File size: 590 Bytes
eadeab6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
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)
|