app.py

import gradio as gr
from utils import analyze_chat

# 创建Gradio界面
iface = gr.Interface(
    fn=analyze_chat,
    inputs=gr.File(label="上传微信聊天记录（JSON / TXT）"),
    outputs=[
        gr.Textbox(label="实体识别结果", lines=4),
        gr.Textbox(label="人物关系（含关系类型）", lines=6),
        gr.HTML(label="人物关系图谱")
    ],
    title="微信聊天人物关系分析",
    description="上传JSON或TXT格式的微信聊天记录，自动识别人物实体和关系类型，并生成互动图谱。",
    theme="compact"
)

if __name__ == "__main__":
    iface.launch()