import gradio as gr
from emotion_predictor import predict_and_plot

def analyze_dialogue(text):
    return predict_and_plot(text)

iface = gr.Interface(
    fn=analyze_dialogue,
    inputs=gr.Textbox(lines=15, label="대화 입력 (형식: 화자: 발화문)"),
    outputs="html",
    title="KOTE 감정 예측 및 시각화",
    description="형식에 맞는 대화를 입력하면, 화자별 부정 감정 예측과 시각화 결과를 보여줍니다."
)

iface.launch()
#11