import gradio as gr
from inference import TruDecide

model = TruDecide()

def predict(message):
    return model.answer(message)

demo = gr.Interface(
    fn=predict,
    inputs=gr.Textbox(lines=2, placeholder="请输入您的问题...", label="问题"),
    outputs=gr.Textbox(label="TruDecide的回答"),
    title="TruDecide.ai - 智能市场分析助手",
    description="我是TruDecide，你的智能市场分析助手。请向我提问关于金融市场的问题！",
    examples=[
        ["什么是股票市场?"],
        ["你是谁?"],
        ["什么是P/E比率?"],
        ["什么是量化交易?"],
        ["ETF是什么?"],
        ["解释一下债券"]
    ],
    flagging_mode="never"
)

demo.launch()