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)