import gradio as gr
from generator import generate_text

def respond(message, history):
    # Trả về generator => Gradio sẽ tự stream
    return generate_text(message)

chatbot = gr.ChatInterface(
    fn=respond,
    title="🤖 Chatbot Tóm Tắt & Tổng Hợp Ý Chính",
    description="Nhập câu hỏi hoặc đoạn văn bản, chatbot sẽ tóm tắt & liệt kê ý chính.",
    type="messages",   # dùng format mới thay vì tuples (cảnh báo ở log)
    examples=[
        ["Tóm tắt nội dung bài viết về biến đổi khí hậu"],
        ["Cho mình ý chính của cuốn sách 'Nhà giả kim'"],
        ["Tóm tắt lại lịch sử Internet ngắn gọn"]
    ],
)

if __name__ == "__main__":
    chatbot.launch(server_name="0.0.0.0", server_port=7860)